DeepMatch: Balancing Deep Covariate Representations for Causal Inference Using Adversarial Training
Nathan Kallus
Abstract
We study optimal covariate balance for causal inferences from observational data when rich covariates and complex relationships necessitate flexible modeling with neural networks. Standard approaches such as propensity weighting and matching/balancing fail in such settings due to miscalibrated propensity nets and inappropriate covariate representations, respectively. We propose a new method based on adversarial training of a weighting and a discriminator network that effectively addresses this methodological gap. This is demonstrated through new theoretical characterizations of the method as well as empirical results using both fully connected architectures to learn complex relationships and convolutional architectures to handle image confounders, showing how this new method can enable strong causal analyses in these challenging settings.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 95b75ca9-bf33-4cd7-9de5-eb66ffedc530Cited by top-tier papers11
- The Causal-Neural Connection: Expressiveness, Learnability, and InferenceKevin Xia, Kai-Zhan Lee, Yoshua Bengio, Elias BareinboimNeurIPS 2021 · 158 citations
- Optimal Transport for Treatment Effect EstimationHao Wang, Jiajun Fan, Zhichao Chen, Haoxuan Li et al.NeurIPS 2023 · 71 citations
- SyncTwin: Treatment Effect Estimation with Longitudinal OutcomesZhaozhi Qian, Yao Zhang, Ioana Bica, Angela M. Wood et al.NeurIPS 2021 · 46 citations
- Adversarial Counterfactual Learning and Evaluation for Recommender SystemDa Xu, Chuanwei Ruan, Evren Körpeoglu, Sushant Kumar et al.NeurIPS 2020 · 36 citations
- Reconsidering Generative Objectives For Counterfactual ReasoningDanni Lu, Chenyang Tao, Junya Chen, Fan Li et al.NeurIPS 2020 · 31 citations
Related papers
- Permutation WeightingDavid Arbour, Drew Dimmery, Arjun SondhiICML 2021 · 24 citations
- Conditional Instrumental Variable Regression with Representation Learning for Causal InferenceDebo Cheng, Ziqi Xu, Jiuyong Li, Lin Liu et al.ICLR 2024 · 14 citations
- Learning to Induce Causal StructureNan Rosemary Ke, Silvia Chiappa, Jane X. Wang, Jörg Bornschein et al.ICLR 2023 · 17 citations
- End-to-End Balancing for Causal Continuous Treatment-Effect EstimationMohammad Taha Bahadori, Eric Tchetgen Tchetgen, David HeckermanICML 2022 · 15 citations
- Deep Learning Methods for Proximal Inference via Maximum Moment RestrictionBenjamin Kompa, David R. Bellamy, Thomas Kolokotrones, James M. Robins et al.NeurIPS 2022 · 22 citations
