English
Related papers

Related papers: Distilling heterogeneous treatment effects: Stable…

200 papers

Interpretability and effectiveness are two essential and indispensable requirements for adopting machine learning methods in reality. In this paper, we propose a knowledge distillation based decision trees extension, dubbed rectified…

Machine Learning · Computer Science 2020-08-24 Yiming Li , Jiawang Bai , Jiawei Li , Xue Yang , Yong Jiang , Shu-Tao Xia

The conditional average treatment effect (CATE) is the best measure of individual causal effects given baseline covariates. However, the CATE only captures the (conditional) average, and can overlook risks and tail events, which are…

Machine Learning · Statistics 2025-06-05 Nathan Kallus , Miruna Oprescu

In this paper we study the problems of estimating heterogeneity in causal effects in experimental or observational studies and conducting inference about the magnitude of the differences in treatment effects across subsets of the…

Machine Learning · Statistics 2022-06-08 Susan Athey , Guido Imbens

Cluster-randomized trials (CRTs) are widely used to evaluate interventions delivered at the clinic, practice, or community level. Although standard analyses typically target average treatment effects, such summaries mask potentially…

Methodology · Statistics 2026-04-16 Changjun Li , Xi Fang , Michael O. Harhay , Andrew B. Forbes , F. Perry Wilson , Guangyu Tong , Fan Li

Estimating heterogeneous treatment effects in survival settings is complicated by right censoring as well as the time-varying nature of the estimand. While the conditional average treatment effect (CATE) provides a natural target, most…

Methodology · Statistics 2026-04-14 Yuming Sun , Jian Kang , Yi Li

Black-box heterogeneous treatment effect (HTE) models are increasingly being used to create personalized policies that assign individuals to their optimal treatments. However, they are difficult to understand, and can be burdensome to…

Machine Learning · Computer Science 2022-08-09 Han Wu , Sarah Tan , Weiwei Li , Mia Garrard , Adam Obeng , Drew Dimmery , Shaun Singh , Hanson Wang , Daniel Jiang , Eytan Bakshy

Machine learning has proved to be very successful for making predictions in travel behavior modeling. However, most machine-learning models have complex model structures and offer little or no explanation as to how they arrive at these…

Machine Learning · Statistics 2019-10-31 Xilei Zhao , Zhengze Zhou , Xiang Yan , Pascal Van Hentenryck

Heterogeneous treatment effects (HTEs) are commonly identified during randomized controlled trials (RCTs). Identifying subgroups of patients with similar treatment effects is of high interest in clinical research to advance precision…

Machine Learning · Computer Science 2022-12-06 Peniel N. Argaw , Elizabeth Healey , Isaac S. Kohane

Matching in causal inference from observational data aims to construct treatment and control groups with similar distributions of covariates, thereby reducing confounding and ensuring an unbiased estimation of treatment effects. This…

Artificial Intelligence · Computer Science 2025-04-15 Sahil Shikalgar , Md. Noor-E-Alam

Designing faithful yet accurate AI models is challenging, particularly in the field of individual treatment effect estimation (ITE). ITE prediction models deployed in critical settings such as healthcare should ideally be (i) accurate, and…

Machine Learning · Computer Science 2024-06-04 Yinjun Wu , Mayank Keoliya , Kan Chen , Neelay Velingker , Ziyang Li , Emily J Getzen , Qi Long , Mayur Naik , Ravi B Parikh , Eric Wong

Treatment effect heterogeneity occurs when individual characteristics influence the effect of a treatment. We propose a novel approach that combines prognostic score matching and conditional inference trees to characterize effect…

This paper provides a link between causal inference and machine learning techniques - specifically, Classification and Regression Trees (CART) - in observational studies where the receipt of the treatment is not randomized, but the…

Machine Learning · Computer Science 2019-05-23 Falco J. Bargagli-Stoffi , Giorgio Gnecco

Estimating heterogeneous treatment effects has become increasingly important in many fields and life and death decisions are now based on these estimates: for example, selecting a personalized course of medical treatment. Recently, a…

Methodology · Statistics 2019-04-01 Sören R. Künzel , Simon J. S. Walter , Jasjeet S. Sekhon

Estimating how a treatment affects different individuals, known as heterogeneous treatment effect estimation, is an important problem in empirical sciences. In the last few years, there has been a considerable interest in adapting machine…

Machine Learning · Computer Science 2024-10-18 Christopher Tran , Keith Burghardt , Kristina Lerman , Elena Zheleva

Inferring the heterogeneous treatment effect is a fundamental problem in the sciences and commercial applications. In this paper, we focus on estimating Conditional Average Treatment Effect (CATE), that is, the difference in the conditional…

Methodology · Statistics 2021-03-23 Haomiao Meng , Xingye Qiao

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

Difference-in-differences (DID) approaches are widely used for estimating causal effects with observational data before and after an intervention. DID traditionally estimates the average treatment effect among the treated after making a…

Methodology · Statistics 2025-06-24 Julia C. Thome , Andrew J. Spieker , Peter F. Rebeiro , Chun Li , Tong Li , Bryan E. Shepherd

Deep learning models have demonstrated promising results in estimating treatment effects (TEE). However, most of them overlook the variations in treatment outcomes among subgroups with distinct characteristics. This limitation hinders their…

Machine Learning · Computer Science 2024-01-31 Seungyeon Lee , Ruoqi Liu , Wenyu Song , Ping Zhang

Understanding treatment effect heterogeneity is vital for scientific and policy research. However, identifying and evaluating heterogeneous treatment effects pose significant challenges due to the typically unknown subgroup structure.…

Methodology · Statistics 2024-11-05 Kwangho Kim , Jisu Kim , Larry A. Wasserman , Edward H. Kennedy

Tailoring treatments to individual needs is a central goal in fields such as medicine. A key step toward this goal is estimating Heterogeneous Treatment Effects (HTE) - the way treatments impact different subgroups. While crucial, HTE…

Machine Learning · Statistics 2025-07-30 Tomer Meir , Uri Shalit , Malka Gorfine