English
Related papers

Related papers: Flexible machine learning estimation of conditiona…

200 papers

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

In an era where diverse and complex data are increasingly accessible, the precise prediction of individual treatment effects (ITE) becomes crucial across fields such as healthcare, economics, and public policy. Current state-of-the-art…

Machine Learning · Statistics 2025-01-28 Baozhen Wang , Xingye Qiao

Researchers are increasingly turning to machine learning (ML) algorithms to investigate causal heterogeneity in randomized experiments. Despite their promise, ML algorithms may fail to accurately ascertain heterogeneous treatment effects…

Methodology · Statistics 2024-04-23 Kosuke Imai , Michael Lingzhi Li

Estimating individualised treatment effect (ITE) -- that is the causal effect of a set of variables (also called exposures, treatments, actions, policies, or interventions), referred to as \textit{composite treatments}, on a set of outcome…

Machine Learning · Computer Science 2025-12-19 Vinod Kumar Chauhan , Lei Clifton , Gaurav Nigam , David A. Clifton

In this paper, we develop inference methods for the distribution of heterogeneous individual treatment effects (ITEs) in the nonseparable triangular model with a binary endogenous treatment and a binary instrument of Vuong and Xu (2017) and…

Econometrics · Economics 2025-09-22 Jun Ma , Vadim Marmer , Zhengfei Yu

Recently, conditional average treatment effect (CATE) estimation has been attracting much attention due to its importance in various fields such as statistics, social and biomedical sciences. This study proposes a partially linear…

Methodology · Statistics 2022-01-31 Shunsuke Horii

While average treatment effects (ATE) and conditional average treatment effects (CATE) provide valuable population- and subgroup-level summaries, they fail to capture uncertainty at the individual level. For high-stakes decision-making,…

Methodology · Statistics 2026-03-31 Juraj Bodik , Yaxuan Huang , Bin Yu

Causal effects may vary among individuals and can even be of opposite signs. When significant effect heterogeneity exists, the population average causal effect might be uninformative for an individual. Due to the fundamental problem of…

Methodology · Statistics 2022-12-12 Richard Post , Zhuozhao Zhan , Edwin van den Heuvel

Instrumental variables (IVs) are widely used for estimating causal effects in the presence of unmeasured confounding. Under the standard IV model, however, the average treatment effect (ATE) is only partially identifiable. To address this,…

Methodology · Statistics 2018-01-08 Linbo Wang , Eric Tchetgen Tchetgen

The average treatment effect (ATE) is popularly used to assess the treatment effect. However, the ATE implicitly assumes a homogenous treatment effect even amongst individuals with different characteristics. In this paper, we mainly focus…

Methodology · Statistics 2016-03-10 Yunjian Yin , Lan Liu , Zhi Geng

The growing demand for personalized decision-making has led to a surge of interest in estimating the Conditional Average Treatment Effect (CATE). Various types of CATE estimators have been developed with advancements in machine learning and…

Machine Learning · Computer Science 2024-11-04 Yiyan Huang , Cheuk Hang Leung , Siyi Wang , Yijun Li , Qi Wu

We present unexpected findings from a large-scale benchmark study evaluating Conditional Average Treatment Effect (CATE) estimation algorithms, i.e., CATE models. By running 16 modern CATE models on 12 datasets and 43,200 sampled variants…

Machine Learning · Statistics 2025-02-21 Haining Yu , Yizhou Sun

Randomized trials are typically designed to detect average treatment effects but often lack the statistical power to uncover individual-level treatment effect heterogeneity, limiting their value for personalized decision-making. To address…

Machine Learning · Statistics 2026-03-19 Rickard Karlsson , Piersilvio De Bartolomeis , Issa J. Dahabreh , Jesse H. Krijthe

In this paper, we introduce a unified estimator to analyze various treatment effects in causal inference, including but not limited to the average treatment effect (ATE) and the quantile treatment effect (QTE). The proposed estimator is…

Methodology · Statistics 2025-03-31 Kuan-Hsun Wu , Li-Pang Chen

While machine learning (ML) methods have received a lot of attention in recent years, these methods are primarily for prediction. Empirical researchers conducting policy evaluations are, on the other hand, pre-occupied with causal problems,…

Machine Learning · Statistics 2019-03-04 Noemi Kreif , Karla DiazOrdaz

The estimation of conditional average treatment effects (CATEs) is an important topic in many scientific fields. CATEs can be estimated with high accuracy if data distributed across multiple parties are centralized. However, it is difficult…

Methodology · Statistics 2025-07-28 Yuji Kawamata , Ryoki Motai , Yukihiko Okada , Akira Imakura , Tetsuya Sakurai

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 Treatment Effect (ATE) estimation is a well-studied problem in causal inference. However, it does not necessarily capture the heterogeneity in the data, and several approaches have been proposed to tackle the issue, including…

Machine Learning · Computer Science 2024-03-19 Raghavendra Addanki , Siddharth Bhandari

Randomization tests and flexible treatment-effect models offer complementary strengths for analyzing data from randomized panel experiments: the former provide valid inference under the known assignment mechanism, while the latter can…

Methodology · Statistics 2026-05-12 Fangnan Zheng , Yao Zhang

This paper develops a performant Bayesian approach to conditional average treatment effect (CATE) estimation in regression discontinuity designs (RDD), an increasingly prevalent form of quasi-experiment that facilitates causal inference.…

Methodology · Statistics 2026-05-18 Rafael Alcantara , P. Richard Hahn , Hedibert F. Lopes