A deep learning-based automatic staging method for early endometrial cancer on MRI images

Abstract

Early treatment increases the 5-year survival rate of patients with endometrial cancer (EC). Deep learning (DL) as a new computer-aided diagnosis method has been widely used in medical image processing which can reduce the misdiagnosis by radiologists. An automatic staging method based on DL for the early diagnosis of EC will benefit both radiologists and patients. To develop an effective and automatic prediction model for early EC diagnosis on magnetic resonance imaging (MRI) images, we retrospectively enrolled 117 patients (73 of stage IA, 44 of stage IB) with a pathological diagnosis of early EC confirmed by postoperative biopsy at our institution from 1 January 2018, to 31 December 2020. Axial T2-weighted image (T2WI), axial diffusion-weighted image (DWI) and sagittal T2WI images from 117 patients have been classified into stage IA and stage IB according to the patient’s pathological diagnosis. Firstly, a semantic segmentation model based on the U-net network is trained to segment the uterine region and the tumor region on the MRI images. Then, the area ratio of the tumor region to the uterine region (TUR) in the segmentation map is calculated. Finally, the receiver operating characteristic curves (ROCs) are plotted by the TUR and the results of the patient’s pathological diagnosis in the test set to find the optimal staging thresholds for stage IA and stage IB. In the test sets, the trained semantic segmentation model yields the average Dice similarity coefficients of uterus and tumor on axial T2WI, axial DWI, and sagittal T2WI were 0.958 and 0.917, 0.956 and 0.941, 0.972 and 0.910 respectively. With pathological diagnostic results as the gold standard, the classification model on axial T2WI, axial DWI, and sagittal T2WI yielded an area under the curve (AUC) of 0.86, 0.85 and 0.94, respectively. In this study, an automatic DL-based segmentation model combining the ROC analysis of TUR on MRI images presents an effective early EC staging method.