arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1912.09040cs.LGstat.ML
收藏

个体治疗效应估计中反事实推理选择偏差的降低

Reducing Selection Bias in Counterfactual Reasoning for Individual Treatment Effects Estimation

University of Alberta(阿尔伯塔大学)

Zichen Zhang, Qingfeng Lan, Lei Ding, Yue Wang, Negar Hassanpour, Russell Greiner

摘要:

反事实推理是一个重要的范式,适用于许多领域,如医疗保健,经济学和教育。在这项工作中,我们提出了一种新的方法来解决这个问题的\textit{selection bias}。我们学习了两组潜在随机变量,其中一组对应于仅导致选择偏差的变量,另一组与结果预测相关。它们由自动编码器学习,其中基于Pearson相关系数(PCC)的额外正则化损失鼓励两组随机变量之间的去相关。这允许通过仅保留与估计个体治疗效果相关的潜在变量来明确减轻选择偏倚。在合成玩具数据集和基准数据集上的实验结果表明,我们的算法能够实现最先进的性能,并改善其对应的结果,没有明确的模型的选择偏差。

英文摘要:

Counterfactual reasoning is an important paradigm applicable in many fields, such as healthcare, economics, and education. In this work, we propose a novel method to address the issue of \textit{selection bias}. We learn two groups of latent random variables, where one group corresponds to variables that only cause selection bias, and the other group is relevant for outcome prediction. They are learned by an auto-encoder where an additional regularized loss based on Pearson Correlation Coefficient (PCC) encourages the de-correlation between the two groups of random variables. This allows for explicitly alleviating selection bias by only keeping the latent variables that are relevant for estimating individual treatment effects. Experimental results on a synthetic toy dataset and a benchmark dataset show that our algorithm is able to achieve state-of-the-art performance and improve the result of its counterpart that does not explicitly model the selection bias.

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

↑