Treatment Effect Estimation using Invariant Risk Minimization

03/13/2021
by   Abhin Shah, et al.
19

Inferring causal individual treatment effect (ITE) from observational data is a challenging problem whose difficulty is exacerbated by the presence of treatment assignment bias. In this work, we propose a new way to estimate the ITE using the domain generalization framework of invariant risk minimization (IRM). IRM uses data from multiple domains, learns predictors that do not exploit spurious domain-dependent factors, and generalizes better to unseen domains. We propose an IRM-based ITE estimator aimed at tackling treatment assignment bias when there is little support overlap between the control group and the treatment group. We accomplish this by creating diversity: given a single dataset, we split the data into multiple domains artificially. These diverse domains are then exploited by IRM to more effectively generalize regression-based models to data regions that lack support overlap. We show gains over classical regression approaches to ITE estimation in settings when support mismatch is more pronounced.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
08/23/2022

Treatment Effect Estimation with Unmeasured Confounders in Data Fusion

In the presence of unmeasured confounders, we address the problem of tre...
research
07/30/2019

Incremental causal effects

This is a draft. The ignorability assumption is a key assumption in caus...
research
05/11/2023

Integrating nearest neighbors on neural network models for treatment effect estimation

Treatment effect estimation is of high-importance for both researchers a...
research
07/29/2021

Double-Robust Two-Way-Fixed-Effects Regression For Panel Data

We propose a new estimator for the average causal effects of a binary tr...
research
05/22/2019

Measuring Average Treatment Effect from Heavy-tailed Data

Heavy-tailed metrics are common and often critical to product evaluation...
research
01/14/2020

Learning Overlapping Representations for the Estimation of Individualized Treatment Effects

The choice of making an intervention depends on its potential benefit or...
research
07/09/2019

Characterization of Overlap in Observational Studies

Overlap between treatment groups is required for nonparametric estimatio...

Please sign up or login with your details

Forgot password? Click here to reset