Multiscale fused network with additive channel–spatial attention for image segmentation

作者:

Highlights:

摘要

Class imbalance is a challenge in image segmentation. Deep learning has demonstrated potential advantages as a promising approach in solving such challenging problems because of its powerful ability to extract features. To treat class imbalance in medical image segmentation effectively, in this study, deep learning technology is employed. Further, a new approach, referred to as multiscale fused network with additive channel–spatial attention (MSF-ACSA), is developed. In particular, using the U-shape convolutional network as the structure of the main body network, an additive channel–spatial attention (ACSA) module, which exploits high-level features to guide low-level features for optimal responses, is designed. Therefore, the proposed MSF-ACSA can availably focus on regions of interest while adaptively recalibrating the channel significance of feature maps. Simultaneously, a multiscale fusion with deep supervision module is introduced to incorporate fine-grained details with coarse-grained semantics from diverse scale features. Furthermore, a focal weighted Tversky loss function is proposed to mitigate class imbalance issues. Numerous experimental results indicate that the proposed MSF-ACSA achieves the best performance when compared with state-of-the-art approaches for medical image segmentation.

论文关键词:Deep learning,Class imbalance,Semantic segmentation,Attention mechanism,Multiscale fusion

论文评审过程:Received 19 September 2020, Revised 18 December 2020, Accepted 3 January 2021, Available online 7 January 2021, Version of Record 8 January 2021.

论文官网地址:https://doi.org/10.1016/j.knosys.2021.106754