Optimal Transport for Treatment Effect Estimation
Published in Conference on Neural Information Processing Systems 2023 (NeurIPS 2023), 2023
Recommended citation: Hao Wang, Jiajun Fan, Zhichao Chen, Haoran Li, Weiming Liu, Tianqiao Liu, Quanyu Dai, Yichao Wang, Zhenhua Dong, Ruiming Tang. "Optimal Transport for Treatment Effect Estimation." NeurIPS 2023. https://arxiv.org/abs/2310.18286
We propose an optimal transport-based framework for treatment effect estimation that addresses selection bias via balanced representation learning, achieving substantially better performance than state-of-the-art methods.