Diagnosis of prostate cancer in a Chinese population by using machine learning methods

Abstract

An early diagnosis of prostate cancer (PC) is key for the successful treatment. Although invasive prostate biopsies can provide a definitive diagnosis, the number of biopsies should be reduced to avoid side effects and risks especially for the men with the low risk of cancer. Therefore, an accurate model is in need to predict PC with the aim of reducing unnecessary biopsies. In this study, we developed predictive models using four machine learning methods including Support Vector Machine (SVM), Least Squares Support Vector Machine (LS-SVM), Artificial Neural Network (ANN) and Random Forest (RF) to detect PC cases using available prebiopsy information. The models were constructed and evaluated on a cohort of 1625 Chinese men with prostate biopsies from Hong Kong hospital. All the models have the excellent performances in detecting significant PC cases, with ANN achieving the highest accuracy of 0.9527 and the AUC value of 0.9755. RF outperformed the other three methods in classifying benign, significant and insignificant PC cases, with an accuracy of 0.9741 and a F1 score of 0.8290.