Diffusion-weighted imaging-based radiomics model using automatic machine learning to differentiate cerebral cystic metastases from brain abscesses

Abstract

Abstract Objectives To develop a radiomics model based on diffusion-weighted imaging (DWI) utilizing automated machine learning method to differentiate cerebral cystic metastases from brain abscesses. Materials and methods A total of 186 patients with cerebral cystic metastases (n = 98) and brain abscesses (n = 88) from two clinical institutions were retrospectively included. The datasets (129 from institution A) were randomly portioned into separate 75% training and 25% internal testing sets. Radiomics features were extracted from DWI images using two subregions of the lesion (cystic core and solid wall). A thorough image preprocessing method was applied to DWI images to ensure the robustness of radiomics features before feature extraction. Then the Tree-based Pipeline Optimization Tool (TPOT) was utilized to search for the best optimized machine learning pipeline, using a fivefold cross-validation in the training set. The external test set (57 from institution B) was used to evaluate the model’s performance. Results Seven distinct TPOT models were optimized to distinguish between cerebral cystic metastases and abscesses either based on different features combination or using wavelet transform. The optimal model demonstrated an AUC of 1.00, an accuracy of 0.97, sensitivity of 1.00, and specificity of 0.93 in the internal test set, based on the combination of cystic core and solid wall radiomics signature using wavelet transform. In the external test set, this model reached 1.00 AUC, 0.96 accuracy, 1.00 sensitivity, and 0.93 specificity. Conclusion The DWI-based radiomics model established by TPOT exhibits a promising predictive capacity in distinguishing cerebral cystic metastases from abscesses.