Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction
Undral Byambadalai, Tatsushi Oka, Shota Yasui
Abstract
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 96e16f0f-66d8-485a-a46f-c9f61f3dbb56Cited by top-tier papers3
- Beyond the Average: Distributional Causal Inference under Imperfect ComplianceUndral Byambadalai, Tomu Hirata, Tatsushi Oka, Shota YasuiNeurIPS 2025 · 3 citations
- On Efficient Estimation of Distributional Treatment Effects under Covariate-Adaptive RandomizationUndral Byambadalai, Tomu Hirata, Tatsushi Oka, Shota YasuiICML 2025
- A Diffusion-Based Method for Learning the Multi-Outcome Distribution of Medical TreatmentsYuchen Ma, Jonas Schweisthal, Hengrui Zhang, Stefan FeuerriegelKDD 2025
Builds on5
- Machine Learning for Variance Reduction in Online ExperimentsYongyi Guo, Dominic Coey, Mikael Konutgan, Wenting Li et al.NeurIPS 2021 · 47 citations
- Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic RegressionJunhyung Park, Uri Shalit, Bernhard Schölkopf, Krikamol MuandetICML 2021 · 46 citations
- Causal Isotonic Calibration for Heterogeneous Treatment EffectsLars van der Laan, Ernesto Ulloa-Pérez, Marco Carone, Alex LuedtkeICML 2023 · 18 citations
- Estimating Heterogeneous Treatment Effects: Mutual Information Bounds and Learning AlgorithmsXingzhuo Guo, Yuchen Zhang, Jianmin Wang, Mingsheng LongICML 2023 · 11 citations
- Proximal Causal Learning of Conditional Average Treatment EffectsErik Sverdrup, Yifan CuiICML 2023 · 7 citations
Related papers
- Modeling Covariate Transition for Efficient Estimation of Longitudinal Treatment Effects in Randomized ExperimentsNaoki Chihara, Tatsushi Oka, Yasuko Matsubara, Yasushi Sakurai et al.ICML 2026
- Finite Population Regression Adjustment and Non-asymptotic Guarantees for Treatment Effect EstimationMehrdad Ghadiri, David Arbour, Tung Mai, Cameron Musco et al.NeurIPS 2023 · 9 citations
- An Efficient Doubly-Robust Test for the Kernel Treatment EffectDiego Martinez-Taboada, Aaditya Ramdas, Edward KennedyNeurIPS 2023 · 17 citations
- Coordinated Double Machine LearningNitai Fingerhut, Matteo Sesia, Yaniv RomanoICML 2022 · 5 citations
- Double Machine Learning Density Estimation for Local Treatment Effects with InstrumentsYonghan Jung, Jin Tian, Elias BareinboimNeurIPS 2021 · 15 citations
