English
Related papers

Related papers: Distilling heterogeneous treatment effects: Stable…

200 papers

Conditional density estimation (CDE) goes beyond regression by modeling the full conditional distribution, providing a richer understanding of the data than just the conditional mean in regression. This makes CDE particularly useful in…

Machine Learning · Computer Science 2024-10-16 Lincen Yang , Matthijs van Leeuwen

Surrogate models play a crucial role in retrospectively interpreting complex and powerful black box machine learning models via model distillation. This paper focuses on using model-based trees as surrogate models which partition the…

Machine Learning · Statistics 2023-10-06 Julia Herbinger , Susanne Dandl , Fiona K. Ewald , Sofia Loibl , Giuseppe Casalicchio

Recent work has made important contributions in the development of causally-interpretable meta-analysis. These methods transport treatment effects estimated in a collection of randomized trials to a target population of interest. Ideally,…

Methodology · Statistics 2023-02-08 Justin M. Clark , Kollin W. Rott , James S. Hodges , Jared D. Huling

We address the problem of estimating heterogeneous treatment effects in panel data, adopting the popular Difference-in-Differences (DiD) framework under the conditional parallel trends assumption. We propose a novel doubly robust…

Machine Learning · Statistics 2025-04-29 Hui Lan , Haoge Chang , Eleanor Dillon , Vasilis Syrgkanis

We introduce an algorithm for identifying interpretable subgroups with elevated treatment effects, given an estimate of individual or conditional average treatment effects (CATE). Subgroups are characterized by ``rule sets'' --…

Machine Learning · Statistics 2025-07-15 Albert Chiu

We address a core problem in causal inference: estimating heterogeneous treatment effects using panel data with general treatment patterns. Many existing methods either do not utilize the potential underlying structure in panel data or have…

Machine Learning · Statistics 2024-06-11 Retsef Levi , Elisabeth Paulson , Georgia Perakis , Emily Zhang

The paper proposes an estimator to make inference of heterogeneous treatment effects sorted by impact groups (GATES) for non-randomised experiments. The groups can be understood as a broader aggregation of the conditional average treatment…

Econometrics · Economics 2020-03-30 Daniel Jacob

The bulk of causal inference studies rule out the presence of interference between units. However, in many real-world scenarios, units are interconnected by social, physical, or virtual ties, and the effect of the treatment can spill from…

Methodology · Statistics 2023-11-03 Falco J. Bargagli-Stoffi , Costanza Tortù , Laura Forastiere

In causal inference, estimating heterogeneous treatment effects (HTE) is critical for identifying how different subgroups respond to interventions, with broad applications in fields such as precision medicine and personalized advertising.…

Machine Learning · Computer Science 2024-07-02 Jiehui Zhou , Linxiao Yang , Xingyu Liu , Xinyue Gu , Liang Sun , Wei Chen

Deep ensembles (DE) have emerged as a powerful approach for quantifying predictive uncertainty and distinguishing its aleatoric and epistemic components, thereby enhancing model robustness and reliability. However, their high computational…

Machine Learning · Computer Science 2025-11-19 Kaizheng Wang , Fabio Cuzzolin , David Moens , Hans Hallez

The Sustainable Development Goals (SDGs) of the United Nations provide a blueprint of a better future by 'leaving no one behind', and, to achieve the SDGs by 2030, poor countries require immense volumes of development aid. In this paper, we…

Machine Learning · Statistics 2024-06-18 Milan Kuzmanovic , Dennis Frauen , Tobias Hatt , Stefan Feuerriegel

While deep reinforcement learning has achieved promising results in challenging decision-making tasks, the main bones of its success --- deep neural networks are mostly black-boxes. A feasible way to gain insight into a black-box model is…

Machine Learning · Computer Science 2021-08-17 Zhao-Hua Li , Yang Yu , Yingfeng Chen , Ke Chen , Zhipeng Hu , Changjie Fan

Cluster-randomized trials (CRTs) are widely used to evaluate group-level interventions and increasingly collect multiple outcomes capturing complementary dimensions of benefit and risk. Investigators often seek a single global summary of…

Methodology · Statistics 2026-01-22 Xinyuan Chen , Fan Li

Subgroup analysis of treatment effects plays an important role in applications from medicine to public policy to recommender systems. It allows physicians (for example) to identify groups of patients for whom a given drug or treatment is…

Machine Learning · Statistics 2020-10-20 Hyun-Suk Lee , Yao Zhang , William Zame , Cong Shen , Jang-Won Lee , Mihaela van der Schaar

We develop a Causal-Deep Neural Network (CDNN) model trained in two stages to infer causal impact estimates at an individual unit level. Using only the pre-treatment features in stage 1 in the absence of any treatment information, we learn…

Machine Learning · Computer Science 2022-01-24 Naveen Nair , Karthik S. Gurumoorthy , Dinesh Mandalapu

When evaluating the efficacy of social programs and medical treatments using randomized experiments, the estimated overall average causal effect alone is often of limited value and the researchers must investigate when the treatments do and…

Applications · Statistics 2013-05-27 Kosuke Imai , Marc Ratkovic

Estimating how a treatment affects units individually, known as heterogeneous treatment effect (HTE) estimation, is an essential part of decision-making and policy implementation. The accumulation of large amounts of data in many domains,…

Machine Learning · Computer Science 2022-06-28 Christopher Tran , Elena Zheleva

Individuals do not respond uniformly to treatments, events, or interventions. Sociologists routinely partition samples into subgroups to explore how the effects of treatments vary by covariates like race, gender, and socioeconomic status.…

Other Statistics · Statistics 2019-09-23 Jennie E. Brand , Jiahui Xu , Bernard Koch , Pablo Geraldo

Average and conditional treatment effects are fundamental causal quantities used to evaluate the effectiveness of treatments in various critical applications, including clinical settings and policy-making. Beyond the gold-standard…

With recent achievements in tasks requiring context awareness, foundation models have been adopted to treat large-scale data from electronic health record (EHR) systems. However, previous clinical recommender systems based on foundation…

Artificial Intelligence · Computer Science 2023-02-02 Seunghyun Lee , Da Young Lee , Sujeong Im , Nan Hee Kim , Sung-Min Park