
There is currently great interest in applying neural networks to prediction tasks in medicine. It is important for predictive models to be able to use survival data, where each patient has a known follow-up time and event/censoring indicator. This avoids information loss when training the model and enables generation of predicted survival curves. In this paper, we describe a discrete-time survival model that is designed to be used with neural networks, which we refer to as Nnet-survival. The model is trained with the maximum likelihood method using mini-batch stochastic gradient descent (SGD). The use of SGD enables rapid convergence and application to large datasets that do not fit in memory. The model is flexible, so that the baseline hazard rate and the effect of the input data on hazard probability can vary with follow-up time. It has been implemented in the Keras deep learning framework, and source code for the model and several examples is available online. We demonstrate the performance of the model on both simulated and real data and compare it to existing models Cox-nnet and Deepsurv.
FOS: Computer and information sciences, Computer Science - Machine Learning, QH301-705.5, Data Mining and Machine Learning, R, Machine Learning (stat.ML), Survival analysis, Machine Learning (cs.LG), Computer Science - Computers and Society, Statistics - Machine Learning, Machine learning, Computers and Society (cs.CY), Medicine, Biology (General), Neural networks
FOS: Computer and information sciences, Computer Science - Machine Learning, QH301-705.5, Data Mining and Machine Learning, R, Machine Learning (stat.ML), Survival analysis, Machine Learning (cs.LG), Computer Science - Computers and Society, Statistics - Machine Learning, Machine learning, Computers and Society (cs.CY), Medicine, Biology (General), Neural networks
| selected citations These citations are derived from selected sources. This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically). | 195 | |
| popularity This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network. | Top 1% | |
| influence This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically). | Top 1% | |
| impulse This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network. | Top 1% |
