1. MSGSE-Net: Multi-scale guided squeeze-and-excitation network for subcortical brain structure segmentation
- Author
-
Shidi Fu, Xiang Li, Lin Wang, Ying Wei, and Chuyuan Wang
- Subjects
Computer science ,business.industry ,Cognitive Neuroscience ,Pattern recognition ,Image segmentation ,computer.software_genre ,Convolutional neural network ,Computer Science Applications ,Discriminative model ,Artificial Intelligence ,Feature (computer vision) ,Voxel ,Benchmark (computing) ,Segmentation ,Artificial intelligence ,Focus (optics) ,business ,computer - Abstract
Convolutional neural networks (CNNs) have been achieving remarkable results in medical image segmentation. However, for accurate segmentation of subcortical brain structure in MR images, it is still a challenge due to the ambiguous boundaries, the complex structures, and the various shapes, which limits their clinical application. In this paper, we focus on utilizing multi-scale image contexts and attention mechanisms to improve networks’ ability to learn discriminative feature representation for accurate segmentation and present a novel FCNN architecture called multis-scale guided squeeze-and-excitation network (MSGSE-Net). In particular, we first propose the multi-scale guided squeeze-and-excitation (MSGSE) attention module which can progressively and selectively aggregate discriminative features. In contrast to existing attention modules, the MSGSE module performs an adaptive recalibration that features at different locations of the feature map are recalibrated under the guidance of multi-scale contexts. Then multi-scale spatial attention supervision is adopted to enhance the intra-class homogeneity and inter-class distinction of the attention weights. Moreover, we propose a novel entropy-weighted Dice loss (EDL) to force the network to focus on the ambiguous voxels around the boundaries of subcortical structures. We evaluate the proposed method on two challenging benchmark datasets (the IBSR dataset and the MALC dataset). The experimental results show that our model consistently yields better segmentation performance than several state-of-the-art methods and improves the segmentation Dice score by 1.6% at most compared with baseline method U-Net. Our code is available at https://github.com/neulxlx/MSGSE-Net .
- Published
- 2021
- Full Text
- View/download PDF