SCA-Former: transformer-like network based on stream-cross attention for medical image segmentation.

Journal: Physics in medicine and biology
Published Date:

Abstract

. Deep convolutional neural networks (CNNs) have been widely applied in medical image analysis and achieved satisfactory performances. While most CNN-based methods exhibit strong feature representation capabilities, they face challenges in encoding long-range interaction information due to the limited receptive fields. Recently, the Transformer has been proposed to alleviate this issue, but its cost is greatly enlarging the model size, which may inhibit its promotion.. To take strong long-range interaction modeling ability and small model size into account simultaneously, we propose a Transformer-like block-based U-shaped network for medical image segmentation, dubbed as SCA-Former. Furthermore, we propose a novel stream-cross attention (SCA) module to enforce the network to focus on finding a balance between local and global representations by extracting multi-scale and interactive features along spatial and channel dimensions. And SCA can effectively extract channel, multi-scale spatial, and long-range information for a more comprehensive feature representation.. Experimental results demonstrate that SCA-Former outperforms the current state-of-the-art (SOTA) methods on three public datasets, including GLAS, ISIC 2017 and LUNG.. This work exhibits a promising method to enhance the feature representation of convolutional neural networks and improve segmentation performance.

Authors

  • Chengrui Gao
    College of Computer Science, Sichuan University, Chengdu 610065, China.
  • Junlong Cheng
    College of Information Science and Engineering, Xinjiang University, Urumqi 830000, China; Key Laboratory of software engineering technology, Xinjiang University, China.
  • Ziyuan Yang
    School of Information Engineering, Nanchang University, Nanchang 330031, China.
  • Yingyu Chen
    College of Veterinary Medicine, Wuhan, China.
  • Min Zhu
    Department of Infectious Diseases, Affiliated Taizhou Hospital of Wenzhou Medical University, No.50 Ximeng Road, Taizhou, 317000, China.