$\beta$-Intact-VAE: Identifying and Estimating Causal Effects under Limited Overlap

Abstract

As an important problem in causal inference, we discuss the identification and estimation of treatment effects (TEs) under limited overlap; that is, when subjects with certain features belong to a single treatment group. We use a latent variable to model a prognostic score which is widely used in biostatistics and sufficient for TEs; i.e., we build a generative prognostic model. We prove that the latent variable recovers a prognostic score, and the model identifies individualized treatment effects. The model is then learned as β-Intact-VAE--a new type of variational autoencoder (VAE). We derive the TE error bounds that enable representations balanced for treatment groups conditioned on individualized features. The proposed method is compared with recent methods using (semi-)synthetic datasets.

arXiv commentUpdated version of the NeurIPS 2021 submission (https://openreview.net/forum?id=Z3yd722b5X5). Largely improve readability and the presentation of experimental results. arXiv admin note: text overlap with arXiv:2109.15062, arXiv:2101.06662

References in corpus (19)

• Auto-Encoding Variational Bayes
• Matching Methods for Causal Inference: A Review and a Look Forward
• Semi-Supervised Learning with Deep Generative Models
• Tutorial on Variational Autoencoders
• Robust Inference on Average Treatment Effects with Possibly More Covariates than Observations
• Causal Effect Inference with Deep Latent-Variable Models
• Support and Invertibility in Domain-Invariant Representations
• Identifying Causal-Effect Inference Failure with Uncertainty-Aware Models
• Finite-Sample Optimal Estimation and Inference on Average Treatment Effects Under Unconfoundedness
• An Introduction to Proximal Causal Learning
• Generalization Bounds and Representation Learning for Estimation of Potential Outcomes and Causal Effects
• ICE-BeeM: Identifiable Conditional Energy-Based Deep Models Based on Nonlinear ICA
• Treatment effect estimation with disentangled latent factors
• Latent Causal Invariant Model
• More robust estimation of sample average treatment effects using Kernel Optimal Matching in an observational study of spine surgical interventions
• Intact-VAE: Estimating Treatment Effects under Unobserved Confounding
• Monotone function estimation in the presence of extreme data coarsening: Analysis of preeclampsia and birth weight in urban Uganda
• Estimating Average Treatment Effects with Support Vector Machines
• Deconfounding Scores: Feature Representations for Causal Effect Estimation with Weak Overlap