English
Related papers

Related papers: A Partial Initialization Strategy to Mitigate the …

200 papers

We study the model selection problem in conditional average treatment effect (CATE) prediction. Unlike previous works on this topic, we focus on preserving the rank order of the performance of candidate CATE predictors to enable accurate…

Machine Learning · Statistics 2020-07-20 Yuta Saito , Shota Yasui

Most of the widely used estimators of the average treatment effect (ATE) in causal inference rely on the assumptions of unconfoundedness and overlap. Unconfoundedness requires that the observed covariates account for all correlations…

Statistics Theory · Mathematics 2025-07-01 Yang Cai , Alkis Kalavasis , Katerina Mamali , Anay Mehrotra , Manolis Zampetakis

Augmented inverse probability weighting and G-computation with canonical generalized linear models have become increasingly popular for estimating average treatment effects (ATEs) in randomized experiments. These methods leverage outcome…

Methodology · Statistics 2026-03-13 Muluneh Alene , Stijn Vansteelandt , Kelly Van Lancker

The heterogeneity of treatment effect (HTE) lies at the heart of precision medicine. Randomized controlled trials are gold-standard for treatment effect estimation but are typically underpowered for heterogeneous effects. In contrast, large…

Methodology · Statistics 2024-11-14 Shu Yang , Siyi Liu , Donglin Zeng , Xiaofei Wang

Accurately predicting conditional average treatment effects (CATEs) is crucial in personalized medicine and digital platform analytics. Since the treatments of interest often cannot be directly randomized, observational data is leveraged to…

Methodology · Statistics 2024-11-05 Miruna Oprescu , Nathan Kallus

Treatment effect estimation involves assessing the impact of different treatments on individual outcomes. Current methods estimate Conditional Average Treatment Effect (CATE) using observational datasets where covariates are collected…

Machine Learning · Computer Science 2025-02-10 Lokesh Nagalapatti , Pranava Singhal , Avishek Ghosh , Sunita Sarawagi

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

One of the central goals of causal machine learning is the accurate estimation of heterogeneous treatment effects from observational data. In recent years, meta-learning has emerged as a flexible, model-agnostic paradigm for estimating…

Artificial Intelligence · Computer Science 2024-11-14 Henri Arno , Paloma Rabaey , Thomas Demeester

Adapting to latent confounded shift remains a core challenge in modern AI. This setting is driven by hidden variables that induce spurious correlations between inputs and outputs during training, leading models to rely on non-causal…

Machine Learning · Computer Science 2026-05-14 Jialin Yu , Yuxiang Zhou , Haoxuan Li , Junchi Yu , Mengyue Yang , Yulan He , Nevin L. Zhang , Philip Torr , Ricardo Silva

Conditional Average Treatment Effect (CATE) estimation, at the heart of counterfactual reasoning, is a crucial challenge for causal modeling both theoretically and applicatively, in domains such as healthcare, sociology, or advertising.…

Machine Learning · Computer Science 2025-01-27 Armand Lacombe , Michèle Sebag

Estimating treatment effects plays a crucial role in causal inference, having many real-world applications like policy analysis and decision making. Nevertheless, estimating treatment effects in the longitudinal setting in the presence of…

Machine Learning · Computer Science 2023-02-22 Defu Cao , James Enouen , Yan Liu

When evaluating a two-phase intervention, the cumulative average treatment effect (ATE) is often the primary causal estimand of interest. However, some individuals who do not respond well to the Phase I treatment may subsequently display…

Methodology · Statistics 2026-02-02 Guanglei Hong , Xu Qin , Zhengyan Xu , Fan Yang

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 investigates the use of regularization priors in the context of treatment effect estimation using observational data where the number of control variables is large relative to the number of observations. First, the phenomenon of…

Methodology · Statistics 2016-12-30 P. Richard Hahn , Carlos M. Carvalho , Jingyu He , David Puelz

An important aspect of the performance of algorithms that predict individualized treatment effects (ITE) is moderate calibration, i.e., the average treatment effect among individuals with predicted treatment effect of z being equal to z.…

Methodology · Statistics 2025-12-10 Mohsen Sadatsafavi , Jeroen Hoogland , Thomas P. A. Debray , John Petkau

Average treatment effects (ATE) and conditional average treatment effects (CATE) are foundational causal estimands, but they target changes in expected outcomes and can miss treatment-induced changes in the shape of outcome distributions. A…

Methodology · Statistics 2026-03-17 Amir Saki , Usef Faghihi

Randomized controlled trials (RCTs) are increasingly prevalent in education research, and are often regarded as a gold standard of causal inference. Two main virtues of randomized experiments are that they (1) do not suffer from…

We propose a framework that aligns Conditional Average Treatment Effect (CATE) estimation with profit maximization. Our method recognizes that, for customers with extreme treatment effects, additional estimation accuracy is unlikely to…

Econometrics · Economics 2026-04-21 Artem Timoshenko , Caio Waisman

Individual treatment effect (ITE) represents the expected improvement in the outcome of taking a particular action to a particular target, and plays important roles in decision making in various domains. However, its estimation problem is…

Machine Learning · Computer Science 2020-05-12 Shonosuke Harada , Hisashi Kashima

This paper studies inference in a randomized controlled trial (RCT) with covariate-adaptive randomization (CAR) and imperfect compliance of a binary treatment. In this context, we study inference on the LATE. As in Bugni et al. (2018,2019),…

Econometrics · Economics 2023-07-25 Federico A. Bugni , Mengsi Gao