论文标题

使用机器学习通过观察网络数据进行治疗效果估算

Treatment Effect Estimation with Observational Network Data using Machine Learning

论文作者

Emmenegger, Corinne, Spohn, Meta-Lina, Elmer, Timon, Bühlmann, Peter

论文摘要

治疗效应估计的因果推理方法通常假设独立单位。但是,这个假设通常是值得怀疑的,因为单位可能会相互作用,从而导致它们之间的溢出作用。我们开发了增强的反可能性加权(AIPW),以估算和推断预期的平均治疗效果(EATE),并具有来自单个(社交)网络具有溢出效果的观察数据。与总体效应(例如全球平均治疗效果(GATE),EATE措施,预期和平均在所有单位中)相反,单位的结果如何受到其自身治疗的因果影响,对其他单位的溢出效应边缘化。我们使用插件机学习开发交叉拟合理论,以获得以参数速率收敛的半参数治疗效应估计器,并渐近地遵循高斯分布。使用依赖关系图而不是网络图开发了渐近学,这使我们明确地允许网络中直接邻居以外的溢出效应。我们将AIPW方法应用于瑞士学生生活研究数据,以调查学习花费的时间对学生社交网络的考试绩效的影响。

Causal inference methods for treatment effect estimation usually assume independent units. However, this assumption is often questionable because units may interact, resulting in spillover effects between them. We develop augmented inverse probability weighting (AIPW) for estimation and inference of the expected average treatment effect (EATE) with observational data from a single (social) network with spillover effects. In contrast to overall effects such as the global average treatment effect (GATE), the EATE measures, in expectation and on average over all units, how the outcome of a unit is causally affected by its own treatment, marginalizing over the spillover effects from other units. We develop cross-fitting theory with plugin machine learning to obtain a semiparametric treatment effect estimator that converges at the parametric rate and asymptotically follows a Gaussian distribution. The asymptotics are developed using the dependency graph rather than the network graph, which makes explicit that we allow for spillover effects beyond immediate neighbors in the network. We apply our AIPW method to the Swiss StudentLife Study data to investigate the effect of hours spent studying on exam performance accounting for the students' social network.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源