Adversarial balancing-based representation learning for causal effect inference with observational data

Xin Du, Lei Sun, Wouter Duivesteijn, Alexander Nikolaev, Mykola Pechenizkiy

Onderzoeksoutput: Bijdrage aan tijdschriftTijdschriftartikelAcademic

40 Downloads (Pure)

Uittreksel

Learning causal effects from observational data greatly benefits a variety of domains such as healthcare, education and sociology. For instance, one could estimate the impact of a policy to decrease unemployment rate. The central problem for causal effect inference is dealing with the unobserved counterfactuals and treatment selection bias. The state-of-the-art approaches focus on solving these problems by balancing the treatment and control groups. However, during the learning and balancing process, highly predictive information from the original covariate space might be lost. In order to build more robust estimators, we tackle this information loss problem by presenting a method called Adversarial Balancing-based representation learning for Causal Effect Inference (ABCEI), based on the recent advances in deep learning. ABCEI uses adversarial learning to balance the distributions of treatment and control group in the latent representation space, without any assumption on the form of the treatment selection/assignment function. ABCEI preserves useful information for predicting causal effects under the regularization of a mutual information estimator. We conduct various experiments on several synthetic and real-world datasets. The experimental results show that ABCEI is robust against treatment selection bias, and matches/outperforms the state-of-the-art approaches.
Originele taal-2Engels
Artikelnummer1904.13335v1
Aantal pagina's17
TijdschriftarXiv
StatusGepubliceerd - 30 apr 2019

Vingerafdruk

Education
Experiments
Deep learning

Trefwoorden

  • cs.LG
  • stat.ML

Citeer dit

@article{fad5793898664c1ca8a2c7102e5c4e75,
title = "Adversarial balancing-based representation learning for causal effect inference with observational data",
abstract = "Learning causal effects from observational data greatly benefits a variety of domains such as healthcare, education and sociology. For instance, one could estimate the impact of a policy to decrease unemployment rate. The central problem for causal effect inference is dealing with the unobserved counterfactuals and treatment selection bias. The state-of-the-art approaches focus on solving these problems by balancing the treatment and control groups. However, during the learning and balancing process, highly predictive information from the original covariate space might be lost. In order to build more robust estimators, we tackle this information loss problem by presenting a method called Adversarial Balancing-based representation learning for Causal Effect Inference (ABCEI), based on the recent advances in deep learning. ABCEI uses adversarial learning to balance the distributions of treatment and control group in the latent representation space, without any assumption on the form of the treatment selection/assignment function. ABCEI preserves useful information for predicting causal effects under the regularization of a mutual information estimator. We conduct various experiments on several synthetic and real-world datasets. The experimental results show that ABCEI is robust against treatment selection bias, and matches/outperforms the state-of-the-art approaches.",
keywords = "cs.LG, stat.ML",
author = "Xin Du and Lei Sun and Wouter Duivesteijn and Alexander Nikolaev and Mykola Pechenizkiy",
year = "2019",
month = "4",
day = "30",
language = "English",
journal = "arXiv",
publisher = "Cornell University Library",

}

Adversarial balancing-based representation learning for causal effect inference with observational data. / Du, Xin; Sun, Lei; Duivesteijn, Wouter; Nikolaev, Alexander; Pechenizkiy, Mykola.

In: arXiv, 30.04.2019.

Onderzoeksoutput: Bijdrage aan tijdschriftTijdschriftartikelAcademic

TY - JOUR

T1 - Adversarial balancing-based representation learning for causal effect inference with observational data

AU - Du, Xin

AU - Sun, Lei

AU - Duivesteijn, Wouter

AU - Nikolaev, Alexander

AU - Pechenizkiy, Mykola

PY - 2019/4/30

Y1 - 2019/4/30

N2 - Learning causal effects from observational data greatly benefits a variety of domains such as healthcare, education and sociology. For instance, one could estimate the impact of a policy to decrease unemployment rate. The central problem for causal effect inference is dealing with the unobserved counterfactuals and treatment selection bias. The state-of-the-art approaches focus on solving these problems by balancing the treatment and control groups. However, during the learning and balancing process, highly predictive information from the original covariate space might be lost. In order to build more robust estimators, we tackle this information loss problem by presenting a method called Adversarial Balancing-based representation learning for Causal Effect Inference (ABCEI), based on the recent advances in deep learning. ABCEI uses adversarial learning to balance the distributions of treatment and control group in the latent representation space, without any assumption on the form of the treatment selection/assignment function. ABCEI preserves useful information for predicting causal effects under the regularization of a mutual information estimator. We conduct various experiments on several synthetic and real-world datasets. The experimental results show that ABCEI is robust against treatment selection bias, and matches/outperforms the state-of-the-art approaches.

AB - Learning causal effects from observational data greatly benefits a variety of domains such as healthcare, education and sociology. For instance, one could estimate the impact of a policy to decrease unemployment rate. The central problem for causal effect inference is dealing with the unobserved counterfactuals and treatment selection bias. The state-of-the-art approaches focus on solving these problems by balancing the treatment and control groups. However, during the learning and balancing process, highly predictive information from the original covariate space might be lost. In order to build more robust estimators, we tackle this information loss problem by presenting a method called Adversarial Balancing-based representation learning for Causal Effect Inference (ABCEI), based on the recent advances in deep learning. ABCEI uses adversarial learning to balance the distributions of treatment and control group in the latent representation space, without any assumption on the form of the treatment selection/assignment function. ABCEI preserves useful information for predicting causal effects under the regularization of a mutual information estimator. We conduct various experiments on several synthetic and real-world datasets. The experimental results show that ABCEI is robust against treatment selection bias, and matches/outperforms the state-of-the-art approaches.

KW - cs.LG

KW - stat.ML

M3 - Article

JO - arXiv

JF - arXiv

M1 - 1904.13335v1

ER -