arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2201.08559cs.LG
收藏

两阶段受控神经网络训练的个体疗效估计

Individual Treatment Effect Estimation Through Controlled Neural Network Training in Two Stages

Amazon(亚马逊)

Naveen Nair, Karthik S. Gurumoorthy, Dinesh Mandalapu

摘要:

我们开发了一个Cause-Deep神经网络(CDNN)模型,该模型在两个阶段进行训练,以推断单个单元水平的因果影响估计。在没有任何治疗信息的情况下,仅使用阶段1中的治疗前特征,我们学习最能代表结果的协变量的编码。在$2^{nd}$阶段,我们通过引入治疗指标变量以及编码协变量,进一步寻求预测第1阶段的不明结局。我们证明,即使没有显式计算处理残差,我们的方法仍然满足所需的局部奈曼正交性,使其对滋扰参数的小扰动具有鲁棒性。此外,通过与表示学习方法建立联系,我们创建了一个框架,从该框架中可以导出我们算法的多个变体。我们在公开的数据集上进行了初步实验,以比较这些变体,并在选择我们的CDNN方法的最佳变体时获得指导。在三个基准数据集上对CDNN与最先进方法进行评估时,我们观察到CDNN具有很强的竞争力,并且通常会产生最准确的个体治疗效果估计。我们强调了CDNN在可扩展性方面的强大优势。

英文摘要:

We develop a Causal-Deep Neural Network (CDNN) model trained in two stages to infer causal impact estimates at an individual unit level. Using only the pre-treatment features in stage 1 in the absence of any treatment information, we learn an encoding for the covariates that best represents the outcome. In the $2^{nd}$ stage we further seek to predict the unexplained outcome from stage 1, by introducing the treatment indicator variables alongside the encoded covariates. We prove that even without explicitly computing the treatment residual, our method still satisfies the desirable local Neyman orthogonality, making it robust to small perturbations in the nuisance parameters. Furthermore, by establishing connections with the representation learning approaches, we create a framework from which multiple variants of our algorithm can be derived. We perform initial experiments on the publicly available data sets to compare these variants and get guidance in selecting the best variant of our CDNN method. On evaluating CDNN against the state-of-the-art approaches on three benchmarking datasets, we observe that CDNN is highly competitive and often yields the most accurate individual treatment effect estimates. We highlight the strong merits of CDNN in terms of its extensibility to multiple use cases.

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

↑