提出新算法提升因果推断中平均处理效应的自适应估计效率
Optimistic Algorithms for Adaptive Estimation of the Average Treatment Effect
- 基于乐观原则设计自适应采样策略,兼顾探索与利用
- 理论和实验均优于现有方法,显著提升估计精度
- 适合需要高效样本利用的实时因果分析场景
平均处理效应(ATE)的估计与推断是因果推断的核心,常作为复杂场景方法的基础。尽管传统上在批量设置下分析,但鞅理论的最新进展为自适应方法铺平了道路,可增强下游推断的效能。然而,对自适应算法的理解与开发仍处于早期阶段。现有工作或侧重渐近分析而忽略有限样本下的探索-利用权衡,或依赖更简单但次优的估计器。本文研究利用渐近最优的增广逆概率加权(AIPW)估计器的自适应采样过程,揭示了渐近方法掩盖的挑战,并提出一种类似多臂赌博机中乐观性的新型算法设计原则。该原理性方法在理论和实证上均显著优于先前方法,推动了自适应因果推断在理论与实践上的进展。
原文摘要 · Abstract (English)
Estimation and inference for the Average Treatment Effect (ATE) is a cornerstone of causal inference and often serves as the foundation for developing procedures for more complicated settings. Although traditionally analyzed in a batch setting, recent advances in martingale theory have paved the way for adaptive methods that can enhance the power of downstream inference. Despite these advances, progress in understanding and developing adaptive algorithms remains in its early stages. Existing work either focus on asymptotic analyses that overlook exploration-exploitation tradeoffs relevant in finite-sample regimes or rely on simpler but suboptimal estimators. In this work, we address these limitations by studying adaptive sampling procedures that take advantage of the asymptotically optimal Augmented Inverse Probability Weighting (AIPW) estimator. Our analysis uncovers challenges obscured by asymptotic approaches and introduces a novel algorithmic design principle reminiscent of optimism in multiarmed bandits. This principled approach enables our algorithm to achieve significant theoretical and empirical gains compared to prior methods. Our findings mark a step forward in advancing adaptive causal inference methods in theory and practice.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。