MRI radiomics: A machine learning approach for the risk stratification of endometrial cancer patients

Abstract

PurposeTo investigate radiomics and machine learning (ML) as possible tools to enhance MRI-based risk stratification in patients with endometrial cancer (EC). MethodFrom two institutions, 133 patients (Institution1 = 104 and Institution2 = 29) with EC and pre-operative MRI were retrospectively enrolled and divided in two a low-risk and a high-risk group according to EC stage and grade. T2-weighted (T2w) images were three-dimensionally annotated to obtain volumes of interest of the entire tumor. A PyRadiomics based and previously validated pipeline was used to extract radiomics features and perform feature selection. In particular, feature stability, variance and pairwise correlation were analyzed. Then, the least absolute shrinkage and selection operator technique and recursive feature elimination were used to obtain the final feature set. The performance of a Support Vector Machine (SVM) algorithm was assessed on the dataset from Institution 1 via 2-fold cross-validation. Then, the model was trained on the entire Institution 1 dataset and tested on the external test set from Institution 2. ResultsIn total, 1197 radiomics features were extracted. After the exclusion of unstable, low variance and intercorrelated features least absolute shrinkage and selection operator and recursive feature elimination identified 4 features that were used to build the predictive ML model. It obtained an accuracy of 0.71 and 0.72 in the train and test sets respectively. ConclusionsWhole-lesion T2w-derived radiomics showed encouraging results and good generalizability for the identification of low-risk EC patients.