DeepMatch: Balancing Deep Covariate Representations for Causal Inference Using Adversarial Training
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 42 papers in the map (data through 7 Oct 2026). First posted on arXiv 15 Feb 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 6; 2022: 11; 2023: 8; 2024: 8; 2025: 8; 2026: 1.
42 papers.
Sort:
By area:
- 2021The Causal-Neural Connection: Expressiveness, Learnability, and Inferencecited by 97
- 2022Neural Causal Models for Counterfactual Identification and Estimationcited by 54
- 2021Causal Effect Inference for Structured Treatmentscited by 38
- 2023Optimal Transport for Treatment Effect Estimationcited by 32
- 2022Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimationcited by 25
- 2022Image-based Treatment Effect Heterogeneitycited by 20
- 2024Neural Causal Abstractionscited by 20
- 2023Towards Causal Foundation Model: on Duality between Causal Inference and Attentioncited by 17
- 2022Deep Multi-Modal Structural Equations For Causal Effect Estimation With Unstructured Proxiescited by 13
- 2022Deep Causal Learning: Representation, Discovery and Inferencecited by 11
- 2021Finding Valid Adjustments under Non-ignorability with Minimal DAG Knowledgecited by 10
- 2023Integrating Earth Observation Data into Causal Inference: Challenges and Opportunitiescited by 8
30 more
- 2021Identifiable Energy-based Representations: An Application to Estimating Heterogeneous Causal Effectscited by 7
- 2024Continuous Treatment Effect Estimation Using Gradient Interpolation and Kernel Smoothingcited by 7
- 2022Neural Score Matching for High-Dimensional Causal Inferencecited by 6
- 2024Contrastive Balancing Representation Learning for Heterogeneous Dose-Response Curves Estimationcited by 6
- 2022Outcome Assumptions and Duality Theory for Balancing Weightscited by 5
- 2021Matched sample selection with GANs for mitigating attribute confoundingcited by 4
- 2022Estimating Causal Effects Under Image Confounding Bias with an Application to Poverty in Africacited by 4
- 2024Implicitly Guided Design with PropEn: Match your Data to Follow the Gradientcited by 4
- 2023Calibrated and Conformal Propensity Scores for Causal Effect Estimationcited by 3
- 2024PairNet: Training with Observed Pairs to Estimate Individual Treatment Effectcited by 3
- 2025Enhancing Treatment Effect Estimation via Active Learning: A Counterfactual Covering Perspectivecited by 3
- 2021Treatment Effect Estimation using Invariant Risk Minimizationcited by 2
- 2024DeCaf: A Causal Decoupling Framework for OOD Generalization on Node Classificationcited by 2
- 2022Estimating causal effects with optimization-based methods: A review and empirical comparisoncited by 1
- 2023Causal Dynamic Variational Autoencoder for Counterfactual Regression in Longitudinal Datacited by 1
- 2023Causal inference with Machine Learning-Based Covariate Representationcited by 1
- 2024Causal Effect Estimation using identifiable Variational AutoEncoder with Latent Confounders and Post-Treatment Variablescited by 1
- 2024Towards Representation Learning for Weighting Problems in Design-Based Causal Inferencecited by 1
- 2025Federated Instrumental Variable Analysis via Federated Generalized Method of Momentscited by 1
- 2022Causal Inference for Chatting Handoffcited by 0
- 2022Nonlinear Causal Discovery via Kernel Anchor Regressioncited by 0
- 2023Variational Counterfactual Prediction under Runtime Domain Corruptioncited by 0
- 2023Adversarial Distribution Balancing for Counterfactual Reasoningcited by 0
- 2025Leveraging a Simulator for Learning Causal Representations from Post-Treatment Covariates for CATEcited by 0
- 2025A Computational Theory for Efficient Mini Agent Evaluation with Causal Guaranteescited by 0
- 2025TSCAN: Context-Aware Uplift Modeling via Two-Stage Training for Online Merchant Business Diagnosiscited by 0
- 2025A Generative Framework for Causal Estimation via Importance-Weighted Diffusion Distillationcited by 0
- 2025Synergizing Deconfounding and Temporal Generalization For Time-series Counterfactual Outcome Estimationcited by 0
- 2025PIPCFR: Pseudo-outcome Imputation with Post-treatment Variables for Individual Treatment Effect Estimationcited by 0
- 2026Causal Risk Minimization for High-Dimensional Treatmentscited by 0
- 2026Causal Risk Minimization for High-Dimensional Treatments26 May 2026
- 2025PIPCFR: Pseudo-outcome Imputation with Post-treatment Variables for Individual Treatment Effect Estimation21 Dec 2025
- 2025Synergizing Deconfounding and Temporal Generalization For Time-series Counterfactual Outcome Estimation20 Nov 2025
- 2025Federated Instrumental Variable Analysis via Federated Generalized Method of Moments27 May 2025
- 2025A Generative Framework for Causal Estimation via Importance-Weighted Diffusion Distillation16 May 2025
- 2025Enhancing Treatment Effect Estimation via Active Learning: A Counterfactual Covering Perspective8 May 2025
- 2025TSCAN: Context-Aware Uplift Modeling via Two-Stage Training for Online Merchant Business Diagnosis26 Apr 2025
- 2025A Computational Theory for Efficient Mini Agent Evaluation with Causal Guarantees27 Mar 2025
- 2025Leveraging a Simulator for Learning Causal Representations from Post-Treatment Covariates for CATE7 Feb 2025
- 2024DeCaf: A Causal Decoupling Framework for OOD Generalization on Node Classification27 Oct 2024
- 2024Towards Representation Learning for Weighting Problems in Design-Based Causal Inference24 Sep 2024
- 2024Causal Effect Estimation using identifiable Variational AutoEncoder with Latent Confounders and Post-Treatment Variables13 Aug 2024
30 more
- 2024PairNet: Training with Observed Pairs to Estimate Individual Treatment Effect6 Jun 2024
- 2024Implicitly Guided Design with PropEn: Match your Data to Follow the Gradient28 May 2024
- 2024Contrastive Balancing Representation Learning for Heterogeneous Dose-Response Curves Estimation21 Mar 2024
- 2024Continuous Treatment Effect Estimation Using Gradient Interpolation and Kernel Smoothing27 Jan 2024
- 2024Neural Causal Abstractions5 Jan 2024
- 2023Adversarial Distribution Balancing for Counterfactual Reasoning28 Nov 2023
- 2023Causal inference with Machine Learning-Based Covariate Representation3 Nov 2023
- 2023Optimal Transport for Treatment Effect Estimation27 Oct 2023
- 2023Causal Dynamic Variational Autoencoder for Counterfactual Regression in Longitudinal Data16 Oct 2023
- 2023Towards Causal Foundation Model: on Duality between Causal Inference and Attention1 Oct 2023
- 2023Variational Counterfactual Prediction under Runtime Domain Corruption23 Jun 2023
- 2023Calibrated and Conformal Propensity Scores for Causal Effect Estimation1 Jun 2023
- 2023Integrating Earth Observation Data into Causal Inference: Challenges and Opportunities30 Jan 2023
- 2022Deep Causal Learning: Representation, Discovery and Inference7 Nov 2022
- 2022Nonlinear Causal Discovery via Kernel Anchor Regression30 Oct 2022
- 2022Causal Inference for Chatting Handoff6 Oct 2022
- 2022Neural Causal Models for Counterfactual Identification and Estimation30 Sep 2022
- 2022Image-based Treatment Effect Heterogeneity13 Jun 2022
- 2022Estimating Causal Effects Under Image Confounding Bias with an Application to Poverty in Africa13 Jun 2022
- 2022Deep Multi-Modal Structural Equations For Causal Effect Estimation With Unstructured Proxies18 Mar 2022
- 2022Outcome Assumptions and Duality Theory for Balancing Weights17 Mar 2022
- 2022Neural Score Matching for High-Dimensional Causal Inference1 Mar 2022
- 2022Estimating causal effects with optimization-based methods: A review and empirical comparison28 Feb 2022
- 2022Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimation2 Feb 2022
- 2021Identifiable Energy-based Representations: An Application to Estimating Heterogeneous Causal Effects6 Aug 2021
- 2021The Causal-Neural Connection: Expressiveness, Learnability, and Inference2 Jul 2021
- 2021Finding Valid Adjustments under Non-ignorability with Minimal DAG Knowledge22 Jun 2021
- 2021Causal Effect Inference for Structured Treatments3 Jun 2021
- 2021Matched sample selection with GANs for mitigating attribute confounding24 Mar 2021
- 2021Treatment Effect Estimation using Invariant Risk Minimization13 Mar 2021
- 2021The Causal-Neural Connection: Expressiveness, Learnability, and Inferencecited by 97
- 2022Neural Causal Models for Counterfactual Identification and Estimationcited by 54
- 2021Causal Effect Inference for Structured Treatmentscited by 38
- 2023Optimal Transport for Treatment Effect Estimationcited by 32
- 2022Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimationcited by 25
- 2022Image-based Treatment Effect Heterogeneitycited by 20
- 2024Neural Causal Abstractionscited by 20
- 2023Towards Causal Foundation Model: on Duality between Causal Inference and Attentioncited by 17
- 2022Deep Multi-Modal Structural Equations For Causal Effect Estimation With Unstructured Proxiescited by 13
- 2022Deep Causal Learning: Representation, Discovery and Inferencecited by 11
- 2021Finding Valid Adjustments under Non-ignorability with Minimal DAG Knowledgecited by 10
- 2023Integrating Earth Observation Data into Causal Inference: Challenges and Opportunitiescited by 8
12 most cited of 41 All 41 (JSON)
- 2026Causal Risk Minimization for High-Dimensional Treatments26 May 2026
- 2025PIPCFR: Pseudo-outcome Imputation with Post-treatment Variables for Individual Treatment Effect Estimation21 Dec 2025
- 2025Synergizing Deconfounding and Temporal Generalization For Time-series Counterfactual Outcome Estimation20 Nov 2025
- 2025Federated Instrumental Variable Analysis via Federated Generalized Method of Moments27 May 2025
- 2025Enhancing Treatment Effect Estimation via Active Learning: A Counterfactual Covering Perspective8 May 2025
- 2025TSCAN: Context-Aware Uplift Modeling via Two-Stage Training for Online Merchant Business Diagnosis26 Apr 2025
- 2025A Computational Theory for Efficient Mini Agent Evaluation with Causal Guarantees27 Mar 2025
- 2025Leveraging a Simulator for Learning Causal Representations from Post-Treatment Covariates for CATE7 Feb 2025
- 2024DeCaf: A Causal Decoupling Framework for OOD Generalization on Node Classification27 Oct 2024
- 2024Towards Representation Learning for Weighting Problems in Design-Based Causal Inference24 Sep 2024
- 2024Causal Effect Estimation using identifiable Variational AutoEncoder with Latent Confounders and Post-Treatment Variables13 Aug 2024
- 2024PairNet: Training with Observed Pairs to Estimate Individual Treatment Effect6 Jun 2024
12 newest of 41 All 41 (JSON)