arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.13805cs.LGecon.EMstat.ML

基于分布学习的加权治疗效应估计

Weighting-Based Treatment Effect Estimation via Distribution Learning

Dongcheng Zhang, Kunpeng Zhang

AI总结:

本文提出一种基于分布学习的加权方法,通过学习协变量条件分布并利用密度比作为权重,以缓解模型误设,实验表明其在估计ATT上优于现有加权方法且具有双稳健性。

摘要:

现有的治疗效应估计加权方法通常建立在倾向得分或协变量平衡的思想之上。它们通常对治疗分配或结果模型施加强假设以获得无偏估计,如线性或特定函数形式,这容易导致模型误设的主要缺陷。在本文中,我们旨在通过开发一种基于分布学习的加权方法来缓解这些问题。我们首先学习以治疗分配为条件的协变量的真实潜在分布,然后利用治疗组中协变量密度与对照组中协变量密度之比作为估计治疗效应的权重。具体而言,我们提出通过变量替换的可逆变换来近似治疗组和对照组中协变量的分布。为了展示我们方法的优越性、鲁棒性和泛化性,我们使用合成数据和真实数据进行了大量实验。从实验结果中,我们发现,我们的方法在使用观测数据估计平均处理效应(ATT)方面优于几种先进的仅加权基准方法,并且在结合加权与一些先进结果建模方法的双稳健估计框架下保持了其优势。

英文摘要:

Existing weighting methods for treatment effect estimation are often built upon the idea of propensity scores or covariate balance. They usually impose strong assumptions on treatment assignment or outcome model to obtain unbiased estimation, such as linearity or specific functional forms, which easily leads to the major drawback of model mis-specification. In this paper, we aim to alleviate these issues by developing a distribution learning-based weighting method. We first learn the true underlying distribution of covariates conditioned on treatment assignment, then leverage the ratio of covariates' density in the treatment group to that of the control group as the weight for estimating treatment effects. Specifically, we propose to approximate the distribution of covariates in both treatment and control groups through invertible transformations via change of variables. To demonstrate the superiority, robustness, and generalizability of our method, we conduct extensive experiments using synthetic and real data. From the experiment results, we find that our method for estimating average treatment effect on treated (ATT) with observational data outperforms several cutting-edge weighting-only benchmarking methods, and it maintains its advantage under a doubly-robust estimation framework that combines weighting with some advanced outcome modeling methods.

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

↑