English
Related papers

Related papers: CATE meets ML -- The Conditional Average Treatment…

200 papers

Conventional research on educational effects typically either employs a "years of schooling" measure of education, or dichotomizes attainment as a point-in-time treatment. Yet, such a conceptualization of education is misaligned with the…

Applications · Statistics 2026-02-11 Aleksei Opacic

Estimating how a treatment affects different individuals, known as heterogeneous treatment effect estimation, is an important problem in empirical sciences. In the last few years, there has been a considerable interest in adapting machine…

Machine Learning · Computer Science 2024-10-18 Christopher Tran , Keith Burghardt , Kristina Lerman , Elena Zheleva

We introduce a new preference-based framework for conditional treatment effect estimation and policy learning, built on the Conditional Preference-based Treatment Effect (CPTE). CPTE requires only that outcomes be ranked under a preference…

Machine Learning · Statistics 2026-02-04 Dovid Parnas , Mathieu Even , Julie Josse , Uri Shalit

We propose plug-in (PI) and double machine learning (DML) estimators of average treatment effect (ATE), average treatment effect on the treated (ATET) and local average treatment effect (LATE) in the multivariate sample selection model with…

Econometrics · Economics 2025-11-18 Sofiia Dolgikh , Bodan Potanin

We consider the problem of estimating the average treatment effect (ATE) when both randomized control trial (RCT) data and external real-world data (RWD) are available. We decompose the ATE estimand as the difference between a pooled-ATE…

Methodology · Statistics 2025-01-22 Mark van der Laan , Sky Qiu , Jens Magelund Tarp , Lars van der Laan

Understanding how treatment effects vary across patient characteristics is essential for personalized medicine, yet randomized controlled trials (RCTs) are often underpowered to detect heterogeneous treatment effects (HTEs). We propose a…

Methodology · Statistics 2025-07-22 Amir Asiaee , Chiara Di Gravio , Cole Beck , Yuting Mei , Samhita Pal , Jared D. Huling

We examine the challenges in ranking multiple treatments based on their estimated effects when using linear regression or its popular double-machine-learning variant, the Partially Linear Model (PLM), in the presence of treatment effect…

Econometrics · Economics 2024-11-06 Apoorva Lal

Matching and weighting methods for observational studies involve the choice of an estimand, the causal effect with reference to a specific target population. Commonly used estimands include the average treatment effect in the treated (ATT),…

Methodology · Statistics 2023-07-12 Noah Greifer , Elizabeth A. Stuart

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

This paper provides a link between causal inference and machine learning techniques - specifically, Classification and Regression Trees (CART) - in observational studies where the receipt of the treatment is not randomized, but the…

Machine Learning · Computer Science 2019-05-23 Falco J. Bargagli-Stoffi , Giorgio Gnecco

We investigate the problem of estimating the average treatment effect (ATE) under a very general setup where the covariates can be high-dimensional, highly correlated, and can have sparse nonlinear effects on the propensity and outcome…

Machine Learning · Statistics 2025-08-26 Jianqing Fan , Soham Jana , Sanjeev Kulkarni , Qishuo Yin

We study attribute control in language models through the method of Causal Average Treatment Effect (Causal ATE). Existing methods for the attribute control task in Language Models (LMs) check for the co-occurrence of words in a sentence…

Computation and Language · Computer Science 2024-02-19 Rahul Madhavan , Kahini Wadhawan

This paper introduces a generalized ps-BART model for the estimation of Average Treatment Effect (ATE) and Conditional Average Treatment Effect (CATE) in continuous treatments, addressing limitations of the Bayesian Causal Forest (BCF)…

Machine Learning · Statistics 2024-09-11 Hugo Gobato Souto , Francisco Louzada Neto

Difference-in-differences is based on a parallel trends assumption, which states that changes over time in average potential outcomes are independent of treatment assignment, possibly conditional on covariates. With time-varying treatments,…

Methodology · Statistics 2024-06-25 Nicholas Illenberger , Iván Díaz , Audrey Renson

Estimating treatment effects (TE) from observational data is a critical yet complex task in many fields, from healthcare and economics to public policy. While recent advances in machine learning and causal inference have produced powerful…

Machine Learning · Computer Science 2025-08-15 Jeroen Berrevoets , Julianna Piskorz , Robert Davis , Harry Amad , Jim Weatherall , Mihaela van der Schaar

Causal learning is the key to obtaining stable predictions and answering \textit{what if} problems in decision-makings. In causal learning, it is central to seek methods to estimate the average treatment effect (ATE) from observational…

Machine Learning · Statistics 2022-12-07 Yiyan Huang , Cheuk Hang Leung , Qi Wu , Xing Yan

In this paper, we explore optimal treatment allocation policies that target distributional welfare. Most literature on treatment choice has considered utilitarian welfare based on the conditional average treatment effect (ATE). While…

Methodology · Statistics 2025-04-30 Yifan Cui , Sukjin Han

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

Randomized controlled trials (RCTs) are considered the gold standard for estimating the average treatment effect (ATE) of interventions. One use of RCTs is to study the causes of global poverty -- a subject explicitly cited in the 2019…

Machine Learning · Computer Science 2023-05-26 Connor T. Jerzak , Fredrik Johansson , Adel Daoud

The Average Treatment Effect (ATE) is a foundational metric in causal inference, widely used to assess intervention efficacy in randomized controlled trials (RCTs). However, in many applications -- particularly in healthcare -- this static…

Machine Learning · Computer Science 2025-07-23 Julianna Piskorz , Krzysztof Kacprzyk , Harry Amad , Mihaela van der Schaar