On Inductive Biases for Heterogeneous Treatment Effect Estimation

06/07/2021
by   Alicia Curth, et al.
3

We investigate how to exploit structural similarities of an individual's potential outcomes (POs) under different treatments to obtain better estimates of conditional average treatment effects in finite samples. Especially when it is unknown whether a treatment has an effect at all, it is natural to hypothesize that the POs are similar - yet, some existing strategies for treatment effect estimation employ regularization schemes that implicitly encourage heterogeneity even when it does not exist and fail to fully make use of shared structure. In this paper, we investigate and compare three end-to-end learning strategies to overcome this problem - based on regularization, reparametrization and a flexible multi-task architecture - each encoding inductive bias favoring shared behavior across POs. To build understanding of their relative strengths, we implement all strategies using neural networks and conduct a wide range of semi-synthetic experiments. We observe that all three approaches can lead to substantial improvements upon numerous baselines and gain insight into performance differences across various experimental settings.

READ FULL TEXT
POST COMMENT

Comments

There are no comments yet.

Authors

page 1

page 2

page 3

page 4

01/31/2019

Learning Triggers for Heterogeneous Treatment Effects

The causal effect of a treatment can vary from person to person based on...
10/25/2018

Heterogeneous Treatment Effect Estimation through Deep Learning

Estimating heterogeneous treatment effect is an important task in causal...
02/02/2022

Can Transformers be Strong Treatment Effect Estimators?

In this paper, we develop a general framework based on the Transformer a...
07/24/2019

Sharp bounds on the relative treatment effect for ordinal outcomes

For ordinal outcomes, the average treatment effect is often ill-defined ...
12/17/2021

Selection bias in the treatment effect for a principal stratum

Estimation of treatment effect for principal strata has been studied for...
10/01/2021

Confounder importance learning for treatment effect inference

We address applied and computational issues for the problem of multiple ...
10/26/2020

Relative Contrast Estimation and Inference for Treatment Recommendation

When there are resource constraints, it is important to rank or estimate...
This week in AI

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