English
Related papers

Related papers: Continuous Treatment Effect Estimation Using Gradi…

200 papers

We study the problem of learning conditional average treatment effects (CATE) from observational data with unobserved confounders. The CATE function maps baseline covariates to individual causal effect predictions and is key for…

Machine Learning · Statistics 2018-10-09 Nathan Kallus , Xiaojie Mao , Angela Zhou

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

The strata-specific treatment effect or so-called blip for a randomly drawn strata of confounders defines a random variable and a corresponding cumulative distribution function. However, the CDF is not pathwise differentiable, necessitating…

Methodology · Statistics 2018-12-31 Jonathan Levy , Mark van der Laan

Treatment effect estimation can assist in effective decision-making in e-commerce, medicine, and education. One popular application of this estimation lies in the prediction of the impact of a treatment (e.g., a promotion) on an outcome…

Machine Learning · Computer Science 2023-09-26 Xiaofeng Lin , Guoxi Zhang , Xiaotian Lu , Han Bao , Koh Takeuchi , Hisashi Kashima

The Average Treatment Effect (ATE) is a global measure of the effectiveness of an experimental treatment intervention. Classical methods of its estimation either ignore relevant covariates or do not fully exploit them. Moreover, past work…

Methodology · Statistics 2013-11-05 Emil Pitkin , Richard Berk , Lawrence Brown , Andreas Buja , Ed George , Kai Zhang , Linda Zhao

This paper provides estimation and inference methods for a conditional average treatment effects (CATE) characterized by a high-dimensional parameter in both homogeneous cross-sectional and unit-heterogeneous dynamic panel data settings. In…

Machine Learning · Statistics 2022-12-13 Vira Semenova , Matt Goldman , Victor Chernozhukov , Matt Taddy

Recent years have seen a swell in methods that focus on estimating "individual treatment effects". These methods are often focused on the estimation of heterogeneous treatment effects under ignorability assumptions. This paper hopes to draw…

Methodology · Statistics 2021-08-12 Brian G. Vegetabile

Evaluating treatment effect heterogeneity widely informs treatment decision making. At the moment, much emphasis is placed on the estimation of the conditional average treatment effect via flexible machine learning algorithms. While these…

Methodology · Statistics 2021-05-07 Lihua Lei , Emmanuel J. Candès

Individual Treatment Effect (ITE) estimation is an extensively researched problem, with applications in various domains. We model the case where there exists heterogeneous non-compliance to a randomly assigned treatment, a typical situation…

Machine Learning · Statistics 2020-10-26 Thibaud Rahier , Amélie Héliou , Matthieu Martin , Christophe Renaudin , Eustache Diemert

In nonseparable triangular models with a binary endogenous treatment and a binary instrumental variable, Vuong and Xu (2017) established identification results for individual treatment effects (ITEs) under the rank invariance assumption.…

Econometrics · Economics 2025-03-10 Jun Ma , Vadim Marmer , Zhengfei Yu

We investigate the problem of machine learning-based (ML) predictive inference on individual treatment effects (ITEs). Previous work has focused primarily on developing ML-based meta-learners that can provide point estimates of the…

Machine Learning · Computer Science 2023-08-30 Ahmed Alaa , Zaid Ahmad , Mark van der Laan

Individual Treatment Effect (ITE) prediction is an important area of research in machine learning which aims at explaining and estimating the causal impact of an action at the granular level. It represents a problem of growing interest in…

Individualized randomized experiments are central to online platforms for optimizing personalized decisions in complex environments. In two-sided markets, however, standard treatment effect estimation is often invalid due to strong temporal…

Methodology · Statistics 2026-05-27 Shuguang Yu , Ting Li , Yuchen Lu , Chengchun Shi , Fan Zhou , Zhichao Zou , Peng Zhen , Hongtu Zhu

Evaluation of intervention in a multiagent system, e.g., when humans should intervene in autonomous driving systems and when a player should pass to teammates for a good shot, is challenging in various engineering and scientific fields.…

Artificial Intelligence · Computer Science 2024-02-20 Keisuke Fujii , Koh Takeuchi , Atsushi Kuribayashi , Naoya Takeishi , Yoshinobu Kawahara , Kazuya Takeda

We regularly consider answering counterfactual questions in practice, such as "Would people with diabetes take a turn for the better had they choose another medication?". Observational studies are growing in significance in answering such…

Machine Learning · Computer Science 2022-08-16 Guanglin Zhou , Lina Yao , Xiwei Xu , Chen Wang , Liming Zhu

Uncertainty quantification for individual treatment effects (ITEs) is a daunting challenge in causal inference. Motivated by recent advances in conformal prediction, several works aim to construct distribution-free prediction sets for ITEs…

Methodology · Statistics 2026-05-07 Chongguang Tao , Zheng Zhou , Yuhong Yang

Inferring the causal effects of time-varying treatments is often hindered by highly variable inverse propensity weights, particularly in settings with limited covariate overlap. Building on the key framework of Imai and Ratkovic (2015), we…

Methodology · Statistics 2025-11-06 Yige Li , María de los Angeles Resa , José R. Zubizarreta

Predicated on the increasing abundance of electronic health records, we investi- gate the problem of inferring individualized treatment effects using observational data. Stemming from the potential outcomes model, we propose a novel multi-…

Machine Learning · Computer Science 2017-05-30 Ahmed M. Alaa , Mihaela van der Schaar

The quantification of treatment effects plays an important role in a wide range of applications, including policy making and bio-pharmaceutical research. In this article, we study the quantile treatment effect (QTE) while addressing two…

Statistics Theory · Mathematics 2025-03-25 Jiachen Sun , Yin Xia

Within the field of causal inference, we consider the problem of estimating heterogeneous treatment effects from data. We propose and validate a novel approach for learning feature representations to aid the estimation of the conditional…

Machine Learning · Statistics 2022-06-23 Michael C. Burkhart , Gabriel Ruiz