English
Related papers

Related papers: Distilling heterogeneous treatment effects: Stable…

200 papers

Causal inference and the estimation of causal effects plays a central role in decision-making across many areas, including healthcare and economics. Estimating causal effects typically requires an estimator that is tailored to each problem…

Treatment effects of stochastic policy shifts quantify differences in outcomes across counterfactual scenarios with varying treatment distributions. Stochastic policy shifts may be of interest in settings where it is unrealistic or…

Methodology · Statistics 2026-03-31 Michael Jetsupphasuk , Chenwei Fang , Didong Li , Michael G. Hudgens

In this extended abstract paper, we address the problem of interpretability and targeted regularization in causal machine learning models. In particular, we focus on the problem of estimating individual causal/treatment effects under…

Machine Learning · Computer Science 2022-07-13 Alberto Caron , Gianluca Baio , Ioanna Manolopoulou

Randomized controlled trials (RCTs) are the accepted standard for treatment effect estimation but they can be infeasible due to ethical reasons and prohibitive costs. Single-arm trials, where all patients belong to the treatment group, can…

Randomized trials and observational studies, more often than not, run over a certain period of time. The treatment effect evolves during this period which provides crucial insights into the treatment response and the long-term effects. Many…

Methodology · Statistics 2020-04-13 Shu Li , Peter Bühlmann

Knowledge Distillation (KD) aims to transfer knowledge from a large teacher model to a smaller student model. While contrastive learning has shown promise in self-supervised learning by creating discriminative representations, its…

Computer Vision and Pattern Recognition · Computer Science 2025-05-14 Nikolaos Giakoumoglou , Tania Stathaki

Applied Difference-in-Differences studies often involve outcomes that are discrete, mixed, censored, or otherwise non-continuously distributed, while policy questions frequently concern distributional effects rather than mean effects alone.…

Econometrics · Economics 2026-05-22 Nelly K. Djuazon , Emmanuel Selorm Tsyawo

Randomized controlled trials (RCTs) are often underpowered to detect treatment heterogeneity in subgroups defined by cross-classifications of multiple covariates, due to sparse sample sizes in some strata. External RCT data can help, but…

Methodology · Statistics 2026-04-23 Youqi Yang , Walter Dempsey , Bhramar Mukherjee

Heterogeneous treatment effects (HTEs) are increasingly estimated using machine learning models that produce highly personalized predictions of treatment effects. In practice, however, predicted treatment effects are rarely interpreted,…

We propose a new estimation method for heterogeneous causal effects which utilizes a regression discontinuity (RD) design for multiple datasets with different thresholds. The standard RD design is frequently used in applied researches, but…

Econometrics · Economics 2019-05-14 Takayuki Toda , Ayako Wakano , Takahiro Hoshino

Difference-in-differences (DID) is a popular approach to identify the causal effects of treatments and policies in the presence of unmeasured confounding. DID identifies the sample average treatment effect in the treated (SATT). However, a…

Methodology · Statistics 2024-06-21 Audrey Renson , Ellicott C. Matthay , Kara E. Rudolph

Accurately estimating personalized treatment effects within a study site (e.g., a hospital) has been challenging due to limited sample size. Furthermore, privacy considerations and lack of resources prevent a site from leveraging…

Machine Learning · Statistics 2022-06-17 Xiaoqing Tan , Chung-Chou H. Chang , Ling Zhou , Lu Tang

Regression discontinuity design (RDD) is widely adopted for causal inference under intervention determined by a continuous variable. While one is interested in treatment effect heterogeneity by subgroups in many applications, RDD typically…

Methodology · Statistics 2024-11-11 Shonosuke Sugasawa , Takuya Ishihara , Daisuke Kurisu

Personalized medicine seeks to identify the causal effect of treatment for a particular patient as opposed to a clinical population at large. Most investigators estimate such personalized treatment effects by regressing the outcome of a…

Machine Learning · Statistics 2021-09-02 Eric V. Strobl , Shyam Visweswaran

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

The conditional average treatment effect (CATE) is frequently estimated to refute the homogeneous treatment effect assumption. Under this assumption, all units making up the population under study experience identical benefit from a given…

In health and social sciences, it is critically important to identify subgroups of the study population where there is notable heterogeneity of treatment effects (HTE) with respect to the population average. Decision trees have been…

Methodology · Statistics 2024-05-28 Falco J. Bargagli-Stoffi , Riccardo Cadei , Kwonsang Lee , Francesca Dominici

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

Interpretability plays a crucial role in the application of statistical learning to estimate heterogeneous treatment effects (HTE) in complex diseases. In this study, we leverage a rule-based workflow, namely causal rule learning (CRL), to…

Machine Learning · Computer Science 2025-11-24 Ying Wu , Hanzhong Liu , Kai Ren , Shujie Ma , Xiangyu Chang

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