Abstract:Background and objective: Segmentation of rectal cancerous regions using 2D Magnetic Resonance Imaging (MRI) images is a critical step in radiation therapy. The shape of rectal cancer has significant variations and the shape of some surrounding organs is similar to that of rectal cancer; these conditions significantly affect the segmentation accuracy of rectal cancer and lead to incorrect segmentation. Therefore, automatic segmentation of rectal cancer is urgently needed, and it is a great challenge. For this task, the existing deep learning-based approaches have two shortcomings: 1) The U-Net network plays an important role in the field of medical segmentation. However, the designs of encoders and decoders in traditional U-Net networks are relatively simple and cannot extract good features, resulting in incorrect segmentation results. 2) Conventional neural networks extract high-level features that often do not include sufficient high-resolution contour information, resulting in ambiguity in contour segmentation. In this paper, we propose an improved U-Net network based on contour prediction, aiming at effective segmentation of rectal cancer. Methods: We designed a new U-Net network by improving the traditional U-Net network. We made four improvements: 1) We replaced the encoders with the SENet network. 2) A global pooling layer was added after the last encoder. 3) We added the Spatial and Channel Squeeze & Excitation (SCSE) attention mechanism module to each decoder. 4) We concatenated the output results of each decoder. In addition, the model implemented content segmentation and contour segmentation for rectal cancer in parallel, so that both the content and contour information was learned by the network to enhance the segmentation accuracy. Results: Our data were obtained from the Shanxi Provincial Cancer Hospital and included 3773 2D MRI rectal cancer images. The proposed method achieved an Mean Intersection over Union of 0.894 (MIoU) on the test set. Compared with state-of-the-art methods, our method had the best performance on the test set, and its MIoU metric was 0.123 higher than that of the second-best model. At the same time, the effectiveness of the improvements to our method was demonstrated through ablation experiments. Conclusions: Our method can help radiologists to segment effectively, save their time and energy, and enable them to focus on cases that are not easily segmented because of the complex shape of rectal cancer.

Residual UNet with spatial and channel attention for automatic magnetic resonance image segmentation of rectal cancer

Improved UNet‐based Magnetic Resonance Imaging Segmentation of Demyelinating Diseases with Small Lesion Regions

Improved U-Net based on contour prediction for efficient segmentation of rectal cancer

Automatic segmentation of rectal tumors from MRI using multiscale densely connected convolutional neural network based on attention mechanism

Segmentation of brain tumor MRI image based on improved attention module Unet network

Brain tumor magnetic resonance image segmentation by a multiscale contextual attention module combined with a deep residual UNet (MCA-ResUNet)

Imaging segmentation mechanism for rectal tumors using improved U-Net

Residual-Attention UNet++: A Nested Residual-Attention U-Net for Medical Image Segmentation

Improved MRI Brain Tumor Segmentation Method Based on UNet++

DPA‐UNet rectal cancer image segmentation based on visual attention

MRI Image Segmentation of Brain Tumor Based on Res-UNet

Attention Augmented ConvNeXt UNet For Rectal Tumour Segmentation

CA-Res2UNet++: a Deep Residual UNet-based Method for Brain Tumor Segmentation in Multimodal MRI

DA-ResUNet: a Novel Method for Brain Tumor Segmentation Based on U-Net with Residual Block and CBAM

Image Segmentation of Rectal Tumor Based on Improved U-Net Model with Deep Learning.

Research on the magnetic resonance imaging brain tumor segmentation algorithm based on DO‐UNet

Covariance Self-Attention Dual Path UNet for Rectal Tumor Segmentation

Joint Dense Residual and Recurrent Attention Network for DCE-MRI Breast Tumor Segmentation

mResU-Net: multi-scale residual U-Net-based brain tumor segmentation from multimodal MRI

Ultrasound image segmentation of renal tumors based on UNet++ with fusion of multiscale residuals and dual attention

Segmentation of rectal tumor from multi-parametric MRI images using an attention-based fusion network