Counterfactual Identifiability via Dynamic Optimal Transport
About
We address the open question of counterfactual identification for high-dimensional multivariate outcomes from observational data. Pearl (2000) argues that counterfactuals must be identifiable (i.e., recoverable from the observed data distribution) to justify causal claims. A recent line of work on counterfactual inference shows promising results but lacks identification, undermining the causal validity of its estimates. To address this, we establish a foundation for multivariate counterfactual identification using continuous-time flows, including non-Markovian settings under standard criteria. We characterise the conditions under which flow matching yields a unique, monotone, and rank-preserving counterfactual transport map with tools from dynamic optimal transport, ensuring consistent inference. Building on this, we validate the theory in controlled scenarios with counterfactual ground-truth and demonstrate improvements in axiomatic counterfactual soundness on real images.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Counterfactual Inference | Counterfactual Ellipses Front-door setting | μAPE1.35 | 14 | |
| Counterfactual Inference | Counterfactual Ellipses Markovian setting | μAPE0.76 | 14 | |
| Counterfactual Effectiveness | MIMIC Chest X-ray 192×192 (test) | |ΔAUC| (Sex)0.067 | 13 | |
| Disease prediction | MIMIC Chest X-ray 192x192 | |ΔAUC| (%)0.023 | 10 | |
| Race Prediction | MIMIC Chest X-ray 192x192 | Absolute Delta AUC (%)0.05 | 10 | |
| Sex Prediction | MIMIC Chest X-ray 192x192 | Absolute Delta AUC18.7 | 10 | |
| Age Prediction | MIMIC Chest X-ray 192x192 | MAE (yr)0.333 | 10 | |
| Counterfactual Generation | MIMIC Chest X-ray 192x192 (test) | Composition MAE0.1835 | 4 |