Timezone: »
Estimating individual treatment effect (ITE) is a challenging problem in causal inference, due to the missing counterfactuals and the selection bias. Existing ITE estimation methods mainly focus on balancing the distributions of control and treated groups, but ignore the local similarity information that is helpful. In this paper, we propose a local similarity preserved individual treatment effect (SITE) estimation method based on deep representation learning. SITE preserves local similarity and balances data distributions simultaneously, by focusing on several hard samples in each mini-batch. Experimental results on synthetic and three real-world datasets demonstrate the advantages of the proposed SITE method, compared with the state-of-the-art ITE estimation methods.
Author Information
Liuyi Yao (State University of New York at Buffalo)
Sheng Li (University of Georgia)
Yaliang Li (Tencent Medical AI Lab)
Mengdi Huai (State University of New York at Buffalo)
Jing Gao (University at Buffalo)
Aidong Zhang (SUNY Buffalo)
More from the Same Authors
-
2022 : Variational Graph Auto-Encoders for Heterogeneous Information Network »
Abhishek Dalvi · Ayan Acharya · Jing Gao · Vasant Honavar