Abstract
Neural networks have been investigated as models for survival data using a training criterion similar to that of the Cox proportional hazards model, a criterion not designed for clinical prediction. In this paper, we develop a new survival learning algorithm where a neural network ensemble minimizes the integrated Brier score. We compare the results obtained with this method to a standard implementation of random survival forests in R and to an ensemble of linear units.
Keywords
This is a preview of subscription content, log in via an institution.
Buying options
Tax calculation will be finalised at checkout
Purchases are for personal use only
Learn about institutional subscriptionsReferences
Schwarzer, G., Vach, W., Schumacher, M.: On the misuses of artificial neural networks for prognostic and diagnostic classification in oncology. Stat. Med. 19(4), 541–561 (2000)
Luck, M., Sylvain, T., Cardinal, H., Lodi, A., Bengio, Y.: Deep learning for patient-specific kidney graft survival analysis. CoRR abs/1705.10245 (2017)
Chapfuwa, P., et al.: Adversarial time-to-event modeling. In: ICML (2018)
Fotso, S.: Deep neural networks for survival analysis based on a multi-task framework. CoRR abs/1801.05512 (2018)
Giunchiglia, E., Nemchenko, A., van der Schaar, M.: RNN-SURV: a deep recurrent model for survival analysis. In: Kůrková, V., Manolopoulos, Y., Hammer, B., Iliadis, L., Maglogiannis, I. (eds.) ICANN 2018. LNCS, vol. 11141, pp. 23–32. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-01424-7_3
Katzman, J., Shaham, U., Cloninger, A., Bates, J., Jiang, T., Kluger, Y.: DeepSurv: personalized treatment recommender system using a cox proportional hazards deep neural network. BMC Med. Res. Methodol. 18, 24 (2018)
Manyam, R.B., Zhang, Y., Keeling, W.B., Binongo, J., Kayatta, M., Carter, S.: Deep learning approach for predicting 30 day readmissions after coronary artery bypass graft surgery. CoRR abs/1812.00596 (2018)
Nezhad, M.Z., Sadati, N., Yang, K., Zhu, D.: A deep active survival analysis approach for precision treatment recommendations: application of prostate cancer. Expert Syst. Appl. 115, 16–26 (2019)
Harrell, F.E., Califf, R.M., Pryor, D.B., Lee, K.L., Rosati, R.A.: Evaluating the yield of medical tests. JAMA 247(18), 2543–2546 (1982)
Brier, G.W.: Verification of forecasts expressed in terms of probability. Mon. Weather Rev. 78(1), 1–3 (1950)
Ishwaran, H., Kogalur, U., Blackstone, E., Lauer, M.: Random survival forests. Ann. Appl. Stat. 2(3), 841–860 (2008)
Therneau, T., Grambsch, P.: Modeling Survival Data: Extending the Cox Model. Statistics for Biology and Health. Springer, New York (2000). https://doi.org/10.1007/978-1-4757-3294-8
Fleming, T.R., Harrington, D.P.: Counting Processes and Survival Analysis. Wiley, Hoboken (2005)
Broët, P., De Rycke, Y., Tubert-Bitter, P., Lellouch, J., Asselain, B., Moreau, T.: A semiparametric approach for the two-sample comparison of survival times with long-term survivors. Biometrics 57(3), 844–852 (2001)
Klambauer, G., Unterthiner, T., Mayr, A., Hochreiter, S.: Self-normalizing neural networks. CoRR abs/1706.02515 (2017)
Graf, E., Schmoor, C., Sauerbrei, W., Schumacher, M.: Assessment and comparison of prognostic classification schemes for survival data. Stat. Med. 18(17–18), 2529–2545 (1999)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2020 Springer Nature Switzerland AG
About this paper
Cite this paper
de Montigny, S., Broët, P. (2020). Adapting Ensemble Neural Networks to Clinical Prediction in High-Dimensional Settings. In: Goutte, C., Zhu, X. (eds) Advances in Artificial Intelligence. Canadian AI 2020. Lecture Notes in Computer Science(), vol 12109. Springer, Cham. https://doi.org/10.1007/978-3-030-47358-7_15
Download citation
DOI: https://doi.org/10.1007/978-3-030-47358-7_15
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-47357-0
Online ISBN: 978-3-030-47358-7
eBook Packages: Computer ScienceComputer Science (R0)