Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
MissCNF trains Causal Normalizing Flows directly on partially observed data by maximizing the marginal likelihood of each sample, eliminating the need for listwise deletion or ad-hoc imputation. By exploiting the causal DAG's autoregressive structure, the approach restricts intractable marginalization strictly to missing variables in the observed set's ancestral closure while dropping non-ancestral missing nodes entirely. Across missing rates up to 90% under MCAR, MAR, and MNAR regimes, it reliably recovers the true data generating process and achieves the lowest counterfactual error in 20 of 24 nonlinear benchmark settings.
Causal generative models can now recover true joint distributions and counterfactual queries even when not a single complete record exists in the dataset.
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 \emph{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.