Machine learning-based radiomics for histological classification of parotid tumors using morphological MRI: a comparative study

Abstract

Objectives: To evaluate the effectiveness of machine learning models based on morphological magnetic resonance imaging (MRI) radiomics in the classification of parotid tumors.

Methods: In total, 298 patients with parotid tumors were randomly assigned to a training and test set at a ratio of 7:3. Radiomics features were extracted from the morphological MRI images and screened using the Select K Best and LASSO algorithm. Three-step machine learning models with XGBoost, SVM, and DT algorithms were developed to classify the parotid neoplasms into four subtypes. The ROC curve was used to measure the performance in each step. Diagnostic confusion matrices of these models were calculated for the test cohort and compared with those of the radiologists.

Results: Six, twelve, and eight optimal features were selected in each step of the three-step process, respectively. XGBoost produced the highest area under the curve (AUC) for all three steps in the training cohort (0.857, 0.882, and 0.908, respectively), and for the first step in the test cohort (0.826), but produced slightly lower AUCs than SVM in the latter two steps in the test cohort (0.817 vs. 0.833, and 0.789 vs. 0.821, respectively). The total accuracies of XGBoost and SVM in the confusion matrices (70.8% and 59.6%) outperformed those of DT and the radiologist (46.1% and 49.2%).

Conclusion: This study demonstrated that machine learning models based on morphological MRI radiomics might be an assistive tool for parotid tumor classification, especially for preliminary screening in absence of more advanced scanning sequences, such as DWI.

Key points: • Machine learning algorithms combined with morphological MRI radiomics could be useful in the preliminary classification of parotid tumors. • XGBoost algorithm performed better than SVM and DT in subtype differentiation of parotid tumors, while DT seemed to have a poor validation performance. • Using morphological MRI only, the XGBoost and SVM algorithms outperformed radiologists in the four-type classification task for parotid tumors, thus making these models a useful assistant diagnostic tool in clinical practice.

Keywords: Algorithms; Diagnosis; Machine learning; Magnetic resonance imaging; Parotid tumor.