Lune

ICML2026Top-tier venue

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

2026Year
1Citations
1Top-tier citations

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.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 93b497cc-0345-4d49-aa02-cc6ec49049bb

Cited by top-tier papers1

Ask how each one uses it

Builds on5

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines