English

SurvSurf: a partially monotonic neural network for first-hitting time prediction of intermittently observed discrete and continuous sequential events

Machine Learning 2025-04-08 v1 Artificial Intelligence Machine Learning Statistics Theory Applications Statistics Theory

Abstract

We propose a neural-network based survival model (SurvSurf) specifically designed for direct and simultaneous probabilistic prediction of the first hitting time of sequential events from baseline. Unlike existing models, SurvSurf is theoretically guaranteed to never violate the monotonic relationship between the cumulative incidence functions of sequential events, while allowing nonlinear influence from predictors. It also incorporates implicit truths for unobserved intermediate events in model fitting, and supports both discrete and continuous time and events. We also identified a variant of the Integrated Brier Score (IBS) that showed robust correlation with the mean squared error (MSE) between the true and predicted probabilities by accounting for implied truths about the missing intermediate events. We demonstrated the superiority of SurvSurf compared to modern and traditional predictive survival models in two simulated datasets and two real-world datasets, using MSE, the more robust IBS and by measuring the extent of monotonicity violation.

Keywords

Cite

@article{arxiv.2504.04997,
  title  = {SurvSurf: a partially monotonic neural network for first-hitting time prediction of intermittently observed discrete and continuous sequential events},
  author = {Yichen Kelly Chen and Sören Dittmer and Kinga Bernatowicz and Josep Arús-Pous and Kamen Bliznashki and John Aston and James H. F. Rudd and Carola-Bibiane Schönlieb and James Jones and Michael Roberts},
  journal= {arXiv preprint arXiv:2504.04997},
  year   = {2025}
}

Comments

41 pages, 18 figures (including supplemental information). Submitted to RSS: Data Science and Artificial Intelligence