English

On Training Survival Models with Scoring Rules

Machine Learning 2024-11-14 v2 Artificial Intelligence Computation Machine Learning

Abstract

Scoring rules are an established way of comparing predictive performances across model classes. In the context of survival analysis, they require adaptation in order to accommodate censoring. This work investigates using scoring rules for model training rather than evaluation. Doing so, we establish a general framework for training survival models that is model agnostic and can learn event time distributions parametrically or non-parametrically. In addition, our framework is not restricted to any specific scoring rule. While we focus on neural network-based implementations, we also provide proof-of-concept implementations using gradient boosting, generalized additive models, and trees. Empirical comparisons on synthetic and real-world data indicate that scoring rules can be successfully incorporated into model training and yield competitive predictive performance with established time-to-event models.

Keywords

Cite

@article{arxiv.2403.13150,
  title  = {On Training Survival Models with Scoring Rules},
  author = {Philipp Kopper and David Rügamer and Raphael Sonabend and Bernd Bischl and Andreas Bender},
  journal= {arXiv preprint arXiv:2403.13150},
  year   = {2024}
}

Comments

9 pages, 3 figures

R2 v1 2026-06-28T15:26:34.775Z