English
Related papers

Related papers: SurvCaus : Representation Balancing for Survival C…

200 papers

We propose an empirically stable and asymptotically efficient covariate-balancing approach to the problem of estimating survival causal effects in data with conditionally-independent censoring. This addresses a challenge often encountered…

Robust estimation of heterogeneous treatment effects is a fundamental challenge for optimal decision-making in domains ranging from personalized medicine to educational policy. In recent years, predictive machine learning has emerged as a…

Machine Learning · Statistics 2025-06-23 Maximilian Schuessler , Erik Sverdrup , Robert Tibshirani

One of the major challenges in estimating conditional potential outcomes and conditional average treatment effects (CATE) is the presence of hidden confounders. Since testing for hidden confounders cannot be accomplished only with…

Machine Learning · Computer Science 2025-06-17 Ahmed Aloui , Juncheng Dong , Ali Hasan , Vahid Tarokh

Estimating individual level treatment effects (ITE) from observational data is a challenging and important area in causal machine learning and is commonly considered in diverse mission-critical applications. In this paper, we propose an…

Machine Learning · Computer Science 2019-06-10 Sungyub Kim , Yongsu Baek , Sung Ju Hwang , Eunho Yang

Estimating counterfactual outcomes over time has the potential to unlock personalized healthcare by assisting decision-makers to answer ''what-iF'' questions. Existing causal inference approaches typically consider regular, discrete-time…

Machine Learning · Computer Science 2022-06-17 Nabeel Seedat , Fergus Imrie , Alexis Bellot , Zhaozhi Qian , Mihaela van der Schaar

We develop a novel approach to partially identify causal estimands, such as the average treatment effect (ATE), from observational data. To better satisfy the stable unit treatment value assumption (SUTVA) we utilize stochastic…

Methodology · Statistics 2024-07-30 Brian Knaeble , Braxton Osting , Placede Tshiaba

Learning the Individual Treatment Effect (ITE) is essential for personalized decision-making, yet causal inference has traditionally focused on aggregated treatment effects. While integrating conformal prediction with causal inference can…

Methodology · Statistics 2025-01-23 Chenyin Gao , Peter B. Gilbert , Larry Han

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

Observational studies have recently received significant attention from the machine learning community due to the increasingly available non-experimental observational data and the limitations of the experimental studies, such as…

Machine Learning · Computer Science 2023-01-26 Vinod Kumar Chauhan , Soheila Molaei , Marzia Hoque Tania , Anshul Thakur , Tingting Zhu , David A. Clifton

Reliable estimation of treatment effects from observational data is important in many disciplines such as medicine. However, estimation is challenging when unconfoundedness as a standard assumption in the causal inference literature is…

Machine Learning · Computer Science 2024-10-15 Jonas Schweisthal , Dennis Frauen , Maresa Schröder , Konstantin Hess , Niki Kilbertus , Stefan Feuerriegel

Estimating causal effects on time-to-event outcomes from observational data is particularly challenging due to censoring, limited sample sizes, and non-random treatment assignment. The need for answering such "when-if" questions--how the…

Machine Learning · Computer Science 2025-11-19 Jessy Xinyi Han , Devavrat Shah

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

While average treatment effects (ATE) and conditional average treatment effects (CATE) provide valuable population- and subgroup-level summaries, they fail to capture uncertainty at the individual level. For high-stakes decision-making,…

Methodology · Statistics 2026-03-31 Juraj Bodik , Yaxuan Huang , Bin Yu

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

There is intense interest in applying machine learning to problems of causal inference in fields such as healthcare, economics and education. In particular, individual-level causal inference has important applications such as precision…

Machine Learning · Statistics 2017-05-17 Uri Shalit , Fredrik D. Johansson , David Sontag

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 conditional average treatment effects (CATE) is challenging, especially when treatment information is missing. Although this is a widespread problem in practice, CATE estimation with missing treatments has received little…

Machine Learning · Statistics 2023-04-19 Milan Kuzmanovic , Tobias Hatt , Stefan Feuerriegel

In an era where diverse and complex data are increasingly accessible, the precise prediction of individual treatment effects (ITE) becomes crucial across fields such as healthcare, economics, and public policy. Current state-of-the-art…

Machine Learning · Statistics 2025-01-28 Baozhen Wang , Xingye Qiao

The conditional average treatment effect (CATE) is frequently estimated to refute the homogeneous treatment effect assumption. Under this assumption, all units making up the population under study experience identical benefit from a given…

In the data-driven era, large-scale datasets are routinely collected and analyzed using machine learning (ML) and artificial intelligence (AI) to inform decisions in high-stakes domains such as healthcare, employment, and criminal justice,…

Machine Learning · Computer Science 2026-05-13 Drago Plecko