Adaptive Data-Borrowing for Improving Treatment Effect Estimation using External Controls
Qinwei Yang, Jingyi Li, Peng Wu
Abstract
Randomized controlled trials (RCTs) often exhibit limited inferential efficiency in estimating treatment effects due to small sample sizes. In recent years, the combination of external controls has gained increasing attention as a means of improving the efficiency of RCTs. However, external controls are not always comparable to RCTs, and direct borrowing without careful evaluation can introduce substantial bias and reduce the efficiency of treatment effect estimation. In this paper, we propose a novel influence-based adaptive sample borrowing approach that effectively quantifies the "comparability" of each sample in the external controls using influence function theory. Given a selected set of borrowed external controls, we further derive a semiparametric efficient estimator under an exchangeability assumption. Recognizing that the exchangeability assumption may not hold for all possible borrowing sets, we conduct a detailed analysis of the asymptotic bias and variance of the proposed estimator under violations of exchangeability. Building on this bias-variance trade-off, we further develop a data-driven approach to select the optimal subset of external controls for borrowing. Extensive simulations and realworld applications demonstrate that the proposed approach significantly enhances treatment effect estimation efficiency in RCTs, outperforming existing 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 feec8f75-4615-42a2-a833-99138b6c39bbBuilds on6
- Multiply Robust Federated Estimation of Targeted Average Treatment EffectsLarry Han, Zhu Shen, José R. ZubizarretaNeurIPS 2023 · 26 citations
- Policy Learning for Balancing Short-Term and Long-Term RewardsPeng Wu, Ziyu Shen, Feng Xie, Zhongyao Wang et al.ICML 2024 · 16 citations
- Learning the Optimal Policy for Balancing Short-Term and Long-Term RewardsQinwei Yang, Xueqing Liu, Yan Zeng, Ruocheng Guo et al.NeurIPS 2024 · 13 citations
- Learning Counterfactual Outcomes Under Rank PreservationPeng Wu, Haoxuan Li, Chunyuan Zheng, Yan Zeng et al.NeurIPS 2025 · 7 citations
- A Relative Error-Based Evaluation Framework of Heterogeneous Treatment Effect EstimatorsJiayi Guo, Haoxuan Li, Ye Tian, Peng WuICLR 2026 · 2 citations
Related papers
- Enhancing Statistical Validity and Power in Hybrid Controlled Trials: A Randomization Inference Approach with Conformal Selective BorrowingKe Zhu, Shu Yang, Xiaofei WangICML 2025
- Adaptive Instrument Design for Indirect ExperimentsYash Chandak, Shiv Shankar, Vasilis Syrgkanis, Emma BrunskillICLR 2024 · 5 citations
- Efficient Adaptive Experimentation with NoncomplianceMiruna Oprescu, Brian Cho, Nathan KallusNeurIPS 2025
- Sample Constrained Treatment Effect EstimationRaghavendra Addanki, David Arbour, Tung Mai, Cameron Musco et al.NeurIPS 2022 · 10 citations
- Quantifying Treatment Effects: Estimating Risk Ratios via Observational StudiesAhmed Boughdiri, Julie Josse, Erwan ScornetICML 2025
