Automatic cervical tumors segmentation in PET/MRI by parallel encoder U-net.
Authors
Affiliations (3)
Affiliations (3)
- Department of Biomedical Engineering, School of Medical Technology, Beijing Institute of Technology, Haidian District, Beijing, 100081, China.
- Department Radiology, Shengjing Hospital of China Medical University, Tiexi District, Shenyang, 110004, Liaoning, China.
- Department of Biomedical Engineering, School of Medical Technology, Beijing Institute of Technology, Haidian District, Beijing, 100081, China. [email protected].
Abstract
Automatic segmentation of cervical tumors is important in quantitative analysis and radiotherapy planning. A parallel encoder U-Net (PEU-Net) integrating the multi-modality information of PET/MRI was proposed to segment cervical tumor, which consisted of two parallel encoders with the same structure for PET and MR images. The features of the two modalities were extracted separately and fused at each layer of the decoder. Res2Net module on skip connection aggregated the features of various scales and refined the segmentation performance. PET/MRI images of 165 patients with cervical cancer were included in this study. U-Net, TransUNet, and nnU-Net with single or multi-modality (PET or/and T2WI) input were used for comparison. The Dice similarity coefficient (DSC) with volume data, DSC and the 95th percentile of Hausdorff distance (HD95) with tumor slices were calculated to evaluate the performance. The proposed PEU-Net exhibited the best performance (DSC<sub>3d</sub>: 0.726 ± 0.204, HD<sub>95</sub>: 4.603 ± 4.579 mm), DSC<sub>2d</sub> (0.871 ± 0.113) was comparable to the best result of TransUNet with PET/MRI (0.873 ± 0.125). The networks with multi-modality input outperformed those with single-modality images as input. The results showed that the proposed PEU-Net could use multi-modality information more effectively through the redesigned structure and achieved competitive performance.