Deep Learning for Patient-Specific Kidney Graft Survival Analysis

An accurate model of patient-specific kidney graft survival distributions can help to improve shared-decision making in the treatment and care of patients. In this paper, we propose a deep learning method that directly models the survival function instead of estimating the hazard function to predict survival times for graft patients based on the principle of multi-task learning. By learning to jointly predict the time of the event, and its rank in the cox partial log likelihood framework, our deep learning approach outperforms, in terms of survival time prediction quality and concordance index, other common methods for survival analysis, including the Cox Proportional Hazards model and a network trained on the cox partial log-likelihood.

[1]  Ivan Bratko,et al.  Machine Learning for Survival Analysis: A Case Study on Recurrence of Prostate Cancer , 1999, AIMDM.

[2]  Douglas E Schaubel,et al.  A Comprehensive Risk Quantification Score for Deceased Donor Kidneys: The Kidney Donor Risk Index , 2009, Transplantation.

[3]  B. Efron The Efficiency of Cox's Likelihood Function for Censored Data , 1977 .

[4]  Adler J. Perotte,et al.  Deep Survival Analysis , 2016, MLHC.

[5]  Xiaoqian Jiang,et al.  Predicting accurate probabilities with a ranking loss , 2012, ICML.

[6]  Balaji Krishnapuram,et al.  On Ranking in Survival Analysis: Bounds on the Concordance Index , 2007, NIPS.

[7]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

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

[9]  Russell Greiner,et al.  Learning Patient-Specific Cancer Survival Distributions as a Sequence of Dependent Regressors , 2011, NIPS.

[10]  David R. Cox,et al.  Regression models and life tables (with discussion , 1972 .

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

[12]  E. Kaplan,et al.  Nonparametric Estimation from Incomplete Observations , 1958 .

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

[14]  José Antonio Gómez-Ruiz,et al.  A combined neural network and decision trees model for prognosis of breast cancer relapse , 2003, Artif. Intell. Medicine.

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

[16]  H. Ishwaran Variable importance in binary regression trees and forests , 2007, 0711.2434.

[17]  Amir Hossein Hashemian,et al.  A Comparison Between Cox Regression and Parametric Methods in Analyzing Kidney Transplant Survival , 2013 .