English

Counterfactual Identifiability via Dynamic Optimal Transport

Machine Learning 2026-03-25 v2 Artificial Intelligence Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2510.08294,
  title  = {Counterfactual Identifiability via Dynamic Optimal Transport},
  author = {Fabio De Sousa Ribeiro and Ainkaran Santhirasekaram and Ben Glocker},
  journal= {arXiv preprint arXiv:2510.08294},
  year   = {2026}
}

Comments

Accepted at NeurIPS 2025

R2 v1 2026-07-01T06:26:58.031Z