A General Framework for Survival Analysis and Multi-State Modelling

TL;DR

This paper introduces a flexible neural ODE-based framework for multi-state survival analysis, overcoming limitations of traditional models like Cox by modeling complex transitions and individual heterogeneity.

Contribution

It proposes a novel neural ODE approach for multi-state survival modeling, incorporating a variational latent variable model for uncertainty quantification and interpretability.

Findings

State-of-the-art performance on survival datasets

Effective modeling of competing events and transitions

Enhanced interpretability and clustering of multi-state outcomes

Abstract

Survival models are a popular tool for the analysis of time to event data with applications in medicine, engineering, economics, and many more. Advances like the Cox proportional hazard model have enabled researchers to better describe hazard rates for the occurrence of single fatal events, but are unable to accurately model competing events and transitions. Common phenomena are often better described through multiple states, for example: the progress of a disease modeled as healthy, sick and dead instead of healthy and dead, where the competing nature of death and disease has to be taken into account. Moreover, Cox models are limited by modeling assumptions, like proportionality of hazard rates and linear effects. Individual characteristics can vary significantly between observational units, like patients, resulting in idiosyncratic hazard rates and different disease trajectories. These considerations require flexible modeling assumptions. To overcome these issues, we propose the use of neural ordinary differential equations as a flexible and general method for estimating multi-state survival models by directly solving the Kolmogorov forward equations. To quantify the uncertainty in the resulting individual cause-specific hazard rates, we further introduce a variational latent variable model and show that this enables meaningful clustering with respect to multi-state outcomes as well as interpretability regarding covariate values. We show that our model exhibits state-of-the-art performance on popular survival data sets and demonstrate its efficacy in a multi-state setting