Continuous and discrete-time survival prediction with neural networks

Håvard Kvamme () and Ørnulf Borgan ()
Additional contact information
Håvard Kvamme: University of Oslo
Ørnulf Borgan: University of Oslo

Lifetime Data Analysis: An International Journal Devoted to Statistical Methods and Applications for Time-to-Event Data, 2021, vol. 27, issue 4, No 7, 710-736

Abstract: Abstract Due to rapid developments in machine learning, and in particular neural networks, a number of new methods for time-to-event predictions have been developed in the last few years. As neural networks are parametric models, it is more straightforward to integrate parametric survival models in the neural network framework than the popular semi-parametric Cox model. In particular, discrete-time survival models, which are fully parametric, are interesting candidates to extend with neural networks. The likelihood for discrete-time survival data may be parameterized by the probability mass function (PMF) or by the discrete hazard rate, and both of these formulations have been used to develop neural network-based methods for time-to-event predictions. In this paper, we review and compare these approaches. More importantly, we show how the discrete-time methods may be adopted as approximations for continuous-time data. To this end, we introduce two discretization schemes, corresponding to equidistant times or equidistant marginal survival probabilities, and two ways of interpolating the discrete-time predictions, corresponding to piecewise constant density functions or piecewise constant hazard rates. Through simulations and study of real-world data, the methods based on the hazard rate parametrization are found to perform slightly better than the methods that use the PMF parametrization. Inspired by these investigations, we also propose a continuous-time method by assuming that the continuous-time hazard rate is piecewise constant. The method, named PC-Hazard, is found to be highly competitive with the aforementioned methods in addition to other methods for survival prediction found in the literature.

Keywords: Time-to-event prediction; Neural networks; Interpolation; Discretization (search for similar items in EconPapers)
Date: 2021
References: View complete reference list from CitEc
Citations: View citations in EconPapers (3)

Downloads: (external link)
http://link.springer.com/10.1007/s10985-021-09532-6 Abstract (text/html)
Access to the full text of the articles in this series is restricted.

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:spr:lifeda:v:27:y:2021:i:4:d:10.1007_s10985-021-09532-6

Ordering information: This journal article can be ordered from
http://www.springer.com/journal/10985

DOI: 10.1007/s10985-021-09532-6

Access Statistics for this article

Lifetime Data Analysis: An International Journal Devoted to Statistical Methods and Applications for Time-to-Event Data is currently edited by Mei-Ling Ting Lee

More articles in Lifetime Data Analysis: An International Journal Devoted to Statistical Methods and Applications for Time-to-Event Data from Springer
Bibliographic data for series maintained by Sonal Shukla () and Springer Nature Abstracting and Indexing ().