Ensemble learning-based radiomics with multi-sequence magnetic resonance imaging for benign and malignant soft tissue tumor differentiation

Abstract

Many previous studies focused on differentiating between benign and malignant soft tissue tumors using radiomics model based on various magnetic resonance imaging (MRI) sequences, but it is still unclear how to set up the input radiomic features from multiple MRI sequences. Here, we evaluated two types of radiomics models generated using different feature incorporation strategies. In order to differentiate between benign and malignant soft tissue tumors (STTs), we compared the diagnostic performance of an ensemble of random forest (R) models with single-sequence MRI inputs to R models with pooled multi-sequence MRI inputs. One-hundred twenty-five STT patients with preoperative MRI were retrospectively included and consisted of training (n = 100) and test (n = 25) sets. MRI included T1-weighted (T1-WI), T2-weighted (T2-WI), contrast-enhanced (CE)-T1-WI, diffusion-weighted images (DWIs, b = 800 sec/mm2) and apparent diffusion coefficient (ADC) maps. After tumor segmentation on each sequence, 100 original radiomic features were extracted from each sequence image and divided into three-feature sets: T features from T1- and T2-WI, CE features from CE-T1-WI, and D features from DWI and ADC maps. Four radiomics models were built using Lasso and R with four combinations of three-feature sets as inputs: T features (R-T), T+CE features (R-C), T+D features (R-D), and T+CE+D features (R-A) (Type-1 model). An ensemble model was built by soft voting of five, single-sequence-based R models (Type-2 model). AUC, sensitivity, specificity, and accuracy of each model was calculated with five-fold cross validation. In Type-1 model, AUC, sensitivity, specificity, and accuracy were 0.752, 71.8%, 61.1%, and 67.2% in R-T; 0.756, 76.1%, 70.4%, and 73.6% in R-C; 0.750, 77.5%, 63.0%, and 71.2% in R-D; and 0.749, 74.6%, 61.1%, and 68.8% R-A models, respectively. AUC, sensitivity, specificity, and accuracy of Type-2 model were 0.774, 76.1%, 68.5%, and 72.8%. In conclusion, an ensemble method is beneficial to incorporate features from multi-sequence MRI and showed diagnostic robustness for differentiating malignant STTs.