English
Related papers

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

200 papers

We study the problem of inferring heterogeneous treatment effects from time-to-event data. While both the related problems of (i) estimating treatment effects for binary or continuous outcomes and (ii) predicting survival outcomes have been…

Machine Learning · Computer Science 2022-01-25 Alicia Curth , Changhee Lee , Mihaela van der Schaar

Recent years, large scale clinical data like patient surveys and medical record data are playing an increasing role in medical data science. These large-scale clinical data, collectively referred to as "real-world data (RWD)". It is…

Methodology · Statistics 2024-02-23 Guanwenqing He , Ke Wan , Kazushi Maruo , Toshio Shimokawa

Estimating varying treatment effects in randomized trials with noncompliance is inherently challenging since variation comes from two separate sources: variation in the impact itself and variation in the compliance rate. In this setting,…

Applications · Statistics 2024-08-28 Jared D. Fisher , David W. Puelz , Sameer K. Deshpande

In this study, we focus on estimating the heterogeneous treatment effect (HTE) for survival outcome. The outcome is subject to censoring and the number of covariates is high-dimensional. We utilize data from both the randomized controlled…

Methodology · Statistics 2025-03-21 Xin Ye , Shu Yang , Xiaofei Wang , Yanyan Liu

Data analysis and machine learning have become an integrative part of the modern scientific methodology, offering automated procedures for the prediction of a phenomenon based on past observations, unraveling underlying patterns in data and…

Machine Learning · Statistics 2015-06-04 Gilles Louppe

Random forests is a state-of-the-art supervised machine learning method which behaves well in high-dimensional settings although some limitations may happen when $p$, the number of predictors, is much larger than the number of observations…

Methodology · Statistics 2019-02-01 Louis Capitaine , Robin Genuer , Rodolphe Thiébaut

Researchers often have to deal with heterogeneous population with mixed regression relationships, increasingly so in the era of data explosion. In such problems, when there are many candidate predictors, it is not only of interest to…

Methodology · Statistics 2021-02-05 Yan Li , Chun Yu , Yize Zhao , Robert H. Aseltine , Weixin Yao , Kun Chen

Adaptive experiments improve efficiency by adjusting treatment assignments based on past outcomes, but this adaptivity breaks the i.i.d.\ assumptions that underpin classical asymptotics. At the same time, many questions of interest are…

Machine Learning · Statistics 2026-05-05 Houssam Zenati , Bariscan Bozkurt , Arthur Gretton

We study targeted maximum likelihood estimation (TMLE) of the average treatment effect in a semiparametric regression model whose mean function is indexed by a finite-dimensional parameter, while the additive error distribution is left…

Methodology · Statistics 2026-04-20 Mijeong Kim

There is a dearth of robust methods to estimate the causal effects of multiple treatments when the outcome is binary. This paper uses two unique sets of simulations to propose and evaluate the use of Bayesian Additive Regression Trees…

Methodology · Statistics 2020-01-22 Liangyuan Hu , Chenyang Gu , Michael Lopez , Jiayi Ji , Juan Wisnivesky

Observational studies are often conducted to estimate causal effects of treatments or exposures on event-time outcomes. Since treatments are not randomized in observational studies, techniques from causal inference are required to adjust…

Methodology · Statistics 2023-10-23 Han Ji , Arman Oganisian

We investigate the estimation of the causal effect of a treatment variable on an outcome in the presence of a latent confounder. We first show that the causal effect is identifiable under certain conditions when data is available from…

Artificial Intelligence · Computer Science 2025-06-16 Yaroslav Kivva , Sina Akbari , Saber Salehkaleybar , Negar Kiyavash

Federated learning of causal estimands may greatly improve estimation efficiency by leveraging data from multiple study sites, but robustness to heterogeneity and model misspecifications is vital for ensuring validity. We develop a…

Methodology · Statistics 2023-10-06 Larry Han , Jue Hou , Kelly Cho , Rui Duan , Tianxi Cai

Causal machine learning methods which flexibly generate heterogeneous treatment effect estimates could be very useful tools for governments trying to make and implement policy. However, as the critical artificial intelligence literature has…

Econometrics · Economics 2023-09-06 Patrick Rehill , Nicholas Biddle

Scholars from diverse fields increasingly rely on high-frequency spatio-temporal data. Yet, causal inference with these data remains challenging due to spatial spillover and temporal carryover effects. We develop methods to estimate…

Methodology · Statistics 2025-11-03 Lingxiao Zhou , Kosuke Imai , Jason Lyall , Georgia Papadogeorgou

Decision Trees and Random Forests are among the most widely used machine learning models, and often achieve state-of-the-art performance in tabular, domain-agnostic datasets. Nonetheless, being primarily discriminative models they lack…

Machine Learning · Statistics 2020-07-14 Alvaro H. C. Correia , Robert Peharz , Cassio de Campos

Heterogeneous treatment effect estimation in high-stakes applications demands models that simultaneously optimize precision, interpretability, and calibration. Many existing tree-based causal inference techniques, however, exhibit high…

Machine Learning · Computer Science 2025-04-21 Yichen Liu

A new meta-algorithm for estimating the conditional average treatment effects is proposed in the paper. The main idea underlying the algorithm is to consider a new dataset consisting of feature vectors produced by means of concatenation of…

Machine Learning · Statistics 2019-09-10 Lev V. Utkin , Mikhail V. Kots , Viacheslav S. Chukanov

Accurately estimating heterogeneous treatment effects (HTE) in longitudinal settings is essential for personalized decision-making across healthcare, public policy, education, and digital marketing. However, time-varying interventions…

Methodology · Statistics 2025-10-28 Lei Shi , Sizhu Lu , Qiuran Lyu , Peng Ding , Nikos Vlassis

Statistical inference of heterogeneous treatment effects (HTEs) across predefined subgroups is challenging when units interact because treatment effects may vary by pre-treatment variables, post-treatment exposure variables (that measure…

Econometrics · Economics 2024-10-02 Julius Owusu