Survival Models: Proper Scoring Rule and Stochastic Optimization with Competing Risks - Collection des publications du laboratoire Bernoulli
Pré-Publication, Document De Travail Année : 2024

Survival Models: Proper Scoring Rule and Stochastic Optimization with Competing Risks

Résumé

When dealing with right-censored data, where some outcomes are missing due to a limited observation period, survival analysis —known as time-to-event analysis— focuses on predicting the time until an event of interest occurs. Multiple classes of outcomes lead to a classification variant: predicting the most likely event, a less explored area known as competing risks. Classic competing risks models couple architecture and loss, limiting scalability. To address these issues, we design a strictly proper censoring-adjusted separable scoring rule, allowing optimization on a subset of the data as each observation is evaluated independently. The loss estimates outcome probabilities and enables stochastic optimization for competing risks, which we use for efficient gradient boosting trees. SurvivalBoost not only outperforms 12 state-of-the-art models across several metrics on 4 real-life datasets, both in competing risks and survival settings, but also provides great calibration, the ability to predict across any time horizon, and computation times faster than existing methods.
Fichier principal
Vignette du fichier
main.pdf (1.13 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04617672 , version 1 (19-06-2024)
hal-04617672 , version 2 (12-09-2024)
hal-04617672 , version 3 (17-10-2024)
hal-04617672 , version 4 (21-10-2024)

Identifiants

  • HAL Id : hal-04617672 , version 4

Citer

Julie Alberge, Vincent Maladière, Olivier Grisel, Judith Abécassis, Gaël Varoquaux. Survival Models: Proper Scoring Rule and Stochastic Optimization with Competing Risks. 2024. ⟨hal-04617672v4⟩
128 Consultations
95 Téléchargements

Partager

More