Machine Learning Model Development for Malignant Prostate Lesion Prediction Using Texture Analysis Features from Ultrasound Shear-Wave Elastography

Introduction: Artificial intelligence (AI) is increasingly utilized for texture analysis and the development of machine learning (ML) techniques to enhance diagnostic accuracy. ML algorithms are trained to differentiate between normal and malignant conditions based on provided data. Texture feature analysis, including first-order and second-order features, is a critical step in ML development. This study aimed to evaluate quantitative texture features of normal and prostate cancer tissues identified through ultrasound B-mode and shear-wave elastography (SWE) imaging and to develop and assess ML models for predicting and classifying normal versus malignant prostate tissues.
Methodology: First-order and second-order texture features were extracted from B-mode and SWE imaging, including four reconstructed regions of interest (ROIs) from SWE images for normal and malignant tissues. A total of 94 texture features were derived, including features for intensity, Gray-Level Co-Occurrence Matrix (GLCM), Gray-Level Dependence Length Matrix (GLDLM), Gray-Level Run Length Matrix (GLRLM), and Gray-Level Size Zone Matrix (GLSZM). Five ML models were developed and evaluated using 5-fold cross-validation to predict normal and malignant tissues.
Results: Data from 62 patients were analyzed. All ROIs, except those derived from B-mode imaging, exhibited statistically significant differences in features between normal and malignant tissues. Among the developed models, Support Vector Machines (SVM), Random Forest (RF), and Naive Bayes (NB) demonstrated the highest performance across all ROIs. These models consistently achieved strong predictive accuracy for classifying normal versus malignant tissues. Gray Pure SWE and Gray Reconstructed images Provided the highest sensitivity and specificity in PCa prediction by 82%, 90%, and 98%, 96%, respectively.
Conclusions: Texture analysis with machine learning on SWE-US and reconstructed images effectively differentiates malignant from benign prostate lesions, with features like contrast, entropy, and correlation playing a key role. Random Forest, SVM, and Naïve Bayes showed the highest classification performance, while grayscale reconstructions (GPSWE and GRRI) enhanced detection accuracy.