aiqbench analysis
This paper introduces Wasserstein Causal Forests (WCF), a machine learning method designed to estimate treatment effects when outcomes are probability distributions rather than single values. By leveraging optimal transport metrics, the approach effectively captures heterogeneous impacts on both the location and shape of outcome distributions, offering new tools for analyzing complex causal structures in fields like education and economics. While highly effective for distributional shifts, the method shows reduced accuracy compared to standard baselines in multimodal settings, highlighting specific boundaries for its application.
Wasserstein Causal Forests for Distribution-Valued Outcomes
arXiv:2609.35898v1 Announce Type: cross Abstract: This paper proposes Wasserstein Causal Forests (WCF) for settings in which each unit's outcome is itself a probability distribution. This study also defines finite-grid transformed average and conditional average treatment effects, including a reference-distance contrast that asks whether treatment moves unit-level distributions toward a prespecified benchmark. Simulations cover null effects, location and shape changes, limited overlap, equal-mean but different laws, heterogeneous effects, multimodality, and structural zeros. WCF is most accurate on the conditional-law metric in most reported designs and sharply improves reference-effect estimation in the principal location-and-shape settings, but it is less accurate than the forest baselines for multimodal settings. WCF is applied to the famous Project STAR \citep{word1990state}, revealing that small classes alter more than the mean: they raise within-grade mathematics achievement by $0.161$ standard deviations on average (SE $0.028$); but the gain is not a uniform location shift, it is larger in the upper part of the classroom score distribution ($+0.179$ at the ninetieth percentile versus $+0.091$ at the tenth) and, in descriptive stratum estimates, largest in the schools serving the most economically disadvantaged students (highest free-lunch quartile, $+0.262$, versus $+0.092$ to $+0.164$ elsewhere), while overall dispersion is essentially unchanged.
Abstract:This paper proposes Wasserstein Causal Forests (WCF) for settings in which each unit's outcome is itself a probability distribution. This study also defines finite-grid transformed average and conditional average treatment effects, including a reference-distance contrast that asks whether treatment moves unit-level distributions toward a prespecified benchmark. Simulations cover null effects, location and shape changes, limited overlap, equal-mean but different laws, heterogeneous effects, multimodality, and structural zeros. WCF is most accurate on the conditional-law metric in most reported designs and sharply improves reference-effect estimation in the principal location-and-shape settings, but it is less accurate than the forest baselines for multimodal settings. WCF is applied to the famous Project STAR \citep{word1990state}, revealing that small classes alter more than the mean: they raise within-grade mathematics achievement by $0.161$ standard deviations on average (SE $0.028$); but the gain is not a uniform location shift, it is larger in the upper part of the classroom score distribution ($+0.179$ at the ninetieth percentile versus $+0.091$ at the tenth) and, in descriptive stratum estimates, largest in the schools serving the most economically disadvantaged students (highest free-lunch quartile, $+0.262$, versus $+0.092$ to $+0.164$ elsewhere), while overall dispersion is essentially unchanged.
Machine Learning (stat.ML); Machine Learning (cs.LG)
arXiv:2609.35898 [stat.ML]
(or arXiv:2609.35898v1 [stat.ML] for this version)
https://doi.org/10.48550/arXiv.2609.35898 Focus to learn more arXiv-issued DOI via DataCite (pending registration)
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.
Full text under the publisher's open licence · 289 words
Related coverage
Bayesian Causal Forests for Zero-Inflated Continuous Outcomes
arXiv · Oct 15, 2026
Contextual Distributionally Robust Optimization with Causal and Continuous Structure
arXiv · Sep 30, 2026
Generative AI for Validating Physics Laws
arXiv · Aug 27, 2026
PIRAMID: Progress and Plans
LessWrong · Jul 27, 2026