Causal Effect Estimation using identifiable Variational AutoEncoder with Latent Confounders and Post-Treatment Variables

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Xie, Yang, Xu, Ziqi, Cheng, Debo, Li, Jiuyong, Liu, Lin, Zhang, Yinghao, Feng, Zaiwen
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929458440568832
author Xie, Yang
Xu, Ziqi
Cheng, Debo
Li, Jiuyong
Liu, Lin
Zhang, Yinghao
Feng, Zaiwen
author_facet Xie, Yang
Xu, Ziqi
Cheng, Debo
Li, Jiuyong
Liu, Lin
Zhang, Yinghao
Feng, Zaiwen
contents Estimating causal effects from observational data is challenging, especially in the presence of latent confounders. Much work has been done on addressing this challenge, but most of the existing research ignores the bias introduced by the post-treatment variables. In this paper, we propose a novel method of joint Variational AutoEncoder (VAE) and identifiable Variational AutoEncoder (iVAE) for learning the representations of latent confounders and latent post-treatment variables from their proxy variables, termed CPTiVAE, to achieve unbiased causal effect estimation from observational data. We further prove the identifiability in terms of the representation of latent post-treatment variables. Extensive experiments on synthetic and semi-synthetic datasets demonstrate that the CPTiVAE outperforms the state-of-the-art methods in the presence of latent confounders and post-treatment variables. We further apply CPTiVAE to a real-world dataset to show its potential application.
format Preprint
id arxiv_https___arxiv_org_abs_2408_07219
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Causal Effect Estimation using identifiable Variational AutoEncoder with Latent Confounders and Post-Treatment Variables
Xie, Yang
Xu, Ziqi
Cheng, Debo
Li, Jiuyong
Liu, Lin
Zhang, Yinghao
Feng, Zaiwen
Machine Learning
Methodology
Estimating causal effects from observational data is challenging, especially in the presence of latent confounders. Much work has been done on addressing this challenge, but most of the existing research ignores the bias introduced by the post-treatment variables. In this paper, we propose a novel method of joint Variational AutoEncoder (VAE) and identifiable Variational AutoEncoder (iVAE) for learning the representations of latent confounders and latent post-treatment variables from their proxy variables, termed CPTiVAE, to achieve unbiased causal effect estimation from observational data. We further prove the identifiability in terms of the representation of latent post-treatment variables. Extensive experiments on synthetic and semi-synthetic datasets demonstrate that the CPTiVAE outperforms the state-of-the-art methods in the presence of latent confounders and post-treatment variables. We further apply CPTiVAE to a real-world dataset to show its potential application.
title Causal Effect Estimation using identifiable Variational AutoEncoder with Latent Confounders and Post-Treatment Variables
topic Machine Learning
Methodology
url https://arxiv.org/abs/2408.07219