MRI Radiomics-Based Machine Learning Models for Ki67 Expression and Gleason Grade Group Prediction in Prostate Cancer

Abstract

Purpose: The Ki67 index and the Gleason grade group (GGG) are vital prognostic indicators of prostate cancer (PCa). This study investigated the value of biparametric magnetic resonance imaging (bpMRI) radiomics feature-based machine learning (ML) models in predicting the Ki67 index and GGG of PCa. Methods: A total of 122 patients with pathologically proven PCa who had undergone preoperative MRI were retrospectively included. Radiomics features were extracted from T2-weighted imaging (T2WI), diffusion-weighted imaging (DWI), and apparent diffusion coefficient (ADC) maps. Then, recursive feature elimination (RFE) was applied to remove redundant features. ML models for predicting Ki67 expression and GGG were constructed based on bpMRI and different algorithms, including logistic regression (LR), support vector machine (SVM), random forest (RF), and K-nearest neighbor (KNN). The performances of different models were evaluated with receiver operating characteristic (ROC) analysis. In addition, a joint analysis of Ki67 expression and GGG was performed by assessing their Spearman correlation and calculating the diagnostic accuracy for both indices. Results: The ML model based on LR and ADC + T2 (LR_ADC + T2, AUC = 0.8882) performed best in predicting Ki67 expression, and ADC_wavelet-LHH_firstorder_Maximum had the highest feature weighting. The SVM_DWI + T2 (AUC = 0.9248) performed best in predicting GGG, and DWI_wavelet HLL_glcm_SumAverage had the highest feature weighting. The Ki67 and GGG exhibited a weak positive correlation (r = 0.382, p < 0.001), and LR_ADC + DWI had the highest diagnostic accuracy in predicting both (0.6230). Conclusion: The proposed ML models are suitable for predicting both Ki67 expression and GGG in PCa. This algorithm could be used to identify indolent or invasive PCa with a noninvasive, repeatable, and accurate diagnostic method.