Search papers, labs, and topics across Lattice.
MilaCNRSKyotoUkrainian Catholic UniversityUniversité du Québec à Montréal (UQAM)Mar 16, 2026arXiv:2603.15182This paper introduces Sequential Transport (ST), a novel framework for causal mediation analysis that leverages optimal transport (OT) to construct unit-level mediator counterfactuals while respecting the causal dependencies encoded in a mediator DAG. ST avoids cross-world counterfactual assumptions by minimally transporting mediator distributions, either marginally or conditionally, towards their distributions under alternative treatments. The authors demonstrate the consistency of estimated transport maps and induced direct/indirect effect decompositions, and showcase its performance on synthetic and real-world datasets like COMPAS.
Ditch the cross-world counterfactuals: Sequential Transport offers a DAG-aware, optimal transport approach to causal mediation analysis, providing deterministic counterfactual mediators and fine-grained attribution.
We propose sequential transport (ST), a distributional framework for mediation analysis that combines optimal transport (OT) with a mediator directed acyclic graph (DAG). Instead of relying on cross-world counterfactual assumptions, ST constructs unit-level mediator counterfactuals by minimally transporting each mediator, either marginally or conditionally, toward its distribution under an alternative treatment while preserving the causal dependencies encoded by the DAG. For numerical mediators, ST uses monotone (conditional) OT maps based on conditional CDF/quantile estimators; for categorical mediators, it extends naturally via simplex-based transport. We establish consistency of the estimated transport maps and of the induced unit-level decompositions into mutatis mutandis direct and indirect effects under standard regularity and support conditions. When the treatment is randomized or ignorable (possibly conditional on covariates), these decompositions admit a causal interpretation; otherwise, they provide a principled distributional attribution of differences between groups aligned with the mediator structure. Gaussian examples show that ST recovers classical mediation formulas, while additional simulations confirm good performance in nonlinear and mixed-type settings. An application to the COMPAS dataset illustrates how ST yields deterministic, DAG-consistent counterfactual mediators and a fine-grained mediator-level attribution of disparities.
