A Machine Learning Approach to Predict an Early Biochemical Recurrence after a Radical Prostatectomy

Abstract

Background: Approximately 20–50% of prostate cancer patients experience biochemical recurrence (BCR) after radical prostatectomy (RP). Among them, cancer recurrence occurs in about 20–30%. Thus, we aim to reveal the utility of machine learning algorithms for the prediction of early BCR after RP. Methods: A total of 104 prostate cancer patients who underwent magnetic resonance imaging and RP were evaluated. Four well-known machine learning algorithms (i.e., k-nearest neighbors (KNN), multilayer perceptron (MLP), decision tree (DT), and auto-encoder) were applied to build a prediction model for early BCR using preoperative clinical and imaging and postoperative pathologic data. The sensitivity, specificity, and accuracy for detection of early BCR of each algorithm were evaluated. Area under the receiver operating characteristics (AUROC) analyses were conducted. Results: A prediction model using an auto-encoder showed the highest prediction ability of early BCR after RP using all data as input (AUC = 0.638) and only preoperative clinical and imaging data (AUC = 0.656), followed by MLP (AUC = 0.607 and 0.598), KNN (AUC = 0.596 and 0.571), and DT (AUC = 0.534 and 0.495). Conclusion: The auto-encoder-based prediction system has the potential for accurate detection of early BCR and could be useful for long-term follow-up planning in prostate cancer patients after RP.