UCL Discovery
UCL home » Library Services » Electronic resources » UCL Discovery

Shrinkage Bayesian Causal Forests for Heterogeneous Treatment Effects Estimation

Caron, A; Baio, G; Manolopoulou, I; (2022) Shrinkage Bayesian Causal Forests for Heterogeneous Treatment Effects Estimation. Journal of Computational and Graphical Statistics 10.1080/10618600.2022.2067549. (In press). Green open access

[thumbnail of 10618600.2022.2067549.pdf]
Preview
Text
10618600.2022.2067549.pdf - Published Version

Download (2MB) | Preview

Abstract

This article develops a sparsity-inducing version of Bayesian Causal Forests, a recently proposed nonparametric causal regression model that employs Bayesian Additive Regression Trees and is specifically designed to estimate heterogeneous treatment effects using observational data. The sparsity-inducing component we introduce is motivated by empirical studies where not all the available covariates are relevant, leading to different degrees of sparsity underlying the surfaces of interest in the estimation of individual treatment effects. The extended version presented in this work, which we name Shrinkage Bayesian Causal Forest, is equipped with an additional pair of priors allowing the model to adjust the weight of each covariate through the corresponding number of splits in the tree ensemble. These priors improve the model’s adaptability to sparse data generating processes and allow to perform fully Bayesian feature shrinkage in a framework for treatment effects estimation, and thus to uncover the moderating factors driving heterogeneity. In addition, the method allows prior knowledge about the relevant confounding covariates and the relative magnitude of their impact on the outcome to be incorporated in the model. We illustrate the performance of our method in simulated studies, in comparison to Bayesian Causal Forest and other state-of-the-art models, to demonstrate how it scales up with an increasing number of covariates and how it handles strongly confounded scenarios. Finally, we also provide an example of application using real-world data. Supplementary materials for this article are available online.

Type: Article
Title: Shrinkage Bayesian Causal Forests for Heterogeneous Treatment Effects Estimation
Open access status: An open access version is available from UCL Discovery
DOI: 10.1080/10618600.2022.2067549
Publisher version: https://doi.org/10.1080/10618600.2022.2067549
Language: English
Additional information: This is an Open Access article distributed under the terms of the Creative Commons Attribution-NonCommercial-NoDerivatives License (http://creativecommons.org/licenses/by-nc-nd/4.0/), which permits non-commercial re-use, distribution, and reproduction in any medium, provided the original work is properly cited, and is not altered, transformed, or built upon in any way.
Keywords: Bayesian nonparametrics, Causal inference, Heterogeneous treatment effects, Observational studies, Machine learning, Tree ensembles
UCL classification: UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Maths and Physical Sciences
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Maths and Physical Sciences > Dept of Statistical Science
UCL > Provost and Vice Provost Offices > UCL BEAMS
UCL
URI: https://discovery.ucl.ac.uk/id/eprint/10149945
Downloads since deposit
334Downloads
Download activity - last month
Download activity - last 12 months
Downloads by country - last 12 months

Archive Staff Only

View Item View Item