arrow
Return

Learning Decomposed Representations for Treatment Effect Estimation

delete2022-01-01
delete18
delete
OA
AI
A
Anpeng Wu
J
Junkun Yuan
K
Kun Kuang *
黎
黎波 (Bo Li) *
武润泽 cover
武润泽 (Runze Wu)
Q
Qiang Zhu
Wu Fei cover
Wu Fei (Fei Wu)
DOI:10.1109/TKDE.2022.3150807delete
deleteOriginal
deleteShare
deleteSave
Abstract

Abstract

En 中文
In observational studies, confounder separation and balancing are the fundamental problems of treatment effect estimation. Most of the previous methods focused on addressing the problem of confounder balancing by treating all observed pre-treatment variables as confounders, ignoring confounder separation. In general, not all the observed pre-treatment variables are confounders that refer to the common causes of the treatment and the outcome, some variables only contribute to the treatment (i.e., instrumental variables) and some only contribute to the outcome (i.e., adjustment variables). Balancing those non-confounders, including instrumental variables and adjustment variables, would generate additional bias for treatment effect estimation. By modeling the different causal relations among observed pre-treatment variables, treatment variables and outcome variables, we propose a synergistic learning framework to i) separate confounders by learning decomposed representations of both confounders and non-confounders, ii) balance confounder with sample re-weighting technique, and simultaneously iii) estimate the treatment effect in observational studies via counterfactual inference. Empirical results on synthetic and real-world datasets demonstrate that the proposed method can precisely decompose confounders and achieve a more precise estimation of treatment effect than baselines.
Keywords:
Estimation
Instruments
Reactive power
Medical services
Measurement
Germanium
Drugs
Treatment effect
decomposed representation
confounder separation and balancing
counterfactual inference

Journal

IEEE Transactions on Knowledge and Data Engineering cover
IEEE Transactions on Knowledge and Data Engineering
IF:
10.4
Papers:
6.9K
Citations:
3.2W

Organization

Z
zhe jiang university
Scholars:
3
Papers: 2
Citations: 0
T
Tsinghua University
Scholars:
5.6K
Papers: 2.1K
Citations: 0
Cited Papers

Cited Papers

Paradoxical Correlation Pattern Mining
err2018-08-01
err3
errOAAI
errZhou, Wenjun; Xiong, Hui; Duan, Lian; Xiao, Keli; Mee, Robert
errShare
errSave
A Geometric Understanding of Deep Learning
err2020-03-01
err79
errOAAI
errLei, Na; An, Dongsheng; Guo, Yang; Su, Kehua; Liu, Shixia; Luo, Zhongxuan; Yau, Shing-Tung; Gu, Xianfeng
errShare
errSave
Causal Inference
err2020-03-01
err78
errOAAI
errKuang, Kun; Li, Lian; Geng, Zhi; Xu, Lei; Zhang, Kun; Liao, Beishui; Huang, Huaxin; Ding, Peng; Miao, Wang; Jiang, Zhichao
errShare
errSave
A Survey of Data-Driven and Knowledge-Aware eXplainable AI
err2020-01-01
err121
PREAI
errLi, Xiao-Hui; Cao, Caleb Chen; Shi, Yuhan; Bai, Wei; Gao, Han; Qiu, Luyu; Wang, Cong; Gao, Yuanyuan; Zhang, Shenjia; Xue, Xun; Chen, Lei
errShare
errSave
Why Is Green Finance Important?
err2019-01-01
err0
errOAAI
errJeffrey D. Sachs; Wing Thye Woo; Naoyuki Yoshino; Farhad Taghizadeh-Hesary
errShare
errSave
An Exploration of Parameter Redundancy in Deep Networks with Circulant Projections
err2015-12-01
err0
errOAAI
errYu Cheng; Felix X. Yu; Rogerio S. Feris; Sanjiv Kumar; Alok Choudhary; Shi-Fu Chang
errShare
errSave
Dark, Beyond Deep: A Paradigm Shift to Cognitive AI with Humanlike Common Sense
err2020-03-01
err79
errOAAI
errZhu, Yixin; Gao, Tao; Fan, Lifeng; Huang, Siyuan; Edmonds, Mark; Liu, Hangxin; Gao, Feng; Zhang, Chi; Qi, Siyuan; Wu, Ying Nian; Tenenbaum, Joshua B.; Zhu, Song-Chun
errShare
errSave
researcher View more