English
Related papers

Related papers: 2D score based estimation of heterogeneous treatme…

200 papers

Recently, many causal estimators for Conditional Average Treatment Effect (CATE) and instrumental variable (IV) problems have been published and open sourced, allowing to estimate granular impact of both randomized treatments (such as A/B…

Machine Learning · Computer Science 2022-12-21 Egor Kraev , Timo Flesch , Hudson Taylor Lekunze , Mark Harley , Pere Planell Morell

In multi-site randomized trials with many sites and few randomization units per site, an Empirical-Bayes estimator can be used to estimate the variance of the treatment effect across sites. When this estimator indicates that treatment…

Econometrics · Economics 2024-12-12 Clément de Chaisemartin , Antoine Deeb

We address the estimation of conditional average treatment effects (CATEs) for structured treatments (e.g., graphs, images, texts). Given a weak condition on the effect, we propose the generalized Robinson decomposition, which (i) isolates…

Machine Learning · Computer Science 2021-10-29 Jean Kaddour , Yuchen Zhu , Qi Liu , Matt J. Kusner , Ricardo Silva

This article walks through how to estimate conditional average treatment effects (CATEs) with right-censored time-to-event outcomes using the function causal_survival_forest (Cui et al., 2023) in the R package grf (Athey et al., 2019,…

Computation · Statistics 2024-02-27 Erik Sverdrup , Stefan Wager

Conditional effects are commonly used measures for understanding how treatment effects vary across different groups, and are often used to target treatments/interventions to groups who benefit most. In this work we review existing methods…

Machine Learning · Statistics 2026-04-14 Jiacheng Ge , Iván Díaz

This paper introduces a novel approach for estimating heterogeneous treatment effects of binary treatment in panel data, particularly focusing on short panel data with large cross-sectional data and observed confoundings. In contrast to…

Methodology · Statistics 2024-06-05 Meijia Wang , Ignacio Martinez , P. Richard Hahn

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

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

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

We study Federated Causal Inference, an approach to estimate treatment effects from decentralized data across centers. We compare three classes of Average Treatment Effect (ATE) estimators derived from the Plug-in G-Formula, ranging from…

Machine Learning · Statistics 2025-03-26 Rémi Khellaf , Aurélien Bellet , Julie Josse

This paper presents methods to study the causal effect of a binary treatment on a functional outcome with observational data. We define a Functional Average Treatment Effect and develop an outcome regression estimator. We show how to obtain…

Methodology · Statistics 2025-09-08 Kreske Ecker , Xavier de Luna , Lina Schelin

The aim of clinical effectiveness research using repositories of electronic health records is to identify what health interventions 'work best' in real-world settings. Since there are several reasons why the net benefit of intervention may…

Methodology · Statistics 2020-06-19 Jie Zhu , Blanca Gallego

The weighted average treatment effect (WATE) is a causal measure for the comparison of interventions in a specific target population, which may be different from the population where data are sampled from. For instance, when the goal is to…

Methodology · Statistics 2018-04-17 Yebin Tao , Haoda Fu

Causal machine learning holds promise for estimating individual treatment effects from complex data. For successful real-world applications of machine learning methods, it is of paramount importance to obtain reliable insights into which…

Machine Learning · Computer Science 2026-05-22 Joseph Paillard , Angel Reyero Lobo , Vitaliy Kolodyazhniy , Bertrand Thirion , Denis A. Engemann

We introduce a robust framework for heterogeneous treatment effect (HTE) estimation tailored to high-dimensional low sample size (HDLSS) settings. By combining Graph Attention Networks (GAT) to capture structural dependencies among…

Methodology · Statistics 2025-09-16 Byeonghee Lee , Joonsung Kang

Many social and environmental phenomena are associated with macroscopic changes in the built environment, captured by satellite imagery on a global scale and with daily temporal resolution. While widely used for prediction, these images and…

Methodology · Statistics 2024-07-25 Connor T. Jerzak , Ritwik Vashistha , Adel Daoud

There is an increasing interest in estimating heterogeneity in causal effects in randomized and observational studies. However, little research has been conducted to understand heterogeneity in an instrumental variables study. In this work,…

Methodology · Statistics 2021-01-20 Michael Johnson , Jiongyi Cao , Hyunseung Kang

There are many measures to report so-called treatment or causal effects: absolute difference, ratio, odds ratio, number needed to treat, and so on. The choice of a measure, e.g. absolute versus relative, is often debated because it leads to…

Methodology · Statistics 2025-09-23 Bénédicte Colnet , Julie Josse , Gaël Varoquaux , Erwan Scornet

Estimating heterogeneous treatment effects in domains such as healthcare or social science often involves sensitive data where protecting privacy is important. We introduce a general meta-algorithm for estimating conditional average…

Machine Learning · Statistics 2022-02-23 Fengshi Niu , Harsha Nori , Brian Quistorff , Rich Caruana , Donald Ngwe , Aadharsh Kannan

A central goal of causal inference is to detect and estimate the treatment effects of a given treatment or intervention on an outcome variable of interest, where a member known as the heterogeneous treatment effect (HTE) is of growing…

Statistics Theory · Mathematics 2020-10-27 Zijun Gao , Yanjun Han