A deep survival analysis method based on ranking

Survival analyses of populations and the establishment of prognoses for individual patients are important activities in the practice of medicine. Standard survival models, such as the Cox proportional hazards model, require extensive feature engineering or prior knowledge to model at an individual level. Some survival analysis models can avoid these problems by using machine learning extended the CPH model, and higher performance has been reported. In this paper, we propose an innovative loss function that is defined as the sum of an extended mean squared error loss and a pairwise ranking loss based on ranking information on survival data. We apply this loss function to optimize a deep feed-forward neural network (RankDeepSurv), which can be used to model survival data. We demonstrate that the performance of our model, RankDeepSurv, is superior to that of other state-of-the-art survival models based on an analysis of 4 public medical clinical datasets. When modelling the prognosis of nasopharyngeal carcinoma (NPC), RankDeepSurv achieved better prognostic accuracy than the CPH established by clinical experts. The difference between high and low risk groups in the RankDeepSurv model is greater than the difference in the CPH. The results show that our method has considerable potential to model survival data in medical settings.

[1]  Sepp Hochreiter,et al.  Self-Normalizing Neural Networks , 2017, NIPS.

[2]  William Nick Street,et al.  A Neural Network Model for Prognostic Prediction , 1998, ICML.

[3]  F. Markowetz,et al.  The genomic and transcriptomic architecture of 2,000 breast tumours reveals novel subgroups , 2012, Nature.

[4]  Sabine Van Huffel,et al.  Support vector methods for survival analysis: a comparison between ranking and regression approaches , 2011, Artif. Intell. Medicine.

[5]  W. Sauerbrei,et al.  Randomized 2 x 2 trial evaluating hormonal treatment and the duration of chemotherapy in node-positive breast cancer patients. German Breast Cancer Study Group. , 1994, Journal of clinical oncology : official journal of the American Society of Clinical Oncology.

[6]  Uri Shaham,et al.  DeepSurv: personalized treatment recommender system using a Cox proportional hazards deep neural network , 2016, BMC Medical Research Methodology.

[7]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[8]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[9]  Sabine Van Huffel,et al.  Improved performance on high-dimensional survival data by application of Survival-SVM , 2011, Bioinform..

[10]  Jieping Ye,et al.  Transfer Learning for Survival Analysis via Efficient L2,1-Norm Regularized Cox Regression , 2016, 2016 IEEE 16th International Conference on Data Mining (ICDM).

[11]  Tai-Hsien Ou Yang,et al.  Development of a Prognostic Model for Breast Cancer Survival in an Open Challenge Environment , 2013, Science Translational Medicine.

[12]  D Faraggi,et al.  A neural network model for survival data. , 1995, Statistics in medicine.

[13]  Junzhou Huang,et al.  WSISA: Making Survival Prediction from Whole Slide Histopathological Images , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[14]  Rui Sun,et al.  Feasibility and efficacy of chemoradiotherapy for elderly patients with locoregionally advanced nasopharyngeal carcinoma: results from a matched cohort analysis , 2013, Radiation oncology.

[15]  L. Goldman,et al.  The SUPPORT Prognostic Model: Objective Estimates of Survival for Seriously Ill Hospitalized Adults , 1995, Annals of Internal Medicine.

[16]  N. Brünner,et al.  The urokinase system of plasminogen activation and prognosis in 2780 breast cancer patients. , 2000, Cancer research.

[17]  F. Harrell,et al.  Evaluating the yield of medical tests. , 1982, JAMA.

[18]  Jiayu Zhou,et al.  Multi-Task Learning based Survival Analysis for Predicting Alzheimer's Disease Progression with Multi-Source Block-wise Missing Data , 2018, SDM.

[19]  J. Spertus,et al.  Development and Validation of a Prediction Rule for Benefit and Harm of Dual Antiplatelet Therapy Beyond 1 Year After Percutaneous Coronary Intervention. , 2016, JAMA.

[20]  D. Cox.  Regression Models and Life-Tables , 1972 .

[21]  Georg Heigold,et al.  An empirical study of learning rates in deep neural networks for speech recognition , 2013, 2013 IEEE International Conference on Acoustics, Speech and Signal Processing.

[22]  F. Harrell,et al.  Regression modelling strategies for improved prognostic prediction. , 1984, Statistics in medicine.

[23]  Wei Chu,et al.  A Support Vector Approach to Censored Targets , 2007, Seventh IEEE International Conference on Data Mining (ICDM 2007).

[24]  Junzhou Huang,et al.  Deep Correlational Learning for Survival Prediction from Multi-modality Data , 2017, MICCAI.

[25]  J. M. Jerez,et al.  Improvement of breast cancer relapse prediction in high risk intervals using artificial neural networks , 2005, Breast Cancer Research and Treatment.

[26]  Ping Wang,et al.  Machine Learning for Survival Analysis , 2019, ACM Comput. Surv..

[27]  R. Tibshirani,et al.  Semi-Supervised Methods to Predict Patient Survival from Gene Expression Data , 2004, PLoS biology.

[28]  H. Sze,et al.  Radical radiotherapy for nasopharyngeal carcinoma in elderly patients: the importance of co-morbidity assessment. , 2012, Oral oncology.

[29]  Ying Sun,et al.  Baseline serum lactate dehydrogenase levels for patients treated with intensity-modulated radiotherapy for nasopharyngeal carcinoma: a predictor of poor prognosis and subsequent liver metastasis. , 2012, International journal of radiation oncology, biology, physics.

[30]  Jieping Ye,et al.  A Multi-Task Learning Formulation for Survival Analysis , 2016, KDD.

[31]  Rui Sun,et al.  Establishment and Validation of Prognostic Nomograms for Endemic Nasopharyngeal Carcinoma. , 2016, Journal of the National Cancer Institute.

[32]  Hemant Ishwaran,et al.  Random Survival Forests , 2008, Wiley StatsRef: Statistics Reference Online.

[33]  S Van Huffel,et al.  Additive survival least‐squares support vector machines , 2010, Statistics in medicine.

[34]  K. Liestøl,et al.  Survival analysis and neural nets. , 1994, Statistics in medicine.

[35]  David W. Hosmer,et al.  Applied Survival Analysis: Regression Modeling of Time-to-Event Data , 2008 .

[36]  P. Royston,et al.  External validation of a Cox prognostic model: principles and methods , 2013, BMC Medical Research Methodology.

[37]  Jun Yan.  Survival Analysis: Techniques for Censored and Truncated Data , 2004 .

[38]  Sepp Hochreiter,et al.  Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs) , 2015, ICLR.