English
Related papers

Related papers: Efficient Heterogeneous Treatment Effect Estimatio…

200 papers

We are interested in the distribution of treatment effects for an experiment where units are randomized to a treatment but outcomes are measured for pairs of units. For example, we might measure risk sharing links between households…

Econometrics · Economics 2022-10-11 Eric Auerbach , Yong Cai

Partial Label Learning (PLL) aims to learn from the data where each training instance is associated with a set of candidate labels, among which only one is correct. Most existing methods deal with such problem by either treating each…

Machine Learning · Computer Science 2019-06-04 Gengyu Lyu , Songhe Feng , Yi Jin , Guojun Dai , Congyan Lang , Yidong Li

Analysis of effect heterogeneity at the group level is standard practice in empirical treatment evaluation. However, treatments analyzed are often aggregates of multiple underlying treatments which are themselves heterogeneous, e.g.…

Econometrics · Economics 2026-02-27 Phillip Heiler , Michael C. Knaus

When the Stable Unit Treatment Value Assumption is violated and there is interference among units, there is not a uniquely defined Average Treatment Effect, and alternative estimands may be of interest. Among these are average unit-level…

Methodology · Statistics 2025-06-30 Molly Offer-Westort , Drew Dimmery

The Predictive Approaches to Treatment Effect Heterogeneity statement focused on baseline risk as a robust predictor of treatment effect and provided guidance on risk-based assessment of treatment effect heterogeneity in the RCT setting.…

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

The machine learning toolbox for estimation of heterogeneous treatment effects from observational data is expanding rapidly, yet many of its algorithms have been evaluated only on a very limited set of semi-synthetic benchmark datasets. In…

Machine Learning · Computer Science 2021-07-29 Alicia Curth , Mihaela van der Schaar

In many applications, data can be heterogeneous in the sense of spanning latent groups with different underlying distributions. When predictive models are applied to such data the heterogeneity can affect both predictive performance and…

Machine Learning · Statistics 2022-05-04 Thomas Lartigue , Sach Mukherjee

We propose a test-based elastic integrative analysis of the randomized trial and real-world data to estimate treatment effect heterogeneity with a vector of known effect modifiers. When the real-world data are not subject to bias, our…

Methodology · Statistics 2022-11-30 Shu Yang , Chenyin Gao , Donglin Zeng , Xiaofei Wang

We study targeted maximum likelihood estimation (TMLE) of the average treatment effect in a semiparametric regression model whose mean function is indexed by a finite-dimensional parameter, while the additive error distribution is left…

Methodology · Statistics 2026-04-20 Mijeong Kim

Estimating heterogeneous treatment effects is important to tailor treatments to those individuals who would most likely benefit. However, conditional average treatment effect predictors may often be trained on one population but possibly…

Machine Learning · Computer Science 2024-10-18 Christoph Kern , Michael Kim , Angela Zhou

The paper proposes a causal supervised machine learning algorithm to uncover treatment effect heterogeneity in sharp and fuzzy regression discontinuity (RD) designs. We develop a criterion for building an honest ``regression discontinuity…

Econometrics · Economics 2025-09-01 Ágoston Reguly

Recent self-supervised pre-training methods for electroencephalogram (EEG) have shown promising results. However, the pre-trained models typically require full fine-tuning on each downstream task individually to achieve good performance. In…

Machine Learning · Computer Science 2026-04-30 Sicheng Dai , Kai Chen , Hongwang Xiao , Shan Yu , Qiwei Ye

We study low-rank matrix regression in settings where matrix-valued predictors and scalar responses are observed across multiple individuals. Rather than assuming a fully homogeneous coefficient matrices across individuals, we accommodate…

Methodology · Statistics 2025-10-28 Di Wang , Xiaoyu Zhang , Guodong Li , Wenyang Zhang

Conditional average treatment effects (CATEs) allow us to understand the effect heterogeneity across a large population of individuals. However, typical CATE learners assume all confounding variables are measured in order for the CATE to be…

Machine Learning · Computer Science 2022-02-01 Yao Zhang , Jeroen Berrevoets , Mihaela van der Schaar

Heterogeneous treatment effect models allow us to compare treatments at subgroup and individual levels, and are of increasing popularity in applications like personalized medicine, advertising, and education. In this talk, we first survey…

Methodology · Statistics 2022-01-28 Zijun Gao , Trevor Hastie

Treatment effect heterogeneity is of a great concern when evaluating policy impact: "is the treatment Pareto-improving?", "what is the proportion of people who are better off under the treatment?", etc. However, even in the simple case of a…

Econometrics · Economics 2025-09-18 Myungkou Shin

Heterogeneous treatment effect (HTE) estimation from observational data poses significant challenges due to treatment selection bias. Existing methods address this bias by minimizing distribution discrepancies between treatment groups in…

Machine Learning · Computer Science 2026-03-26 Hao Wang , Zhichao Chen , Zhaoran Liu , Xu Chen , Haoxuan Li , Zhouchen Lin

We propose a new framework for online testing of heterogeneous treatment effects. The proposed test, named sequential score test (SST), is able to control type I error under continuous monitoring and detect multi-dimensional heterogeneous…

Methodology · Statistics 2020-02-11 Miao Yu , Wenbin Lu , Rui Song

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