Long short-term memory artificial neural network model for prediction of prostate cancer survival outcomes according to initial treatment strategy: development of an online decision-making support system
The delivery of precision medicine is a primary objective for both clinical and translational investigators. Patients with newly diagnosed prostate cancer (PCa) face the challenge of deciding among multiple initial treatment modalities. The purpose of this study is to utilize artificial neural network (ANN) modeling to predict survival outcomes according to initial treatment modality and to develop an online decision-making support system.
Data were collected retrospectively from 7267 patients diagnosed with PCa between January 1988 and December 2017. The analyses included 19 pretreatment clinicopathological covariates. Multilayer perceptron (MLP), MLP for N-year survival prediction (MLP-N), and long short-term memory (LSTM) ANN models were used to analyze progression to castration-resistant PCa (CRPC)-free survival, cancer-specific survival (CSS), and overall survival (OS), according to initial treatment modality. The performances of the ANN and the Cox-proportional hazards regression models were compared using Harrell’s C-index.
The ANN models provided higher predictive power for 5- and 10-year progression to CRPC-free survival, CSS, and OS compared to the Cox-proportional hazards regression model. The LSTM model achieved the highest predictive power, followed by the MLP-N, and MLP models. We developed an online decision-making support system based on the LSTM model to provide individualized survival outcomes at 5 and 10 years, according to the initial treatment strategy.
The LSTM ANN model may provide individualized survival outcomes of PCa according to initial treatment strategy. Our online decision-making support system can be utilized by patients and health-care providers to determine the optimal initial treatment modality and to guide survival predictions.
KeywordsArtificial intelligence Decision support techniques Prostate cancer Survival
Artificial neural network
Area under the curve
Castration-resistant prostate cancer
Long short-term memory
MLP for N-year survival prediction
This study was supported through a Young Researcher Program Grant from the National Research Foundation of Korea (NRF-2017R1C1B5017516).
Protocol/project development: Koo, KS Lee, Han, Rha, Hong, Yang and Chung. Data collection and management: Koo, KS Lee, YH Lee, and GR Min. Data analysis: Kim and C Min. Manuscript writing/editing: Koo and Chung.
Compliance with ethical standards
Conflict of interest
All of the authors declare that they have no conflicts of interest to declare.
All procedures performed in studies involving human participants were in accordance with the ethical standards of the institutional and/or national research committee and with the 1964 Helsinki Declaration and its later amendments or comparable ethical standards.
Informed consent was not required for the purposes of this study as it was based upon retrospective anonymous patient data and did not involve patient intervention or the use of human tissue samples.
- 1.Koo KC, Cho JS, Bang WJ et al (2018) Cancer-specific mortality among Korean men with localized or locally advanced prostate cancer treated with radical prostatectomy versus radiotherapy: a multi-center study using propensity scoring and competing risk regression analyses. Cancer Res Treat 50:129–137CrossRefGoogle Scholar
- 14.Chi CL, Street WN, Wolberg WH (2007) Application of artificial neural network-based survival analysis on two breast cancer datasets. AMIA Annu Symp Proc 130–134.Google Scholar
- 15.Rumelhart D, Hinton G, Williams R et al (1986) Learning internal representations by error propagation. In: Rumelhart DE, MCclelland JL (eds) Parallel distributed processing 1986. MIT Press, Cambridge, pp 318–362Google Scholar