English
Related papers

Related papers: What Makes Forest-Based Heterogeneous Treatment Ef…

200 papers

Decision trees and random forest remain highly competitive for classification on medium-sized, standard datasets due to their robustness, minimal preprocessing requirements, and interpretability. However, a single tree suffers from high…

Machine Learning · Statistics 2025-12-02 Cencheng Shen , Yuexiao Dong , Carey E. Priebe

Standard supervised learning procedures are validated against a test set that is assumed to have come from the same distribution as the training data. However, in many problems, the test data may have come from a different distribution. We…

Machine Learning · Statistics 2019-08-28 Tim Coleman , Kimberly Kaufeld , Mary Frances Dorn , Lucas Mentch

Estimating treatment effects (TE) from observational data is a critical yet complex task in many fields, from healthcare and economics to public policy. While recent advances in machine learning and causal inference have produced powerful…

Machine Learning · Computer Science 2025-08-15 Jeroen Berrevoets , Julianna Piskorz , Robert Davis , Harry Amad , Jim Weatherall , Mihaela van der Schaar

The average treatment effect (ATE) is popularly used to assess the treatment effect. However, the ATE implicitly assumes a homogenous treatment effect even amongst individuals with different characteristics. In this paper, we mainly focus…

Methodology · Statistics 2016-03-10 Yunjian Yin , Lan Liu , Zhi Geng

Analyses of heterogeneous treatment effects (HTE) are common in applied causal inference research. However, when outcomes are latent variables assessed via psychometric instruments such as educational tests, standard methods ignore the…

Heterogeneity and comorbidity are two interwoven challenges associated with various healthcare problems that greatly hampered research on developing effective treatment and understanding of the underlying neurobiological mechanism. Very few…

Methodology · Statistics 2023-06-27 Richard A Watson , Hengrui Cai , Xinming An , Samuel McLean , Rui Song

Estimation of heterogeneous treatment effects is an essential component of precision medicine. Model and algorithm-based methods have been developed within the causal inference framework to achieve valid estimation and inference. Existing…

Methodology · Statistics 2021-05-10 Ruohong Li , Honglang Wang , Wanzhu Tu

Understanding the factors that trigger or prevent undesirable health outcomes across patient subpopulations is essential for designing targeted interventions. While randomized controlled trials and expert-led patient interviews are standard…

Artificial Intelligence · Computer Science 2026-05-28 Shishir Adhikari , Guido Muscioni , Mark Shapiro , Plamen Petrov , Elena Zheleva

One advantage of decision tree based methods like random forests is their ability to natively handle categorical predictors without having to first transform them (e.g., by using feature engineering techniques). However, in this paper, we…

Machine Learning · Statistics 2018-10-30 Timothy C. Au

The R package DynForest implements random forests for predicting a continuous, a categorical or a (multiple causes) time-to-event outcome based on time-fixed and time-dependent predictors. The main originality of DynForest is that it…

Machine Learning · Statistics 2024-04-12 Anthony Devaux , Cécile Proust-Lima , Robin Genuer

Estimation of individual treatment effect in observational data is complicated due to the challenges of confounding and selection bias. A useful inferential framework to address this is the counterfactual (potential outcomes) model which…

Machine Learning · Statistics 2017-01-23 Min Lu , Saad Sadiq , Daniel J. Feaster , Hemant Ishwaran

Tree-based algorithms such as random forests and gradient boosted trees continue to be among the most popular and powerful machine learning models used across multiple disciplines. The conventional wisdom of estimating the impact of a…

Machine Learning · Statistics 2022-01-03 Markus Loecher , Qi Wu

A new matching method is proposed for the estimation of the average treatment effect of social policy interventions (e.g., training programs or health care measures). Given an outcome variable, a treatment and a set of pre-treatment…

Statistics Theory · Mathematics 2007-06-13 Stefano Iacus , Giuseppe Porro

Estimating a causal effect from observational data can be biased if we do not control for self-selection. This selection is based on confounding variables that affect the treatment assignment and the outcome. Propensity score methods aim to…

Econometrics · Economics 2021-09-10 Daniel Jacob

Recent studies have adopted an approach of selecting accurate and diverse trees based on individual or collective performance within an ensemble for classification and regression problems. This work follows in the wake of these…

Applications · Statistics 2020-05-20 Naz Gul , Nosheen Faiz , Dan Brawn , Rafal Kulakowski , Zardad Khan , Berthold Lausen

Heterogeneous effect estimation plays a crucial role in causal inference, with applications across medicine and social science. Many methods for estimating conditional average treatment effects (CATEs) have been proposed in recent years,…

Statistics Theory · Mathematics 2023-08-22 Edward H. Kennedy

Statisticians show growing interest in estimating and analyzing heterogeneity in causal effects in observational studies. However, there usually exists a trade-off between accuracy and interpretability for developing a desirable estimator…

Methodology · Statistics 2023-06-26 Steven Siwei Ye , Yanzhen Chen , Oscar Hernan Madrid Padilla

Researchers are increasingly turning to machine learning (ML) algorithms to investigate causal heterogeneity in randomized experiments. Despite their promise, ML algorithms may fail to accurately ascertain heterogeneous treatment effects…

Methodology · Statistics 2024-04-23 Kosuke Imai , Michael Lingzhi Li

Recently, many causal estimators for Conditional Average Treatment Effect (CATE) and instrumental variable (IV) problems have been published and open sourced, allowing to estimate granular impact of both randomized treatments (such as A/B…

Machine Learning · Computer Science 2022-12-21 Egor Kraev , Timo Flesch , Hudson Taylor Lekunze , Mark Harley , Pere Planell Morell

In recent years, dynamically growing data and incrementally growing number of classes pose new challenges to large-scale data classification research. Most traditional methods struggle to balance the precision and computational burden when…

Machine Learning · Computer Science 2016-11-01 Tingting Xie , Yuxing Peng , Changjian Wang