Counterfactual Representation Learning with Balancing Weights

by   Serge Assaad, et al.

A key to causal inference with observational data is achieving balance in predictive features associated with each treatment type. Recent literature has explored representation learning to achieve this goal. In this work, we discuss the pitfalls of these strategies - such as a steep trade-off between achieving balance and predictive power - and present a remedy via the integration of balancing weights in causal learning. Specifically, we theoretically link balance to the quality of propensity estimation, emphasize the importance of identifying a proper target population, and elaborate on the complementary roles of feature balancing and weight adjustments. Using these concepts, we then develop an algorithm for flexible, scalable and accurate estimation of causal effects. Finally, we show how the learned weighted representations may serve to facilitate alternative causal learning procedures with appealing statistical features. We conduct an extensive set of experiments on both synthetic examples and standard benchmarks, and report encouraging results relative to state-of-the-art baselines.



There are no comments yet.


page 1

page 2

page 3

page 4


Adversarial Balancing-based Representation Learning for Causal Effect Inference with Observational Data

Learning causal effects from observational data greatly benefits a varie...

Double Robust Representation Learning for Counterfactual Prediction

Causal inference, or counterfactual prediction, is central to decision m...

CoBWeb: a user-friendly web application to estimate causal treatment effects from observational data using multiple algorithms

Background/aims: While randomized controlled trials are the gold standar...

The Counterfactual χ-GAN

Causal inference often relies on the counterfactual framework, which req...

Learning Representations for Counterfactual Inference

Observational studies are rising in importance due to the widespread acc...

Adaptive Multi-Source Causal Inference

Data scarcity is a tremendous challenge in causal effect estimation. In ...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.