Active Treatment Effect Estimation via Limited Samples
Zhiheng Zhang, Haoxiang Wang, Haoxuan Li, Zhouchen Lin
Abstract
Designing experiments for causal effect estimation remains an enduring topic in both machine learning and statistics. While much of the existing statistical literature focuses on using central limit theorems to analyze asymptotic properties of estimators, a parallel line of research has emerged around theoretical tools that provide finite-sample error bounds, offering performance on par with-or superior to-the asymptotic approaches. These finite-sample results are especially relevant in active sampling settings where the sample size is limited (for instance, under privacy or cost constraints). In this paper, we develop a finite-sample estimator with sample complexity analysis and extend its applicability to social networks. Through simulations and real-world experiments, we show that our method achieves higher estimation accuracy with fewer samples than traditional estimators endowed with asymptotic normality and other estimators backed by finite-sample guarantees.
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 c270d571-7547-4bc6-b146-48afa8d408bdCited by top-tier papers6
- Online Experimental Design With Estimation-Regret Trade-off Under Network InterferenceZhiheng Zhang, Zichen WangNeurIPS 2025 · 12 citations
- Unveiling Environmental Sensitivity of Individual Gains in Influence MaximizationXinyan Su, Zhiheng Zhang, Jiyan Qiu, Zhaojuan Yue et al.NeurIPS 2025 · 9 citations
- Design-Based Bandits Under Network Interference: Trade-Off Between Regret and Statistical InferenceZichen Wang, Haoyang Hong, Chuanhao Li, Haoxuan Li et al.NeurIPS 2025 · 3 citations
- 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
Builds on7
- Learning Causal Effects on HypergraphsJing Ma, Mengting Wan, Longqi Yang, Jundong Li et al.KDD 2022 · 61 citations
- Active Bayesian Causal InferenceChristian Toth, Lars Lorch, Christian Knoll, Andreas Krause et al.NeurIPS 2022 · 52 citations
- 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
- Online A-Optimal Design and Active Linear RegressionXavier Fontaine, Pierre Perrault, Michal Valko, Vianney PerchetICML 2021 · 20 citations
- Sample Constrained Treatment Effect EstimationRaghavendra Addanki, David Arbour, Tung Mai, Cameron Musco et al.NeurIPS 2022 · 10 citations
Related papers
- A/B Testing in Dense Large-Scale Networks: Design and InferencePreetam Nandy, Kinjal Basu, Shaunak Chatterjee, Ye TuNeurIPS 2020 · 19 citations
- 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
- Error Bounds for the Network Scale-Up MethodSergio Díaz-Aranda, Juan Marcos Ramirez, Mohit Daga, Jaya Prakash Champati et al.KDD 2025
- Uncertainty for Active Learning on GraphsDominik Fuchsgruber, Tom Wollschläger, Bertrand Charpentier, Antonio Oroz et al.ICML 2024 · 17 citations
- Estimating Causal Effects Using Weighting-Based EstimatorsYonghan Jung, Jin Tian, Elias BareinboimAAAI 2020 · 37 citations
