Adversarially Balanced Representation for Continuous Treatment Effect Estimation
DOI:
https://doi.org/10.1609/aaai.v38i12.29207Keywords:
ML: Representation Learning, ML: Causal LearningAbstract
Individual treatment effect (ITE) estimation requires adjusting for the covariate shift between populations with different treatments, and deep representation learning has shown great promise in learning a balanced representation of covariates. However the existing methods mostly consider the scenario of binary treatments. In this paper, we consider the more practical and challenging scenario in which the treatment is a continuous variable (e.g. dosage of a medication), and we address the two main challenges of this setup. We propose the adversarial counterfactual regression network (ACFR) that adversarially minimizes the representation imbalance in terms of KL divergence, and also maintains the impact of the treatment value on the outcome prediction by leveraging an attention mechanism. Theoretically we demonstrate that ACFR objective function is grounded in an upper bound on counterfactual outcome prediction error. Our experimental evaluation on semi-synthetic datasets demonstrates the empirical superiority of ACFR over a range of state-of-the-art methods.Downloads
Published
2024-03-24
How to Cite
Kazemi, A., & Ester, M. (2024). Adversarially Balanced Representation for Continuous Treatment Effect Estimation. Proceedings of the AAAI Conference on Artificial Intelligence, 38(12), 13085-13093. https://doi.org/10.1609/aaai.v38i12.29207
Issue
Section
AAAI Technical Track on Machine Learning III