English
Related papers

Related papers: SurvITE: Learning Heterogeneous Treatment Effects …

200 papers

Every design choice will have different effects on different units. However traditional A/B tests are often underpowered to identify these heterogeneous effects. This is especially true when the set of unit-level attributes is…

Artificial Intelligence · Computer Science 2016-11-09 Alexander Peysakhovich , Akos Lada

Estimating how a treatment affects units individually, known as heterogeneous treatment effect (HTE) estimation, is an essential part of decision-making and policy implementation. The accumulation of large amounts of data in many domains,…

Machine Learning · Computer Science 2022-06-28 Christopher Tran , Elena Zheleva

Estimating treatment effects from observational data is challenging due to two main reasons: (a) hidden confounding, and (b) covariate mismatch (control and treatment groups not having identical distributions). Long lines of works exist…

Machine Learning · Computer Science 2025-04-30 Praharsh Nanavati , Ranjitha Prasad , Karthikeyan Shanmugam

Treatment effect heterogeneity plays an important role in many areas of causal inference and within recent years, estimation of the conditional average treatment effect (CATE) has received much attention in the statistical community. While…

Methodology · Statistics 2024-12-17 Simon Christoffer Ziersen , Torben Martinussen

In randomized trials involving multiple treatments, bivariate survival outcomes present significant analytical challenges for making decisions. This paper addresses the problem of deriving optimal individualized treatment rules to maximize…

Machine Learning · Statistics 2026-05-29 Kun Ren , Yifan Cui , Wen Su

There is strong interest in estimating how the magnitude of treatment effects of an intervention vary across sub-groups of the population of interest. In our paper, we propose a two-study approach to first propose and then test…

Methodology · Statistics 2020-06-23 Rahul Ladhania , Amelia Haviland , Neeraj Sood , Edward Kennedy , Ateev Mehrotra

Analyzing data from multiple sources offers valuable opportunities to improve the estimation efficiency of causal estimands. However, this analysis also poses many challenges due to population heterogeneity and data privacy constraints.…

Methodology · Statistics 2025-10-23 Rong Zhao , Jason Falvey , Xu Shi , Vernon M. Chinchilli , Chixiang Chen

Undertaking causal inference with observational data is incredibly useful across a wide range of tasks including the development of medical treatments, advertisements and marketing, and policy making. There are two significant challenges…

Machine Learning · Statistics 2022-01-19 Matthew James Vowels , Necati Cihan Camgoz , Richard Bowden

A central focus in survival analysis is examining how covariates influence survival time. These covariate effects are often found to be either time-varying, heterogeneous - such as being specific to patients, treatments, or subgroups - or…

Methodology · Statistics 2025-01-24 Niklas Hagemann , Thomas Kneib , Kathrin Möllenhoff

Understanding treatment effect heterogeneity has become an increasingly popular task in various fields, as it helps design personalized advertisements in e-commerce or targeted treatment in biomedical studies. However, most of the existing…

Methodology · Statistics 2024-07-12 Waverly Wei , Xinwei Ma , Jingshen Wang

Many applications of causal inference require using treatment effects estimated on a study population to make decisions in a separate target population. We consider the challenging setting where there are covariates that are observed in the…

Machine Learning · Computer Science 2024-10-22 Khurram Yamin , Vibhhu Sharma , Ed Kennedy , Bryan Wilder

Often in follow-up studies intermediate events occur in some patients, such as reinterventions or adverse events. These intermediate events directly affect the shapes of their longitudinal profiles. Our work is motivated by two studies in…

Developing tools for estimating heterogeneous treatment effects (HTE) and individualized treatment effects has been an area of active research in recent years. While these tools have proven to be useful in many contexts, a concern when…

Methodology · Statistics 2025-03-07 Mahsa Ashouri , Nicholas C. Henderson

Consider the problem of improving the estimation of conditional average treatment effects (CATE) for a target domain of interest by leveraging related information from a source domain with a different feature space. This heterogeneous…

Machine Learning · Computer Science 2022-10-13 Ioana Bica , Mihaela van der Schaar

To draw real-world evidence about the comparative effectiveness of multiple time-varying treatments on patient survival, we develop a joint marginal structural survival model and a novel weighting strategy to account for time-varying…

Methodology · Statistics 2023-08-08 Liangyuan Hu , Jiayi Ji , Himanshu Joshi , Erick Scott , Fan Li

Domain adaptation and covariate shift are big issues in deep learning and they ultimately affect any causal inference algorithms that rely on deep neural networks. Causal effect variational autoencoder (CEVAE) is trained to predict the…

Machine Learning · Computer Science 2022-09-22 Daniel Jiwoong Im , Kyunghyun Cho , Narges Razavian

Estimating the effects of interventions on patient outcome is one of the key aspects of personalized medicine. Their inference is often challenged by the fact that the training data comprises only the outcome for the administered treatment,…

Randomized controlled experiment has long been accepted as the golden standard for establishing causal link and estimating causal effect in various scientific fields. Average treatment effect is often used to summarize the effect…

Applications · Statistics 2016-10-14 Alex Deng , Pengchuan Zhang , Shouyuan Chen , Dong Woo Kim , Jiannan Lu

Estimating treatment effects is crucial for personalized decision-making in medicine, but this task faces unique challenges in clinical practice. At training time, models for estimating treatment effects are typically trained on…

Machine Learning · Computer Science 2025-10-31 Yuchen Ma , Dennis Frauen , Jonas Schweisthal , Stefan Feuerriegel

Estimating heterogeneous treatment effects (HTEs) from right-censored survival data is critical in high-stakes applications such as precision medicine and individualized policy-making. Yet, the survival analysis setting poses unique…

Machine Learning · Computer Science 2026-03-06 Shahriar Noroozizadeh , Xiaobin Shen , Jeremy C. Weiss , George H. Chen