Online Balanced Experimental Design
David Arbour, Drew Dimmery, Tung Mai, Anup B. Rao
Abstract
We consider the experimental design problem in an online environment, an important practical task for reducing the variance of estimates in randomized experiments which allows for greater precision, and in turn, improved decision mak-ing. In this work, we present algorithms that build on recent advances in online discrepancy minimization which accommodate both arbitrary treatment probabilities and multiple treatments. The proposed algorithms are computationally efficient, minimize covariate imbalance, and include randomization which enables robustness to misspecification. We provide worst case bounds on the expected mean squared error of the causal estimate and show that the proposed estimator is no worse than an implicit ridge regression, which are within a logarithmic factor of the best known results for offline experimental design. We con-clude with a detailed simulation study showing favorable results relative to complete randomization as well as to offline methods for experimental design with time complexities exceeding our algorithm, which has a linear dependence on the number of observations, by polynomial factors.
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 70d1501a-a547-4f62-a381-edb7be55b9fcCited by top-tier papers3
- 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
- Experimental Design for Multi-Channel Imaging via Task-Driven Feature SelectionStefano B. Blumberg, Paddy J. Slator, Daniel C. AlexanderICLR 2024 · 1 citation
- Nonlinear Covariate Balance in Experimental DesignQing Chen, Peng ZhangICML 2026
Builds on1
Related papers
- Sample Constrained Treatment Effect EstimationRaghavendra Addanki, David Arbour, Tung Mai, Cameron Musco et al.NeurIPS 2022 · 10 citations
- Online A-Optimal Design and Active Linear RegressionXavier Fontaine, Pierre Perrault, Michal Valko, Vianney PerchetICML 2021 · 20 citations
- CLIP-OGD: An Experimental Design for Adaptive Neyman Allocation in Sequential ExperimentsJessica Dai, Paula Gradu, Christopher HarshawNeurIPS 2023 · 21 citations
- Ordinal Secretaries with AdviceHasti Nourmohammadi Sigaroudi, Ying Cao, Bo Sun, Xiaoqi TanAAAI 2026
- On Efficient Estimation of Distributional Treatment Effects under Covariate-Adaptive RandomizationUndral Byambadalai, Tomu Hirata, Tatsushi Oka, Shota YasuiICML 2025
