MultiMBNN: Matched and Balanced Causal Inference with Neural Networks
Methodology
2021-08-17 v4 Machine Learning
Multiagent Systems
Abstract
Causal inference (CI) in observational studies has received a lot of attention in healthcare, education, ad attribution, policy evaluation, etc. Confounding is a typical hazard, where the context affects both, the treatment assignment and response. In a multiple treatment scenario, we propose the neural network based MultiMBNN, where we overcome confounding by employing generalized propensity score based matching, and learning balanced representations. We benchmark the performance on synthetic and real-world datasets using PEHE, and mean absolute percentage error over ATE as metrics. MultiMBNN outperforms the state-of-the-art algorithms for CI such as TARNet and Perfect Match (PM).
Cite
@article{arxiv.2004.13446,
title = {MultiMBNN: Matched and Balanced Causal Inference with Neural Networks},
author = {Ankit Sharma and Garima Gupta and Ranjitha Prasad and Arnab Chatterjee and Lovekesh Vig and Gautam Shroff},
journal= {arXiv preprint arXiv:2004.13446},
year = {2021}
}
Comments
7 pages, 3 figures, Accepted in ESANN 2020