Skip to yearly menu bar Skip to main content


Poster

Representation Learning for Treatment Effect Estimation from Observational Data

Liuyi Yao · Sheng Li · Yaliang Li · Mengdi Huai · Jing Gao · Aidong Zhang

Room 517 AB #116

Keywords: [ Representation Learning ]


Abstract:

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.

Live content is unavailable. Log in and register to view live content