arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.02464cs.LGcs.AIstat.ML

跨多环境部分识别处理效应的元学习器

Meta-Learners for Partially-Identified Treatment Effects Across Multiple Environments

LMU Munich(慕尼黑大学);Munich Center for Machine Learning (MCML)(慕尼黑机器学习中心);University of Cambridge(剑桥大学)

Jonas Schweisthal, Dennis Frauen, Mihaela van der Schaar, Stefan Feuerriegel

AI总结:

针对多环境观测数据中存在重叠性和无混淆性违反的CATE估计问题,提出将环境作为工具变量进行部分识别的模型无关元学习器,并在模拟与真实数据上验证了其有效性。

摘要:

从观测数据中估计条件平均处理效应(CATE)与个性化医疗等诸多应用相关。本文聚焦于观测数据来自多环境(如不同医院、医生或国家)这一普遍设定。此外,我们允许违反标准因果假设,即环境内的重叠性和无混淆性。为此,我们放弃点识别,转而关注部分识别。具体而言,我们表明当前文献中关于多环境的假设允许我们将环境解释为工具变量(IV)。这使得我们能够利用跨环境的处理分配机制,改编 IV 文献中的界限以对 CATE 进行部分识别。然后,我们提出了不同的模型无关学习器(即所谓的元学习器)来估计这些界限,它们可与任意机器学习模型结合使用。我们进一步利用模拟和真实数据,通过各种实验证明了元学习器的有效性。最后,我们讨论了元学习器在工具变量设定(如存在不依从的随机对照试验)中部分识别的适用性。

英文摘要:

Estimating the conditional average treatment effect (CATE) from observational data is relevant for many applications such as personalized medicine. Here, we focus on the widespread setting where the observational data come from multiple environments, such as different hospitals, physicians, or countries. Furthermore, we allow for violations of standard causal assumptions, namely, overlap within the environments and unconfoundedness. To this end, we move away from point identification and focus on partial identification. Specifically, we show that current assumptions from the literature on multiple environments allow us to interpret the environment as an instrumental variable (IV). This allows us to adapt bounds from the IV literature for partial identification of CATE by leveraging treatment assignment mechanisms across environments. Then, we propose different model-agnostic learners (so-called meta-learners) to estimate the bounds that can be used in combination with arbitrary machine learning models. We further demonstrate the effectiveness of our meta-learners across various experiments using both simulated and real-world data. Finally, we discuss the applicability of our meta-learners to partial identification in instrumental variable settings, such as randomized controlled trials with non-compliance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