Budgeted Active Experimentation for Treatment Effect Estimation from Observational and Randomized Data
Jiacan Gao, Xinyan Su, Mingyuan Ma, Yiyan HUANG, Xiao Xu, Xinrui Wan, Tianqi Gu, Enyun Yu, Jiecheng Guo, Zhiheng Zhang
Abstract
Estimating heterogeneous treatment effects is central to data-driven decision-making, yet industrial applications often face a fundamental tension between limited randomized controlled trial (RCT) budgets and abundant but biased observational data (OBS) collected under historical targeting policies. Although observational logs offer the advantage of scale, they may suffer from severe policy-induced imbalance and overlap violations, rendering standalone estimation unreliable. We propose a budgeted active experimentation framework that iteratively collects informative randomized samples for causal effect estimation via active sampling. By leveraging observational signals, we develop an acquisition function targeting uplift estimation uncertainty, domain discrepancy, and overlap deficits to select the most informative units for randomized experiments. We establish finite-sample deviation bounds, asymptotic normality via martingale CLTs, and minimax lower bounds showing near-optimality in the linear representation setting. Experiments on synthetic datasets support our theoretical findings, and further extensions to industrial neural network-based uplift modeling scenarios show that active sampling can improve sample efficiency over random sampling under limited RCT budgets.
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 93b497cc-0345-4d49-aa02-cc6ec49049bbCited by top-tier papers1
Ask how each one uses itBuilds on5
- Active Bayesian Causal InferenceChristian Toth, Lars Lorch, Christian Knoll, Andreas Krause et al.NeurIPS 2022 · 52 citations
- Active Learning with Neural Networks: Insights from Nonparametric StatisticsYinglun Zhu, Robert NowakNeurIPS 2022 · 15 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
- Enhancing Treatment Effect Estimation via Active Learning: A Counterfactual Covering PerspectiveHechuan Wen, Tong Chen, Mingming Gong, Li Kheng Chai et al.ICML 2025
- Active Treatment Effect Estimation via Limited SamplesZhiheng Zhang, Haoxiang Wang, Haoxuan Li, Zhouchen LinICML 2025
Related papers
- Budgeted Heterogeneous Treatment Effect EstimationTian Qin, Tian-Zuo Wang, Zhi-Hua ZhouICML 2021 · 18 citations
- Observationally Informed Adaptive Causal Experimental DesignErdun Gao, Liang Zhang, Jake Fawkes, Aoqi Zuo et al.KDD 2026 · 1 citation
- ABC3: Active Bayesian Causal Inference with Cohn Criteria in Randomized ExperimentsTaehun Cha, Donghun LeeAAAI 2025
- Causal-EPIG: Causally Aligned Active CATE EstimationErdun Gao, Jake Fawkes, Dino SejdinovicICML 2026 · 3 citations
- Robust Sampling for Active Statistical InferencePuheng Li, Tijana Zrnic, Emmanuel J. CandèsNeurIPS 2025 · 6 citations
