Finite Population Regression Adjustment and Non-asymptotic Guarantees for Treatment Effect Estimation
Mehrdad Ghadiri, David Arbour, Tung Mai, Cameron Musco, Anup B. Rao
Abstract
The design and analysis of randomized experiments is fundamental to many areas, from the physical and social sciences to industrial settings. Regression adjustment is a popular technique to reduce the variance of estimates obtained from experiments, by utilizing information contained in auxiliary covariates. While there is a large literature within the statistics community studying various approaches to regression adjustment and their asymptotic properties, little focus has been given to approaches in the finite population setting with non-asymptotic accuracy bounds. Further, prior work typically assumes that an entire population is exposed to an experiment, whereas practitioners often seek to minimize the number of subjects exposed to an experiment, for ethical and pragmatic reasons. In this work, we study the problems of estimating the sample mean, individual treatment effects, and average treatment effect with regression adjustment. We propose approaches that use techniques from randomized numerical linear algebra to sample a subset of the population on which to perform an experiment. We give non-asymptotic accuracy bounds for our methods and demonstrate that they compare favorably with prior approaches.
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 b557d62d-0ab2-48b2-934b-e6aeac115a26Cited by top-tier papers7
- Causal-EPIG: Causally Aligned Active CATE EstimationErdun Gao, Jake Fawkes, Dino SejdinovicICML 2026 · 3 citations
- Observationally Informed Adaptive Causal Experimental DesignErdun Gao, Liang Zhang, Jake Fawkes, Aoqi Zuo et al.KDD 2026 · 1 citation
- Budgeted Active Experimentation for Treatment Effect Estimation from Observational and Randomized DataJiacan Gao, Xinyan Su, Mingyuan Ma, Yiyan HUANG et al.ICML 2026 · 1 citation
- Active Policy Optimization for Individualized Dosing via Gradient Variance MinimizationYi Wan, Xin Wang, Huanhuan ChenICML 2026
- Enhancing Treatment Effect Estimation via Active Learning: A Counterfactual Covering PerspectiveHechuan Wen, Tong Chen, Mingming Gong, Li Kheng Chai et al.ICML 2025
Builds on5
- Causal-BALD: Deep Bayesian Active Learning of Outcomes to Infer Treatment-Effects from Observational DataAndrew Jesson, Panagiotis Tigas, Joost van Amersfoort, Andreas Kirsch et al.NeurIPS 2021 · 42 citations
- Subquadratic Kronecker Regression with Applications to Tensor DecompositionMatthew Fahrbach, Gang Fu, Mehrdad GhadiriNeurIPS 2022 · 24 citations
- Budgeted Heterogeneous Treatment Effect EstimationTian Qin, Tian-Zuo Wang, Zhi-Hua ZhouICML 2021 · 18 citations
- Sample Constrained Treatment Effect EstimationRaghavendra Addanki, David Arbour, Tung Mai, Cameron Musco et al.NeurIPS 2022 · 10 citations
- Online Balanced Experimental DesignDavid Arbour, Drew Dimmery, Tung Mai, Anup B. RaoICML 2022 · 5 citations
Related papers
- Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance ReductionUndral Byambadalai, Tatsushi Oka, Shota YasuiICML 2024 · 7 citations
- On Efficient Estimation of Distributional Treatment Effects under Covariate-Adaptive RandomizationUndral Byambadalai, Tomu Hirata, Tatsushi Oka, Shota YasuiICML 2025
- Beyond the Average: Distributional Causal Inference under Imperfect ComplianceUndral Byambadalai, Tomu Hirata, Tatsushi Oka, Shota YasuiNeurIPS 2025 · 3 citations
- A/B Testing in Dense Large-Scale Networks: Design and InferencePreetam Nandy, Kinjal Basu, Shaunak Chatterjee, Ye TuNeurIPS 2020 · 19 citations
- Modeling Covariate Transition for Efficient Estimation of Longitudinal Treatment Effects in Randomized ExperimentsNaoki Chihara, Tatsushi Oka, Yasuko Matsubara, Yasushi Sakurai et al.ICML 2026
