Time-to-Event Prediction with Neural Networks and Cox Regression

Abstract

New methods for time-to-event prediction are proposed by extending the Cox proportional hazards model with neural networks. Building on methodology from nested case-control studies, we propose a loss function that scales well to large data sets, and enables fitting of both proportional and non-proportional extensions of the Cox model. Through simulation studies, the proposed loss function is verified to be a good approximation for the Cox partial log-likelihood. The proposed methodology is compared to existing methodologies on real-world data sets, and is found to be highly competitive, typically yielding the best performance in terms of Brier score and binomial log-likelihood. A python package for the proposed methods is available at https://github.com/havakv/pycox.

References in corpus (7)

• Decoupled Weight Decay Regularization
• DeepSurv: Personalized Treatment Recommender System Using A Cox Proportional Hazards Deep Neural Network
• On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima
• Train longer, generalize better: closing the generalization gap in large batch training of neural networks
• Entity Embeddings of Categorical Variables
• Deep Neural Networks for Survival Analysis Based on a Multi-Task Framework
• Deep Learning for Patient-Specific Kidney Graft Survival Analysis

Cited by in corpus (17)

• mlr3proba: An R Package for Machine Learning in Survival Analysis
• SurvSHAP(t): Time-dependent explanations of machine learning survival models
• Medical Imaging and Machine Learning
• Predicting Time-to-conversion for Dementia of Alzheimer's Type using Multi-modal Deep Survival Analysis
• A survey on datasets for fairness-aware machine learning
• Variational Bayes survival analysis for unemployment modelling
• The Concordance Index decomposition: A measure for a deeper understanding of survival prediction models
• Tutorial on survival modeling with applications to omics data
• Multi-modal Heart Failure Risk Estimation based on Short ECG and Sampled Long-Term HRV
• Scaling Survival Analysis in Healthcare with Federated Survival Forests: A Comparative Study on Heart Failure and Breast Cancer Genomics
• Federated Survival Forests
• Survival stacking: casting survival analysis as a classification problem
• Beyond Cox Models: Assessing the Performance of Machine-Learning Methods in Non-Proportional Hazards and Non-Linear Survival Analysis
• Deep Learning-Based Discrete Calibrated Survival Prediction
• ICTSurF: Implicit Continuous-Time Survival Functions with Neural Networks
• Spectral Survival Analysis
• Maximum Likelihood Estimation of Flexible Survival Densities with Importance Sampling