Neuro-Causal Factor Analysis

Alex Markham, Mingyu Liu, Bryon Aragam, Liam Solus

Keywords: variational autoencoder, latent variable models, generative models, causality, factor analysis, causal discovery

TL;DR: We introduce Neuro-Causal Factor Analysis, a fully nonparametric approach for modeling data with latent causal factors via deep generative models constrained to respect the associated Markov factorization.

Abstract:

We revisit nonlinear factor analysis from a comparatively new perspective given by advancements in causal discovery and deep learning, introducing a framework for Neuro-Causal Factor Analysis (NCFA). Our approach is fully nonparametric: It identifies factors via latent causal discovery methods and then uses a variational autoencoder (VAE) that is constrained to abide by the Markov factorization of the distribution with respect to the learned graph. We evaluate NCFA on real and synthetic data sets, finding that it performs comparably to standard VAEs on data reconstruction tasks but with the advantages of sparser architecture, lower model complexity, and causal interpretability. Unlike traditional factor analysis methods, our NCFA method allows learning and reasoning about the latent factors underlying observed data from a justifiably causal perspective, even when the relations between factors and measurements are highly nonlinear.

Submission Number: 48