Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Fabio De Sousa Ribeiro, Ainkaran Santhirasekaram, Ben Glocker• 2025

Related benchmarks

TaskDatasetResultRank
Counterfactual InferenceCounterfactual Ellipses Front-door setting
μAPE1.35
14
Counterfactual InferenceCounterfactual Ellipses Markovian setting
μAPE0.76
14
Counterfactual EffectivenessMIMIC Chest X-ray 192×192 (test)
|ΔAUC| (Sex)0.067
13
Disease predictionMIMIC Chest X-ray 192x192
|ΔAUC| (%)0.023
10
Race PredictionMIMIC Chest X-ray 192x192
Absolute Delta AUC (%)0.05
10
Sex PredictionMIMIC Chest X-ray 192x192
Absolute Delta AUC18.7
10
Age PredictionMIMIC Chest X-ray 192x192
MAE (yr)0.333
10
Counterfactual GenerationMIMIC Chest X-ray 192x192 (test)
Composition MAE0.1835
4
Showing 8 of 8 rows

Other info

Follow for update