English
Related papers

Related papers: SurvITE: Learning Heterogeneous Treatment Effects …

200 papers

Given a dataset of individuals each described by a covariate vector, a treatment, and an observed outcome on the treatment, the goal of the individual treatment effect (ITE) estimation task is to predict outcome changes resulting from a…

Machine Learning · Computer Science 2024-06-07 Lokesh Nagalapatti , Pranava Singhal , Avishek Ghosh , Sunita Sarawagi

Causal inference methods are widely applied in the fields of medicine, policy, and economics. Central to these applications is the estimation of treatment effects to make decisions. Current methods make binary yes-or-no decisions based on…

Machine Learning · Computer Science 2020-04-24 Will Y. Zou , Smitha Shyam , Michael Mui , Mingshi Wang , Jan Pedersen , Zoubin Ghahramani

We propose a novel method for predicting time-to-event in the presence of cure fractions based on flexible survivals models integrated into a deep neural network framework. Our approach allows for non-linear relationships and…

Machine Learning · Statistics 2024-11-11 Victor Medina-Olivares , Stefan Lessmann , Nadja Klein

In biomedical science, analyzing treatment effect heterogeneity plays an essential role in assisting personalized medicine. The main goals of analyzing treatment effect heterogeneity include estimating treatment effects in clinically…

Methodology · Statistics 2022-12-06 Waverly Wei , Maya Petersen , Mark J van der Laan , Zeyu Zheng , Chong Wu , Jingshen Wang

The task of predicting long-term patient outcomes using supervised machine learning is a challenging one, in part because of the high variance of each patient's trajectory, which can result in the model over-fitting to the training data.…

Machine Learning · Computer Science 2026-02-09 Thomas Frost , Kezhi Li , Steve Harris

There is growing interest in estimating and analyzing heterogeneous treatment effects in experimental and observational studies. We describe a number of meta-algorithms that can take advantage of any supervised learning or regression method…

Statistics Theory · Mathematics 2019-06-18 Sören R. Künzel , Jasjeet S. Sekhon , Peter J. Bickel , Bin Yu

Estimating the joint effect of a multivariate, continuous exposure is crucial, particularly in environmental health where interest lies in simultaneously evaluating the impact of multiple environmental pollutants on health. We develop novel…

Accurately predicting conditional average treatment effects (CATEs) is crucial in personalized medicine and digital platform analytics. Since the treatments of interest often cannot be directly randomized, observational data is leveraged to…

Methodology · Statistics 2024-11-05 Miruna Oprescu , Nathan Kallus

This study proposes a novel framework based on the RuleFit method to estimate Heterogeneous Treatment Effect (HTE) in a randomized clinical trial. To achieve this, we adopted S-learner of the metaalgorithm for our proposed framework. The…

Methodology · Statistics 2023-07-28 Mayu Hiraishi , Ke Wan , Kensuke Tanioka , Hiroshi Yadohisa , Toshio Shimokawa

Heterogeneous treatment effect (HTE) estimation is critical in medical research. It provides insights into how treatment effects vary among individuals, which can provide statistical evidence for precision medicine. While most existing…

Machine Learning · Statistics 2025-04-25 Ke Wan , Kensuke Tanioka , Toshio Shimokawa

Recent development in data-driven decision science has seen great advances in individualized decision making. Given data with individual covariates, treatment assignments and outcomes, researchers can search for the optimal individualized…

Methodology · Statistics 2021-12-16 Weibin Mo , Yufeng Liu

We propose a framework for testing the homogeneity of conditional average treatment effects (CATEs) across multiple experimental and observational studies. Our approach leverages multiple randomized trials to assess whether treatment…

Econometrics · Economics 2026-02-25 Ana Armendariz , Martin Huber

The accurate prediction of survival times for patients with severe diseases remains a critical challenge despite recent advances in artificial intelligence. This study introduces "SurvTimeSurvival: Survival Analysis On Patients With…

Machine Learning · Computer Science 2023-11-17 Hung Le , Ong Eng-Jon , Bober Miroslaw

Estimating heterogeneous treatment effects (HTEs) is crucial for precision medicine. While multiple studies can improve the generalizability of results, leveraging them for estimation is statistically challenging. Existing approaches often…

Methodology · Statistics 2025-12-22 Cathy Shyr , Boyu Ren , Prasad Patil , Giovanni Parmigiani

We study treatment-effect estimation using panel data. The treatment may be non-binary, non-absorbing, and the outcome may be affected by treatment lags. We make a parallel-trends assumption, and propose event-study estimators of the effect…

Econometrics · Economics 2026-05-13 Clément de Chaisemartin , Xavier D'Haultfœuille

Estimating long-term causal effects by combining long-term observational and short-term experimental data is a crucial but challenging problem in many real-world scenarios. In existing methods, several ideal assumptions, e.g. latent…

Machine Learning · Computer Science 2025-05-12 Ruichu Cai , Junjie Wan , Weilin Chen , Zeqin Yang , Zijian Li , Peng Zhen , Jiecheng Guo

Causal effect estimation seeks to determine the impact of an intervention from observational data. However, the existing causal inference literature primarily addresses treatment effects on frequently occurring events. But what if we are…

Machine Learning · Statistics 2025-06-18 Jiyuan Tan , Jose Blanchet , Vasilis Syrgkanis

Estimating individual-level treatment effect from observational data is a fundamental problem in causal inference and has attracted increasing attention in the fields of education, healthcare, and public policy.In this work, we concentrate…

Machine Learning · Computer Science 2025-07-10 Hui Meng , Keping Yang , Xuyu Peng , Bo Zheng

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

Treatment effect estimation is a fundamental problem in causal inference. We focus on designing efficient randomized controlled trials, to accurately estimate the effect of some treatment on a population of $n$ individuals. In particular,…

Machine Learning · Computer Science 2022-10-14 Raghavendra Addanki , David Arbour , Tung Mai , Cameron Musco , Anup Rao