On Inductive Biases for Heterogeneous Treatment Effect Estimation

by   Alicia Curth, et al.

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.



There are no comments yet.


page 1

page 2

page 3

page 4


Learning Triggers for Heterogeneous Treatment Effects

The causal effect of a treatment can vary from person to person based on...

Heterogeneous Treatment Effect Estimation through Deep Learning

Estimating heterogeneous treatment effect is an important task in causal...

Can Transformers be Strong Treatment Effect Estimators?

In this paper, we develop a general framework based on the Transformer a...

Sharp bounds on the relative treatment effect for ordinal outcomes

For ordinal outcomes, the average treatment effect is often ill-defined ...

Selection bias in the treatment effect for a principal stratum

Estimation of treatment effect for principal strata has been studied for...

Confounder importance learning for treatment effect inference

We address applied and computational issues for the problem of multiple ...

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.