English
Related papers

Related papers: K-Fold Causal BART for CATE Estimation

200 papers

Datasets from field experiments with covariate-adaptive randomizations (CARs) usually contain extra covariates in addition to the strata indicators. We propose to incorporate these additional covariates via auxiliary regressions in the…

Econometrics · Economics 2022-09-07 Liang Jiang , Peter C. B. Phillips , Yubo Tao , Yichong Zhang

Causal inference methods are widely applied in various decision-making domains such as precision medicine, optimal policy and economics. Central to causal inference is the treatment effect estimation of intervention strategies, such as…

Artificial Intelligence · Computer Science 2021-05-31 Tri Dung Duong , Qian Li , Guandong Xu

This paper presents a novel nonlinear regression model for estimating heterogeneous treatment effects from observational data, geared specifically towards situations with small effect sizes, heterogeneous effects, and strong confounding.…

Methodology · Statistics 2019-11-14 P. Richard Hahn , Jared S. Murray , Carlos Carvalho

This paper introduces an innovative Bayesian machine learning algorithm to draw interpretable inference on heterogeneous causal effects in the presence of imperfect compliance (e.g., under an irregular assignment mechanism). We show,…

Methodology · Statistics 2021-12-02 Falco J. Bargagli-Stoffi , Kristof De-Witte , Giorgio Gnecco

In observational study, the propensity score has the central role to estimate causal effects. Since the propensity score is usually unknown, estimating by appropriate procedures is an indispensable step. A point to note that a causal effect…

Methodology · Statistics 2023-01-19 Shunichiro Orihara

We consider estimation of the target population average treatment effect (TATE) when outcome information is unavailable. Instead, we observe the outcome in multiple source populations and wish to combine the treatment effects therein to…

Methodology · Statistics 2025-05-16 Zehao Su , Helene Charlotte Rytgaard , Henrik Ravn , Frank Eriksson

Estimating individualised treatment effect (ITE) -- that is the causal effect of a set of variables (also called exposures, treatments, actions, policies, or interventions), referred to as \textit{composite treatments}, on a set of outcome…

Machine Learning · Computer Science 2025-12-19 Vinod Kumar Chauhan , Lei Clifton , Gaurav Nigam , David A. Clifton

We present a new package in R implementing Bayesian additive regression trees (BART). The package introduces many new features for data analysis using BART such as variable selection, interaction detection, model diagnostic plots,…

Machine Learning · Statistics 2014-11-25 Adam Kapelner , Justin Bleich

This paper develops a novel stochastic tree ensemble method for nonlinear regression, which we refer to as XBART, short for Accelerated Bayesian Additive Regression Trees. By combining regularization and stochastic search strategies from…

Machine Learning · Statistics 2021-06-04 Jingyu He , P. Richard Hahn

Estimating the conditional average treatment effect (CATE) from observational data plays a crucial role in areas such as e-commerce, healthcare, and economics. Existing studies mainly rely on the strong ignorability assumption that there…

Machine Learning · Computer Science 2025-01-28 Chuan Zhou , Yaxuan Li , Chunyuan Zheng , Haiteng Zhang , Haoxuan Li , Mingming Gong

We provide a new flexible framework for inference with the instrumental variable model. Rather than using linear specifications, functions characterizing the effects of instruments and other explanatory variables are estimated using machine…

Machine Learning · Statistics 2021-02-03 Robert E. McCulloch , Rodney A. Sparapani , Brent R. Logan , Purushottam W. Laud

There is interest in learning about the causal effects of modern contraceptive use on empowerment outcomes. Data on this question often come from family planning (FP) programs that increase access to FP and facilitate contraceptive use…

Methodology · Statistics 2026-03-31 Lucas Godoy Garraza , Ilene Speizer , Leontine Alkema

This paper introduces Type 2 Tobit Bayesian Additive Regression Trees (TOBART-2). BART can produce accurate individual-specific treatment effect estimates. However, in practice estimates are often biased by sample selection. We extend the…

Econometrics · Economics 2025-11-04 Eoghan O'Neill

We present a method for incorporating missing data in non-parametric statistical learning without the need for imputation. We focus on a tree-based method, Bayesian Additive Regression Trees (BART), enhanced with "Missingness Incorporated…

Machine Learning · Statistics 2014-02-14 Adam Kapelner , Justin Bleich

Estimating heterogeneous treatment effects is important to tailor treatments to those individuals who would most likely benefit. However, conditional average treatment effect predictors may often be trained on one population but possibly…

Machine Learning · Computer Science 2024-10-18 Christoph Kern , Michael Kim , Angela Zhou

Conditional Average Treatment Effect (CATE) estimation, at the heart of counterfactual reasoning, is a crucial challenge for causal modeling both theoretically and applicatively, in domains such as healthcare, sociology, or advertising.…

Machine Learning · Computer Science 2025-01-27 Armand Lacombe , Michèle Sebag

Current implementations of Bayesian Additive Regression Trees (BART) are based on axis-aligned decision rules that recursively partition the feature space using a single feature at a time. Several authors have demonstrated that oblique…

Machine Learning · Statistics 2024-11-14 Paul-Hieu V. Nguyen , Ryan Yee , Sameer K. Deshpande

The average treatment effect can obscure important heterogeneity when individuals respond differently to a treatment. While the conditional average treatment effect (CATE) function captures such heterogeneity, it is difficult to communicate…

Methodology · Statistics 2026-05-18 Anders Munch , Thomas A. Gerds

The proliferation of data has sparked significant interest in leveraging findings from one study to estimate treatment effects in a different target population without direct outcome observations. However, the transfer learning process is…

Methodology · Statistics 2025-05-15 Seok-Jin Kim , Hongjie Liu , Molei Liu , Kaizheng Wang

We introduce a new preference-based framework for conditional treatment effect estimation and policy learning, built on the Conditional Preference-based Treatment Effect (CPTE). CPTE requires only that outcomes be ranked under a preference…

Machine Learning · Statistics 2026-02-04 Dovid Parnas , Mathieu Even , Julie Josse , Uri Shalit