English
Related papers

Related papers: Continuous Treatment Effect Estimation Using Gradi…

200 papers

Accurately estimating treatment effects over time is crucial in fields such as precision medicine, epidemiology, economics, and marketing. Many current methods for estimating treatment effects over time assume that all confounders are…

Machine Learning · Statistics 2025-11-11 Mouad El Bouchattaoui , Myriam Tami , Benoit Lepetit , Paul-Henry Cournède

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 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

Continuous treatments (e.g., doses) arise often in practice, but many available causal effect estimators are limited by either requiring parametric models for the effect curve, or by not allowing doubly robust covariate adjustment. We…

Methodology · Statistics 2017-04-21 Edward H. Kennedy , Zongming Ma , Matthew D. McHugh , Dylan S. Small

We study the problem of model selection in causal inference, specifically for conditional average treatment effect (CATE) estimation. Unlike machine learning, there is no perfect analogue of cross-validation for model selection as we do not…

Machine Learning · Computer Science 2024-04-30 Divyat Mahajan , Ioannis Mitliagkas , Brady Neal , Vasilis Syrgkanis

Heterogeneous treatment effects (HTEs) are commonly identified during randomized controlled trials (RCTs). Identifying subgroups of patients with similar treatment effects is of high interest in clinical research to advance precision…

Machine Learning · Computer Science 2022-12-06 Peniel N. Argaw , Elizabeth Healey , Isaac S. Kohane

State-level policy studies often conduct heterogeneity analyses that quantify how treatment effects vary across state characteristics. These analyses may be used to inform state-specific policy decisions, or to infer how the effect of a…

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

Randomized clinical trials typically aim to estimate a marginal treatment effect. While covariate adjustment can improve precision, it may change the estimand in nonlinear models due to noncollapsibility, leading to conditional rather than…

Methodology · Statistics 2026-05-25 Leticia Wuethrich , Torsten Hothorn

We consider the problem of estimating the effects of a binary treatment on a continuous outcome of interest from observational data in the absence of confounding by unmeasured factors. We provide a new estimator of the population average…

Methodology · Statistics 2020-08-04 James Robins , Mariela Sued , Quanhong Lei-Gomez , Andrea Rotnitzky

We consider the problem of estimating the average treatment effect (ATE) in a semi-supervised learning setting, where a very small proportion of the entire set of observations are labeled with the true outcome but features predictive of the…

Methodology · Statistics 2020-10-27 David Cheng , Ashwin Ananthakrishnan , Tianxi Cai

In semi-logarithmic regressions, treatment coefficients are often interpreted as approximations of the average treatment effect (ATE) in percentage points. This paper highlights the overlooked bias of this approximation under treatment…

Econometrics · Economics 2026-02-04 Ying Zeng

This paper develops a novel nonparametric identification method for treatment effects in settings where individuals self-select into treatment sequences. I propose an identification strategy which relies on a dynamic version of standard…

Econometrics · Economics 2023-12-05 Pedro Picchetti

Causal inference from observational data requires untestable identification assumptions. If these assumptions apply, machine learning (ML) methods can be used to study complex forms of causal effect heterogeneity. Recently, several ML…

Methodology · Statistics 2023-12-20 Richard Post , Isabel van den Heuvel , Marko Petkovic , Edwin van den Heuvel

Counterfactual inference has become a ubiquitous tool in online advertisement, recommendation systems, medical diagnosis, and econometrics. Accurate modeling of outcome distributions associated with different interventions -- known as…

Machine Learning · Statistics 2021-07-13 Krikamol Muandet , Motonobu Kanagawa , Sorawit Saengkyongam , Sanparith Marukatat

Patients in clinical studies often exhibit heterogeneous treatment effect (HTE). Classical subgroup analyses provide inferential tools to test for effect modification, while modern machine learning methods estimate the Conditional Average…

Applications · Statistics 2026-01-05 Nan Miles Xi , Xin Huang , Lin Wang

We study the benign overfitting theory in the prediction of the conditional average treatment effect (CATE), with linear regression models. As the development of machine learning for causal inference, a wide range of large-scale models for…

Econometrics · Economics 2022-02-17 Masahiro Kato , Masaaki Imaizumi

In this paper, we provide efficient estimators and honest confidence bands for a variety of treatment effects including local average (LATE) and local quantile treatment effects (LQTE) in data-rich environments. We can handle very many…

Statistics Theory · Mathematics 2018-01-08 Alexandre Belloni , Victor Chernozhukov , Ivan Fernández-Val , Christian Hansen

The average treatment effect (ATE) is widely used to evaluate the effectiveness of drugs and other medical interventions. In safety-critical applications like medicine, reliable inferences about the ATE typically require valid uncertainty…

Machine Learning · Computer Science 2025-10-13 Maresa Schröder , Justin Hartenstein , Stefan Feuerriegel

Estimation of individual treatment effect in observational data is complicated due to the challenges of confounding and selection bias. A useful inferential framework to address this is the counterfactual (potential outcomes) model which…

Machine Learning · Statistics 2017-01-23 Min Lu , Saad Sadiq , Daniel J. Feaster , Hemant Ishwaran