English

Disentangled Representation Learning for Causal Inference with Instruments

Machine Learning 2024-12-09 v1 Artificial Intelligence Machine Learning

Abstract

Latent confounders are a fundamental challenge for inferring causal effects from observational data. The instrumental variable (IV) approach is a practical way to address this challenge. Existing IV based estimators need a known IV or other strong assumptions, such as the existence of two or more IVs in the system, which limits the application of the IV approach. In this paper, we consider a relaxed requirement, which assumes there is an IV proxy in the system without knowing which variable is the proxy. We propose a Variational AutoEncoder (VAE) based disentangled representation learning method to learn an IV representation from a dataset with latent confounders and then utilise the IV representation to obtain an unbiased estimation of the causal effect from the data. Extensive experiments on synthetic and real-world data have demonstrated that the proposed algorithm outperforms the existing IV based estimators and VAE-based estimators.

Keywords

Cite

@article{arxiv.2412.04641,
  title  = {Disentangled Representation Learning for Causal Inference with Instruments},
  author = {Debo Cheng and Jiuyong Li and Lin Liu and Ziqi Xu and Weijia Zhang and Jixue Liu and Thuc Duy Le},
  journal= {arXiv preprint arXiv:2412.04641},
  year   = {2024}
}

Comments

14 pages, 13 figures and 5 tables. Accepted by TNNLS

R2 v1 2026-06-28T20:24:57.385Z