arXiv Machine Learning

Flexible Doubly Robust Proximal Causal Inference for Censored Outcomes

arXiv Machine Learning
Jul 22

When Are Scoring Rules Proper? Bridging Theory and Practice in Survival Model Evaluation

arXiv:2212. 05260v4 Announce Type: replace-cross Abstract: Proper scoring rules encourage probabilistic predictions that match the true underlying distribution and are central to model evaluation, with increasing relevance in automated workflows such as AutoML.

By John Zobolas, Raphael Sonabend, Riccardo De Bin, Johannes Piller, Philipp Kopper, Lukas Burk, Andreas Bender
arXiv Machine Learning
5d ago

Efficient Learning of Survival Analysis Models Under Outcome-Dependent Truncation and Censoring

The paper investigates efficient learning of survival analysis models when data are subject to outcome-dependent truncation and right-censoring. It demonstrates that Cox regression can be trained with first‑order optimization when the baseline hazard is known, but joint learning of the baseline hazard introduces strict suboptimal local minima. Experiments on four datasets show that accounting for truncation reduces prediction bias and yields orders‑of‑magnitude speedups over expectation maximization.

By Itai Zilberstein, Ioannis Anagnostides, Arman Kilic, Tuomas Sandholm
arXiv Statistics ML
Oct 5

Flexible Nonparametric Inference for Causal Effects under the Front-Door Model

The paper introduces new one‑step and targeted minimum loss‑based estimators for the average treatment effect and the average treatment effect on the treated under the front‑door criterion. These estimators use multiple parameterizations of the observed data distribution, can avoid modeling the mediator density, and are compatible with flexible machine‑learning nuisance estimation. The authors derive conditions for root‑n consistency, asymptotic linearity, and develop doubly robust tests for identification assumptions, demonstrating improved efficiency and favorable performance in simulations and real‑world applications in education and emergency medicine.

By Anna Guo, David Benkeser, Razieh Nabi
Hugging Face Trending Papers
Oct 8

Efficient Learning of Survival Analysis Models Under Outcome-Dependent Truncation and Censoring

The paper investigates efficient learning of survival analysis models when data are subject to outcome‑dependent truncation and right‑censoring. It shows that Cox regression can be trained with first‑order methods when the baseline hazard is known, but joint estimation of the baseline hazard introduces non‑convexity and local minima. Experiments on four datasets demonstrate that accounting for truncation reduces bias, works with neural networks, and yields orders‑of‑magnitude speedups over expectation‑maximization.

arXiv Machine Learning
Jun 4

SurvPFN: Towards Foundation Models for Survival Predictions

arXiv:2606. 04564v1 Announce Type: new Abstract: Tabular foundation models (TFMs) have made rapid progress in standard classification and regression, but time-to-event survival prediction tasks have remained largely untouched.

By Samuel B\"ohm (Institute of Epidemiology and Prevention, Medical Center - University of Freiburg, Faculty of Medicine, University of Freiburg, Freiburg, Germany), Lennart Purucker (Department of Computer Science, University of Freiburg, Freiburg, Germany, PriorLabs, Freiburg, Germany), Frank Hutter (Department of Computer Science, University of Freiburg, Freiburg, Germany, PriorLabs, Freiburg, Germany), Pascal Schlosser (Institute of Epidemiology and Prevention, Medical Center - University of Freiburg, Faculty of Medicine, University of Freiburg, Freiburg, Germany, Department of Epidemiology, Johns Hopkins Bloomberg School of Public Health, Baltimore, Maryland, US, CIBSS - Centre for Integrative Biological Signalling Studies, University of Freiburg, Freiburg, Germany)
Hugging Face Trending Papers
Jun 4

Proper Scoring Rules for Right-Censored Survival Data

Proper scoring rules provide a rigorous theoretical basis for the training and evaluation of probabilistic forecasts. However, in the presence of right censoring, the event time is only partially observed, rendering conventional scoring rules inapplicable in their standard form.