On Efficient Estimation of Distributional Treatment Effects under Covariate-Adaptive Randomization
Undral Byambadalai, Tomu Hirata, Tatsushi Oka, Shota Yasui
Abstract
This paper focuses on the estimation of distributional treatment effects in randomized experiments that use covariate-adaptive randomization (CAR). These include designs such as Efron's biased-coin design and stratified block randomization, where participants are first grouped into strata based on baseline covariates and assigned treatments within each stratum to ensure balance across groups. In practice, datasets often contain additional covariates beyond the strata indicators. We propose a flexible distribution regression framework that leverages off-the-shelf machine learning methods to incorporate these additional covariates, enhancing the precision of distributional treatment effect estimates. We establish the asymptotic distribution of the proposed estimator and introduce a valid inference procedure. Furthermore, we derive the semiparametric efficiency bound for distributional treatment effects under CAR and demonstrate that our regressionadjusted estimator attains this bound. Simulation studies and empirical analyses of microcredit programs highlight the practical advantages of our method.
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 1a1cafe7-bd73-413c-ac48-2dc5aa285c1dCited by top-tier papers2
- Beyond the Average: Distributional Causal Inference under Imperfect ComplianceUndral Byambadalai, Tomu Hirata, Tatsushi Oka, Shota YasuiNeurIPS 2025 · 3 citations
- GDR-learners: Orthogonal Learning of Generative Models for Potential OutcomesValentyn Melnychuk, Stefan FeuerriegelICLR 2026 · 1 citation
Builds on6
- 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
- Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance ReductionUndral Byambadalai, Tatsushi Oka, Shota YasuiICML 2024 · 7 citations
- 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
- Using Surrogates in Covariate-adjusted Response-adaptive Randomization Experiments with Delayed OutcomesLei Shi, Waverly Wei, Jingshen WangNeurIPS 2024 · 4 citations
- Active Adaptive Experimental Design for Treatment Effect Estimation with Covariate ChoiceMasahiro Kato, Akihiro Oga, Wataru Komatsubara, Ryo InokuchiICML 2024 · 12 citations
- Quantifying Aleatoric Uncertainty of the Treatment Effect: A Novel Orthogonal LearnerValentyn Melnychuk, Stefan Feuerriegel, Mihaela van der SchaarNeurIPS 2024 · 13 citations
