English
Related papers

Related papers: An Algorithm for Identifying Interpretable Subgrou…

200 papers

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

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

For observational studies, we study the sensitivity of causal inference when treatment assignments may depend on unobserved confounders. We develop a loss minimization approach for estimating bounds on the conditional average treatment…

Methodology · Statistics 2022-03-11 Steve Yadlowsky , Hongseok Namkoong , Sanjay Basu , John Duchi , Lu Tian

The Average Treatment Effect (ATE) is a foundational metric in causal inference, widely used to assess intervention efficacy in randomized controlled trials (RCTs). However, in many applications -- particularly in healthcare -- this static…

Machine Learning · Computer Science 2025-07-23 Julianna Piskorz , Krzysztof Kacprzyk , Harry Amad , Mihaela van der Schaar

Within the field of causal inference, we consider the problem of estimating heterogeneous treatment effects from data. We propose and validate a novel approach for learning feature representations to aid the estimation of the conditional…

Machine Learning · Statistics 2022-06-23 Michael C. Burkhart , Gabriel Ruiz

Conditional effect estimation has great scientific and policy importance because interventions may impact subjects differently depending on their characteristics. Most research has focused on estimating the conditional average treatment…

Methodology · Statistics 2023-04-25 Alec McClean , Zach Branson , Edward H. Kennedy

Estimating the conditional average treatment effect (CATE) from observational data is relevant for many applications such as personalized medicine. Here, we focus on the widespread setting where the observational data come from multiple…

Machine Learning · Computer Science 2024-06-05 Jonas Schweisthal , Dennis Frauen , Mihaela van der Schaar , Stefan Feuerriegel

The dearth of prescribing guidelines for physicians is one key driver of the current opioid epidemic in the United States. In this work, we analyze medical and pharmaceutical claims data to draw insights on characteristics of patients who…

Machine Learning · Computer Science 2020-05-01 Chirag Nagpal , Dennis Wei , Bhanukiran Vinzamuri , Monica Shekhar , Sara E. Berger , Subhro Das , Kush R. Varshney

We study the problem of learning conditional average treatment effects (CATE) from observational data with unobserved confounders. The CATE function maps baseline covariates to individual causal effect predictions and is key for…

Machine Learning · Statistics 2018-10-09 Nathan Kallus , Xiaojie Mao , Angela Zhou

The average treatment effect can obscure important heterogeneity when individuals respond differently to a treatment. While the conditional average treatment effect (CATE) function captures such heterogeneity, it is difficult to communicate…

Methodology · Statistics 2026-05-18 Anders Munch , Thomas A. Gerds

Identifying patient subgroups with different treatment responses is an important task to inform medical recommendations, guidelines, and the design of future clinical trials. Existing approaches for treatment effect estimation primarily…

Methodology · Statistics 2025-12-10 Vincent Jeanselme , Chang Ho Yoon , Fabian Falck , Brian Tom , Jessica Barrett

Precise estimation of treatment effects is crucial for evaluating intervention effectiveness. While deep learning models have exhibited promising performance in learning counterfactual representations for treatment effect estimation (TEE),…

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

One of the most significant challenges in Conditional Average Treatment Effect (CATE) estimation is the statistical discrepancy between distinct treatment groups. To address this issue, we propose a model-agnostic data augmentation method…

Machine Learning · Computer Science 2025-06-17 Ahmed Aloui , Juncheng Dong , Cat P. Le , Vahid Tarokh

Most of the widely used estimators of the average treatment effect (ATE) in causal inference rely on the assumptions of unconfoundedness and overlap. Unconfoundedness requires that the observed covariates account for all correlations…

Statistics Theory · Mathematics 2025-07-01 Yang Cai , Alkis Kalavasis , Katerina Mamali , Anay Mehrotra , Manolis Zampetakis

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

Conditional Average Treatment Effects (CATE) estimation is one of the main challenges in causal inference with observational data. In addition to Machine Learning based-models, nonparametric estimators called meta-learners have been…

Machine Learning · Statistics 2023-06-06 Naoufal Acharki , Ramiro Lugo , Antoine Bertoncello , Josselin Garnier

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

We propose Causal Interaction Trees for identifying subgroups of participants that have enhanced treatment effects using observational data. We extend the Classification and Regression Tree algorithm by using splitting criteria that focus…

Methodology · Statistics 2021-12-08 Jiabei Yang , Issa J. Dahabreh , Jon A. Steingrimsson

Robust estimation of heterogeneous treatment effects is a fundamental challenge for optimal decision-making in domains ranging from personalized medicine to educational policy. In recent years, predictive machine learning has emerged as a…

Machine Learning · Statistics 2025-06-23 Maximilian Schuessler , Erik Sverdrup , Robert Tibshirani

In many practical situations, randomly assigning treatments to subjects is uncommon due to feasibility constraints. For example, economic aid programs and merit-based scholarships are often restricted to those meeting specific income or…