تقرير
Optimal Transport for Treatment Effect Estimation
| العنوان: | Optimal Transport for Treatment Effect Estimation |
|---|---|
| المؤلفون: | Wang, Hao, Chen, Zhichao, Fan, Jiajun, Li, Haoxuan, Liu, Tianqiao, Liu, Weiming, Dai, Quanyu, Wang, Yichao, Dong, Zhenhua, Tang, Ruiming |
| سنة النشر: | 2023 |
| المجموعة: | Computer Science Statistics |
| مصطلحات موضوعية: | Computer Science - Machine Learning, Statistics - Applications, Statistics - Machine Learning |
| الوصف: | Estimating conditional average treatment effect from observational data is highly challenging due to the existence of treatment selection bias. Prevalent methods mitigate this issue by aligning distributions of different treatment groups in the latent space. However, there are two critical problems that these methods fail to address: (1) mini-batch sampling effects (MSE), which causes misalignment in non-ideal mini-batches with outcome imbalance and outliers; (2) unobserved confounder effects (UCE), which results in inaccurate discrepancy calculation due to the neglect of unobserved confounders. To tackle these problems, we propose a principled approach named Entire Space CounterFactual Regression (ESCFR), which is a new take on optimal transport in the context of causality. Specifically, based on the framework of stochastic optimal transport, we propose a relaxed mass-preserving regularizer to address the MSE issue and design a proximal factual outcome regularizer to handle the UCE issue. Extensive experiments demonstrate that our proposed ESCFR can successfully tackle the treatment selection bias and achieve significantly better performance than state-of-the-art methods. Comment: Accepted as NeurIPS 2023 Poster |
| نوع الوثيقة: | Working Paper |
| الوصول الحر: | http://arxiv.org/abs/2310.18286Test |
| رقم الانضمام: | edsarx.2310.18286 |
| قاعدة البيانات: | arXiv |
| الوصف غير متاح. |