English
Related papers

Related papers: Meta-learners for Estimating Heterogeneous Treatme…

200 papers

Estimating heterogeneous treatment effects (HTEs) in time-varying settings is particularly challenging, as the probability of observing certain treatment sequences decreases exponentially with longer prediction horizons. Thus, the observed…

Machine Learning · Computer Science 2026-02-12 Konstantin Hess , Dennis Frauen , Mihaela van der Schaar , Stefan Feuerriegel

Orthogonal meta-learners, such as DR-learner, R-learner and IF-learner, are increasingly used to estimate conditional average treatment effects. They improve convergence rates relative to na\"{\i}ve meta-learners (e.g., T-, S- and…

Methodology · Statistics 2023-11-17 Stijn Vansteelandt , Paweł Morzywołek

Estimating heterogeneous treatment effects in network settings is complicated by interference, meaning that the outcome of an instance can be influenced by the treatment status of others. Existing causal machine learning approaches usually…

Machine Learning · Computer Science 2025-10-27 Daan Caljon , Jente Van Belle , Wouter Verbeke

The machine learning toolbox for estimation of heterogeneous treatment effects from observational data is expanding rapidly, yet many of its algorithms have been evaluated only on a very limited set of semi-synthetic benchmark datasets. In…

Machine Learning · Computer Science 2021-07-29 Alicia Curth , Mihaela van der Schaar

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

We propose a novel method, termed the M-learner, for estimating heterogeneous indirect and total treatment effects and identifying relevant subgroups within a mediation framework. The procedure comprises four key steps. First, we compute…

Machine Learning · Statistics 2025-08-14 Xingyu Li , Qing Liu , Tony Jiang , Hong Amy Xia , Brian P. Hobbs , Peng Wei

The Average Treatment Effect (ATE) is a global measure of the effectiveness of an experimental treatment intervention. Classical methods of its estimation either ignore relevant covariates or do not fully exploit them. Moreover, past work…

Methodology · Statistics 2013-11-05 Emil Pitkin , Richard Berk , Lawrence Brown , Andreas Buja , Ed George , Kai Zhang , Linda Zhao

Estimating conditional average treatment effects (CATEs) from observational data is relevant in many fields such as personalized medicine. However, in practice, the treatment assignment is usually confounded by unobserved variables and thus…

Methodology · Statistics 2023-01-24 Dennis Frauen , Stefan Feuerriegel

Estimating treatment effects is of great importance for many biomedical applications with observational data. Particularly, interpretability of the treatment effects is preferable for many biomedical researchers. In this paper, we first…

Machine Learning · Statistics 2022-06-28 Kan Chen , Qishuo Yin , Qi Long

We consider the problem of estimating the effects of a binary treatment on a continuous outcome of interest from observational data in the absence of confounding by unmeasured factors. We provide a new estimator of the population average…

Methodology · Statistics 2020-08-04 James Robins , Mariela Sued , Quanhong Lei-Gomez , Andrea Rotnitzky

We develop a Bayesian "sum-of-trees" model where each tree is constrained by a regularization prior to be a weak learner, and fitting and inference are accomplished via an iterative Bayesian backfitting MCMC algorithm that generates samples…

Methodology · Statistics 2010-10-08 Hugh A. Chipman , Edward I. George , Robert E. McCulloch

One of the major challenges in estimating conditional potential outcomes and conditional average treatment effects (CATE) is the presence of hidden confounders. Since testing for hidden confounders cannot be accomplished only with…

Machine Learning · Computer Science 2025-06-17 Ahmed Aloui , Juncheng Dong , Ali Hasan , Vahid Tarokh

Estimation of heterogeneous long-term treatment effects (HLTEs) is widely used for personalized decision-making in marketing, economics, and medicine, where short-term randomized experiments are often combined with long-term observational…

Machine Learning · Computer Science 2026-04-02 Haorui Ma , Dennis Frauen , Valentyn Melnychuk , Stefan Feuerriegel

Treatment effect estimation involves assessing the impact of different treatments on individual outcomes. Current methods estimate Conditional Average Treatment Effect (CATE) using observational datasets where covariates are collected…

Machine Learning · Computer Science 2025-02-10 Lokesh Nagalapatti , Pranava Singhal , Avishek Ghosh , Sunita Sarawagi

We study the problem of learning conditional average treatment effects (CATE) from high-dimensional, observational data with unobserved confounders. Unobserved confounders introduce ignorance -- a level of unidentifiability -- about an…

Machine Learning · Computer Science 2022-02-02 Andrew Jesson , Sören Mindermann , Yarin Gal , Uri Shalit

We propose a framework that aligns Conditional Average Treatment Effect (CATE) estimation with profit maximization. Our method recognizes that, for customers with extreme treatment effects, additional estimation accuracy is unlikely to…

Econometrics · Economics 2026-04-21 Artem Timoshenko , Caio Waisman

Randomization tests and flexible treatment-effect models offer complementary strengths for analyzing data from randomized panel experiments: the former provide valid inference under the known assignment mechanism, while the latter can…

Methodology · Statistics 2026-05-12 Fangnan Zheng , Yao Zhang

We propose a novel "tree-averaging" model that utilizes the ensemble of classification and regression trees (CART). Each constituent tree is estimated with a subset of similar data. We treat this grouping of subsets as Bayesian ensemble…

Machine Learning · Statistics 2014-08-20 Leo L. Duan , John P. Clancy , Rhonda D. Szczesniak

Recently, from the personalized medicine perspective, there has been an increased demand to identify subgroups of subjects for whom treatment is effective. Consequently, the estimation of heterogeneous treatment effects (HTE) has been…

Methodology · Statistics 2024-08-02 Ryoma Hieda , Shintaro Yuki , Kensuke Tanioka , Hiroshi Yadohisa

In this study, we compared two groups, in which subjects were assigned to either the treatment or the control group. In such trials, if the efficacy of the treatment cannot be demonstrated in a population that meets the eligibility…

Methodology · Statistics 2024-07-26 Shintaro Yuki , Kensuke Tanioka , Hiroshi Yadohisa