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.

Point-Sampling Method Based on 3D U-Net Architecture to Reduce the Influence of False Positive and Solve Boundary Blur Problem in 3D CT Image Segmentation

Interactive Liver Segmentation in CT Volumes Using Fully Convolutional Networks.

3D Multiple-Contextual ROI-Attention Network for Efficient and Accurate Volumetric Medical Image Segmentation.

Render U-Net: A Unique Perspective on Render to Explore Accurate Medical Image Segmentation

Automatic CT Segmentation from Bounding Box Annotations using Convolutional Neural Networks

Multi-Scale Supervised 3D U-Net for Kidneys and Kidney Tumor Segmentation

3D RoI-aware U-Net for Accurate and Efficient Colorectal Tumor Segmentation

3D Medical Image Segmentation based on multi-scale MPU-Net

Cross-layer Connected Network with Adaptive Attention Mechanism for 3D Multi-Organ and Tumor Segmentations from CT

A Novel Hybrid Convolutional Neural Network for Accurate Organ Segmentation in 3D Head and Neck CT Images

Three-Dimensional Liver Image Segmentation Using Generative Adversarial Networks Based on Feature Restoration

Surface-GCN: Learning Interaction Experience for Organ Segmentation in 3D Medical Images.

Segmentation of the prostate and organs at risk in male pelvic CT images using deep learning

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

Multi Scale Supervised 3D U-Net for Kidney and Tumor Segmentation

RMAU-Net: Residual Multi-Scale Attention U-Net For liver and tumor segmentation in CT images

3-D RoI-Aware U-Net for Accurate and Efficient Colorectal Tumor Segmentation

CT-based Multi-Organ Segmentation Using a 3D Self-Attention U-net Network for Pancreatic Radiotherapy

Large-Kernel Attention for 3D Medical Image Segmentation

Multi-organ segmentation network for abdominal CT images based on spatial attention and deformable convolution

CT Male Pelvic Organ Segmentation Using Fully Convolutional Networks with Boundary Sensitive Representation