Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction
Not on the map. Why?
Abstract (arXiv)
We propose a novel regression adjustment method designed for estimating distributional treatment effect parameters in randomized experiments. Randomized experiments have been extensively used to estimate treatment effects in various scientific fields. However, to gain deeper insights, it is essential to estimate distributional treatment effects rather than relying solely on average effects. Our approach incorporates pre-treatment covariates into a distributional regression framework, utilizing machine learning techniques to improve the precision of distributional treatment effect estimators. The proposed approach can be readily implemented with off-the-shelf machine learning methods and remains valid as long as the nuisance components are reasonably well estimated. Also, we establish the asymptotic properties of the proposed estimator and present a uniformly valid inference method. Through simulation results and real data analysis, we demonstrate the effectiveness of integrating machine learning techniques in reducing the variance of distributional treatment effect estimators in finite samples.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 23 of its 82 references that are in the map. First posted on arXiv 22 Jul 2024.
Builds on
23 of 82 references.
References · 23
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
- 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
- 2016Learning Representations for Counterfactual Inferencecited by 327
- 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 285
- 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 254
- 2017Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processescited by 156
- 2021Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic Regressioncited by 40
- 2015The Influence Function of Semiparametric Estimatorscited by 36
- 2013Program Evaluation and Causal Inference with High-Dimensional Datacited by 29
- 2019Localized Debiased Machine Learning: Efficient Inference on Quantile Treatment Effects and Beyondcited by 23
- 2022Robust and Agnostic Learning of Conditional Distributional Treatment Effectscited by 18
11 more
- 2014Designing and Deploying Online Field Experimentscited by 17
- 2021Elastic Net Regularization Paths for All Generalized Linear Modelscited by 17
- 2021Machine Learning for Variance Reduction in Online Experimentscited by 17
- 2023Causal isotonic calibration for heterogeneous treatment effectscited by 16
- 2016High-dimensional regression adjustments in randomized experimentscited by 10
- 2021Estimating Potential Outcome Distributions with Collaborating Causal Networkscited by 8
- 2023Proximal Causal Learning of Conditional Average Treatment Effectscited by 7
- 2024Regression Adjustment for Estimating Distributional Treatment Effects in Randomized Controlled Trialscited by 5
- 2016Generic Inference on Quantile and Quantile Effect Functions for Discrete Outcomescited by 4
- 2017Quantile Treatment Effects in Difference in Differences Models under Dependence Restrictions and with only Two Time Periodscited by 4
- 2021Regression-Adjusted Estimation of Quantile Treatment Effects under Covariate-Adaptive Randomizationscited by 4
Built on by
Citing papers per year: 2025: 4; 2026: 1.
5 papers.
- 2025On Efficient Estimation of Distributional Treatment Effects under Covariate-Adaptive Randomizationcited by 6
- 2025A Diffusion-Based Method for Learning the Multi-Outcome Distribution of Medical Treatmentscited by 3
- 2025Beyond the Average: Distributional Causal Inference under Imperfect Compliancecited by 2
- 2025Efficient and Scalable Estimation of Distributional Treatment Effects with Multi-Task Neural Networkscited by 1
- 2026Modeling Covariate Transition for Efficient Estimation of Longitudinal Treatment Effects in Randomized Experimentscited by 0
- 2026Modeling Covariate Transition for Efficient Estimation of Longitudinal Treatment Effects in Randomized Experiments29 May 2026
- 2025Beyond the Average: Distributional Causal Inference under Imperfect Compliance19 Sep 2025
- 2025Efficient and Scalable Estimation of Distributional Treatment Effects with Multi-Task Neural Networks10 Jul 2025
- 2025On Efficient Estimation of Distributional Treatment Effects under Covariate-Adaptive Randomization6 Jun 2025
- 2025A Diffusion-Based Method for Learning the Multi-Outcome Distribution of Medical Treatments2 Jun 2025