Skip to content
naulon

No retraction, correction or concern found when last checked on Oct 15, 2026.

DeepSurv: personalized treatment recommender system using a Cox proportional hazards deep neural network

Jared L. Katzman, Uri Shaham, Alexander Cloninger and 3 more

BMC Medical Research Methodology · 2018 · CC BY

doi:10.1186/s12874-018-0482-1

Abstract

Background. Medical practitioners use survival models to explore and understand the relationships between patients’ covariates (e.g. clinical and genetic features) and the effectiveness of various treatment options. Standard survival models like the linear Cox proportional hazards model require extensive feature engineering or prior medical knowledge to model treatment interaction at an individual level. While nonlinear survival methods, such as neural networks and survival forests, can inherently model these high-level interaction terms, they have yet to be shown as effective treatment recommender systems.

Methods. We introduce DeepSurv, a Cox proportional hazards deep neural network and state-of-the-art survival method for modeling interactions between a patient’s covariates and treatment effectiveness in order to provide personalized treatment recommendations.

Results. We perform a number of experiments training DeepSurv on simulated and real survival data. We demonstrate that DeepSurv performs as well as or better than other state-of-the-art survival models and validate that DeepSurv successfully models increasingly complex relationships between a patient’s covariates and their risk of failure. We then show how DeepSurv models the relationship between a patient’s features and effectiveness of different treatment options to show how DeepSurv can be used to provide individual treatment recommendations. Finally, we train DeepSurv on real clinical studies to demonstrate how it’s personalized treatment recommendations would increase the survival time of a set of patients.

Conclusions. The predictive and modeling capabilities of DeepSurv will enable medical researchers to use deep neural networks as a tool in their exploration, understanding, and prediction of the effects of a patient’s characteristics on their risk of failure.

Full text

Its licence does not allow us to show the full text. Read it at PubMed Central:

Read at PubMed Central

