Derivative-based neural modelling of cumulative distribution functions for survival analysis

Conference item

Survival models — particularly those able to account for patient comorbidities via competing risks analysis — offer valuable prognostic information to clinicians making critical decisions and represent a growing area of application for machine learning approaches. However, current methods typically involve restrictive parameterisations, discretisation of time or the modelling of only one event cause. In this paper, we highlight how general cumulative distribution functions can be naturally expressed via neural network-based ordinary differential equations and how this observation can be utilised in survival analysis. In particular, we present DeSurv, a neural derivative-based approach capable of avoiding aforementioned restrictions and flexibly modelling competing-risk survival data in continuous time. We apply DeSurv to both single-risk and competing-risk synthetic and real-world datasets and obtain results which compare favourably with current state-of-the-art models.

Authors: Danks, D; Yau, C

• Publication status: Published
• Peer review status: Peer reviewed
• Publisher: PMLR
• Host title: Proceedings of The 25th International Conference on Artificial Intelligence and Statistics
• Pages: 7240-7256
• Series: Proceedings of Machine Learning Research
• Series number: 151
• Publication date: 2022-05-03
• Acceptance date: 2022-01-29
• Event title: 25th International Conference on Artificial Intelligence and Statistics (AISTATS 2022)
• Event location: Virtual event
• Event website: https://aistats.org/aistats2022/
• Event start date: 2022-03-28
• Event end date: 2022-03-30
• EISSN: 2640-3498
• ISSN: 2640-3498
• Language: English