Learning Causal Normalizing Flows from Incomplete Data via Observed-Data Likelihood
Not on the map. Why?
Abstract (arXiv)
Causal Normalizing Flows (CNFs) enable causal inference from observational data given the causal structure, but they assume fully observed training data. We introduce MissCNF, which trains CNFs directly on incomplete data by maximizing the marginal likelihood of each partially observed sample, without discarding rows or constructing a completed dataset. Thanks to the causal structure encoded in the autoregressive factorization of CNFs, only missing variables in the ancestral closure of the observed set are integrated out, while the others are dropped without computation. We further establish the conditions under which MissCNF recovers the true joint distribution, and introduce causal-family positivity, where identification is possible even when no record in the dataset is ever complete. We compare MissCNF with two common strategies for handling missing data: listwise deletion and impute-then-fit pipelines. Across eight synthetic causal benchmarks, three missingness mechanisms, and missing rates up to 90%, MissCNF achieves the lowest KL divergence in 23 of 24 nonlinear MCAR and MAR settings and in all nonlinear MNAR settings, as well as the lowest counterfactual error in 20 of 24 settings. On linear SCMs, where linear imputation performs best, MissCNF ranks in the top two in 22 of 24 settings.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 14 of its 31 references that are in the map. First posted on arXiv 29 Sep 2026.
Builds on
14 of 31 references.
References · 14
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
- 2017Masked Autoregressive Flow for Density Estimationcited by 680
- 2018GAIN: Missing Data Imputation using Generative Adversarial Netscited by 349
- 2015MADE: Masked Autoencoder for Distribution Estimationcited by 306
- 2020Deep Structural Causal Models for Tractable Counterfactual Inferencecited by 210
- 2020Missing Data Imputation using Optimal Transportcited by 99
- 2022Deep End-to-end Causal Inferencecited by 93
- 2022HyperImpute: Generalized Iterative Imputation with Automatic Model Selectioncited by 81
- 2021MIRACLE: Causally-Aware Imputation via Learning Missing Data Mechanismscited by 50
- 2023Causal normalizing flows: from theory to practicecited by 49
- 2023ReMasker: Imputing Tabular Data with Masked Autoencodingcited by 35
- 2023Modeling Causal Mechanisms with Diffusion Models for Interventional and Counterfactual Queriescited by 23
Built on by
Not cited yet.