References 36

  1. 1.

    RW Y, EA S, DJ K, et al. Development and validation of a prediction rule for benefit and harm of dual antiplatelet therapy beyond 1 year after percutaneous coronary intervention. JAMA. 2016; 315(16):1735–49. https://doi.org/10.1001/jama.2016.3775.

    Not in our library10.1001/jama.2016.3775

  2. 2.
  3. 3.
  4. 4.

    Cheng W-Y, Yang T-HO, Anastassiou D. Development of a prognostic model for breast cancer survival in an open challenge environment. Sci Total Environ. 2013;5(181):181-5018150.

    Not in our library10.1126/scitranslmed.3005974

  5. 5.

    Cox DR. In: Kotz S, Johnson NL, (eds).Regression Models and Life-Tables. New York: Springer; 1992, pp. 527–41. 10.1007/978-1-4612-4380-9.

    Not in our library

  6. 6.

    Liestbl K, Andersen PK, Andersen U. Survival analysis and neural nets. Stat Med. 1994;13(12):1189-200.

    Not in our library10.1002/sim.4780131202

  7. 7.

    Street WN. A neural network model for prognostic prediction In: Kaufmann M, editor. Proceedings of the Fifteenth International Conference on Machine Learning. San Francisco: 1998. p. 540–46.

    Not in our library

  8. 8.

    Jerez JM, Franco L, Alba E, Llombart-Cussac A, Lluch A, Ribelles N, Munárriz B, Martín M. Improvement of breast cancer relapse prediction in high risk intervals using artificial neural networks. Breast Cancer Res Treat. 2005; 94(3):265–72. 10.1007/s10549-005-9013-y.

    Not in our library10.1007/s10549-005-9013-y

  9. 9.

    Biganzoli E, Boracchi P, Mariani L, Marubini E. Feed forward neural networks for the analysis of censored survival data: a partial logistic regression approach. Stat Med. 1998;17(10):1169-86.

    Not in our library10.1002/(sici)1097-0258(19980530)17:10<1169::aid-sim796>3.0.co;2-d

  10. 10.

    Faraggi D, Simon R. A neural network model for survival data. Stat Med. 1995;14(1):73-82.

    Not in our library10.1002/sim.4780140108

  11. 11.

    Sargent DJ. Comparison of artificial neural networks with other statistical approaches. Cancer. 2001;91(S8):1636-42.

    Not in our library10.1002/1097-0142(20010415)91:8+<1636::aid-cncr1176>3.0.co;2-d

  12. 12.

    Xiang A, Lapuerta P, Ryutov A, Buckley J, Azen S. Comparison of the performance of neural network methods and cox regression for censored survival data. Comput Stat Data Anal. 2000;34(2):243-57.

    Not in our library10.1016/s0167-9473(99)00098-5

  13. 13.

    Mariani L, Coradini D, Biganzoli E, Boracchi P, Marubini E, Pilotti S, Salvadori B, Silvestrini R, Veronesi U, Zucali R, et al. Prognostic factors for metachronous contralateral breast cancer: a comparison of the linear cox regression model and its artificial neural network extension. Breast Cancer Res Treat. 1997;44(2):167-78.

    Not in our library10.1023/a:1005765403093

  14. 14.

    Therneau T, Grambsch PM. Modeling Survival Data: Extending the Cox Model. 2000.

    Not in our library

  15. 15.

    Ishwaran H, Kogalur UB. Random survival forests for r. R News. 2007;7(2):25-31.

    Not in our library

  16. 16.

    Ishwaran H, Kogalur UB, Blackstone EH, Lauer MS. Random survival forests. Ann Appl Statist. 2008;2(3):841-60.

    Not in our library10.1214/08-aoas169

  17. 17.

    Ranganath R, Perotte A, Elhadad N, Blei D. Deep survival analysis In: Doshi-Velez F, Fackler J, Kale D, Wallace B, Weins J, editors. Proceedings of the 1st Machine Learning for Healthcare Conference. Proceedings of Machine Learning Research, vol 56. Northeastern University, Boston, MA, USA: PMLR: 2016. p. 101–14. http://proceedings.mlr.press/v56/Ranganath16.html.

    Not in our library

  18. 18.

    Srivastava N, Hinton G, Krizhevsky A, Sutskever I, Salakhutdinov R. Dropout: A simple way to prevent neural networks from overfitting. J Mach Learn Res. 2014;15(1):1929-58.

    Not in our library

  19. 19.

    Klambauer G, Unterthiner T, Mayr A, Hochreiter S. Self-normalizing neural networks. In: Advances in Neural Information Processing Systems: 2017. p. 972–81. arXiv preprint. 1706.02515.

    Not in our library

  20. 20.

    Kingma D, Ba J. Adam: A method for stochastic optimization. Proceedings of the 3rd International Conference on Learning Representations (ICLR 2015). 2015. arXiv preprint arXiv:1412.6980. https://dare.uva.nl/search?identifier=a20791d3-1aff-464a-8544-268383c33a75.

    Not in our library

  21. 21.

    Nesterov Y. Gradient methods for minimizing composite functions. Math Program. 2013;140(1):125-61.

    Not in our library10.1007/s10107-012-0629-5

  22. 22.

    Senior A, Heigold G, Ranzato M, Yang K. An empirical study of learning rates in deep neural networks for speech recognition. In: Acoustics, Speech and Signal Processing (ICASSP), 2013 IEEE International Conference on Acoustics, Speech and Signal Processing. IEEE: 2013. p. 6724–8.

    Not in our library

  23. 23.

    Bergstra J, Bengio Y. Random search for hyper-parameter optimization. J Mach Learn Res. 2012;13(1):281-305.

    Not in our library

  24. 24.

    Harrell FE, Lee KL, Califf RM, Pryor DB, Rosati RA. Regression modeling strategies for improved prognostic prediction. Stat Med. 1984;3(2):143-52.

    Not in our library10.1002/sim.4780030207

  25. 25.

    Efron B, Tibshirani RJ. An Introduction to the Bootstrap. 1993.

    Not in our library

  26. 26.

    Austin PC. Generating survival times to simulate cox proportional hazards models with time-varying covariates. Stat Med. 2012;31(29):3946-58.

    Not in our library10.1002/sim.5452

  27. 27.

    Hosmer DW, Lemeshow S, May S. Applied Survival Analysis: Regression Modeling of Time to Event Data. 2nd ed. 2008.

    Not in our library

  28. 28.

    Knaus WA, Harrell FE, Lynn J, Goldman L, Phillips RS, Connors AF, Dawson NV, Fulkerson WJ, Califf RM, Desbiens N, et al. The support prognostic model: objective estimates of survival for seriously ill hospitalized adults. Ann Intern Med. 1995;122(3):191-203.

    Not in our library10.7326/0003-4819-122-3-199502010-00007

  29. 29.

    Curtis C, Shah SP, Chin S-F, Turashvili G, Rueda OM, Dunning MJ, Speed D, Lynch AG, Samarajiwa S, Yuan Y, et al. The genomic and transcriptomic architecture of 2,000 breast tumours reveals novel subgroups. Nature. 2012;486(7403):346-52.

    Not in our library10.1038/nature10983

  30. 30.

    Lakhanpal R, Sestak I, Shadbolt B, Bennett GM, Brown M, Phillips T, Zhang Y, Bullman A, Rezo A. Ihc4 score plus clinical treatment score predicts locoregional recurrence in early breast cancer. The Breast. 2016;29:147-52.

    Not in our library10.1016/j.breast.2016.06.019

  31. 31.

    Foekens JA, Peters HA, Look MP, Portengen H, Schmitt M, Kramer MD, Brünner N, Jänicke F, Meijer-van Gelder ME, Henzen-Logmans SC, et al. The urokinase system of plasminogen activation and prognosis in 2780 breast cancer patients. Cancer Res. 2000;60(3):636-43.

    Not in our library

  32. 32.

    Schumacher M, Bastert G, Bojar H, Huebner K, Olschewski M, Sauerbrei W, Schmoor C, Beyerle C, Neumann R, Rauschecker H. Randomized 2 x 2 trial evaluating hormonal treatment and the duration of chemotherapy in node-positive breast cancer patients. german breast cancer study group. J Clin Oncol. 1994;12(10):2086-93.

    Not in our library10.1200/jco.1994.12.10.2086

  33. 33.

    Altman DG, Royston P. What do we mean by validating a prognostic model? Stat Med. 2000;19(4):453-73.

    Not in our library10.1002/(sici)1097-0258(20000229)19:4<453::aid-sim350>3.0.co;2-5

  34. 34.

    Katzman JL. DeepSurv. GitHub. 2017. 10.5281/zenodo.1134133. https://github.com/jaredleekatzman/DeepSurv.

    Not in our library

  35. 35.

    Sobol IM. Uniformly distributed sequences with an additional uniform property. USSR Comput Math Math Phys. 1976;16(5):236-42.

    Not in our library10.1016/0041-5553(76)90154-3

  36. 36.

    Fox BL. Algorithm 647: Implementation and relative efficiency of quasirandom sequence generators. ACM Trans Math Softw. 1986; 12(4):362–76. 10.1145/22721.356187.

    Not in our library

Cited by papers we hold 18

Reads

Read by people on this page

None yet.

Checked or read by AI through naulon

None yet.

A person counts once a day per paper. We keep no cookie and no address: a reader is a hash that changes every day. Any month under five reads shows no number.

Are you an author?

Is this your paper?

Claim it to add your note and be paid when agents cite it. Free.