Uncovering Hidden Treatment Effect Heterogeneity Through Optimal Transport
The conditional average treatment effect has all but become synonymous with “treatment effect heterogeneity.” However, treatment impacts the entire outcome distribution, not just its mean. Thus, there are two sources of heterogeneity to consider: across covariate strata and across quantiles. We develop a framework for characterizing and testing for both types of heterogeneous effects using the Wasserstein geometry. Specifically, we treat the difference between conditional quantile functions as a surface indexed by covariates and quantiles. This allows us to distinguish between covariate heterogeneity, rank heterogeneity, and covariate-rank interactions. The average of this surface over ranks is the conditional average treatment effect, and its norm is the conditional Wasserstein distance. We then define integrated distributional effects that aggregate distributional movement across strata; this allows us to measure the amount of distributional heterogeneity hidden by marginal analyses. For each new estimand, we develop efficient influence functions, estimation procedures, and interpretable tests. Furthermore, we generalize our framework to multivariate outcomes by exploring integrated Sinkhorn effects. Finally, we validate our theoretical results via simulation and apply our methods to a real-world dataset. Our framework thus unifies mean and distributional notions of heterogeneity.
Bio
Dr. Kyle Schindl is an assistant professor in the Department of Statistics at Iowa State University. In 2025, he completed his PhD in the Department of Statistics and Data Science at Carnegie Mellon University, where he was advised by Zach Branson, Edward H. Kennedy, and Joel Greenhouse. Before joining Carnegie Mellon, he received a Master of Science in Computational Analysis and Public Policy at the University of Chicago. Dr. Schindl is broadly interested in causal inference and experimental design. Much of his research centers around leveraging optimal transport theory to construct new causal inference tools. These causal tools are motivated by interdisciplinary collaborations in epidemiology, biostatistics, and economics.