Heterogeneous Treatment Effects Estimation: When Machine Learning meets multiple treatment regime

05/29/2022
by   Naoufal Acharki, et al.
7

In many scientific and engineering domains, inferring the effect of treatment and exploring its heterogeneity is crucial for optimization and decision making. In addition to Machine Learning based models (e.g. Random Forests or Neural Networks), many meta-algorithms have been developed to estimate the Conditional Average Treatment Effect (CATE) function in the binary setting, with the main advantage of not restraining the estimation to a specific supervised learning method. However, this task becomes more challenging when the treatment is not binary. In this paper, we investigate the Rubin Causal Model under the multi-treatment regime and we focus on estimating heterogeneous treatment effects. We generalize Meta-learning algorithms to estimate the CATE for each possible treatment value. Using synthetic and semi-synthetic simulation datasets, we assess the quality of each meta-learner in observational data, and we highlight in particular the performances of the X-learner.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
08/01/2022

Towards R-learner of conditional average treatment effects with a continuous treatment: T-identification, estimation, and inference

The R-learner has been popular in causal inference as a flexible and eff...
research
04/20/2023

B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under Hidden Confounding

Estimating heterogeneous treatment effects from observational data is a ...
research
05/19/2023

Meta-learning for heterogeneous treatment effect estimation with closed-form solvers

This article proposes a meta-learning method for estimating the conditio...
research
06/10/2022

Efficient Heterogeneous Treatment Effect Estimation With Multiple Experiments and Multiple Outcomes

Learning heterogeneous treatment effects (HTEs) is an important problem ...
research
07/28/2021

Doing Great at Estimating CATE? On the Neglected Assumptions in Benchmark Comparisons of Treatment Effect Estimators

The machine learning toolbox for estimation of heterogeneous treatment e...
research
01/26/2021

Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithms

The need to evaluate treatment effectiveness is ubiquitous in most of em...
research
08/14/2020

Semiparametric Estimation and Inference on Structural Target Functions using Machine Learning and Influence Functions

We aim to construct a class of learning algorithms that are of practical...

Please sign up or login with your details

Forgot password? Click here to reset