CS U-NET: A Medical Image Segmentation Method Integrating Spatial and Contextual Attention Mechanisms Based on U-NET

被引:0
|
作者
Zhang, Fanyang [1 ]
Fan, Zhang [1 ]
机构
[1] Shanghai Univ Engn Sci, Lab Intelligent Control & Robot, Shanghai, Peoples R China
关键词
CBAM; deep learning; medical image segmentation; Swin transformer; U-net;
D O I
10.1002/ima.70072
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Medical image segmentation is a crucial process in medical image analysis, with convolutional neural network (CNN)-based methods achieving notable success in recent years. Among these, U-Net has gained widespread use due to its simple yet effective architecture. However, CNNs still struggle to capture global, long-range semantic information. To address this limitation, we present CS U-NET, a novel method built upon Swin-U-Net, which integrates spatial and contextual attention mechanisms. This hybrid approach combines the strengths of both transformers and U-Net architectures to enhance segmentation performance. In this framework, tokenized image patches are processed through a transformer-based U-shaped encoder-decoder, enabling the learning of both local and global semantic features via skip connections. Our method achieves a Dice Similarity Coefficient of 78.64% and a 95% Hausdorff distance of 21.25 on the Synapse multiorgan segmentation dataset, outperforming Trans-U-Net and other state-of-the-art U-Net variants by 4% and 6%, respectively. The experimental results highlight the significant improvements in prediction accuracy and edge detail preservation provided by our approach.
引用
收藏
页数:10
相关论文
共 50 条
  • [31] Medical Image Segmentation Based on 3D U-net
    Chen, Silu
    Hu, Guanghao
    Sun, Jun
    2020 19TH INTERNATIONAL SYMPOSIUM ON DISTRIBUTED COMPUTING AND APPLICATIONS FOR BUSINESS ENGINEERING AND SCIENCE (DCABES 2020), 2020, : 130 - 133
  • [32] Review of Medical Image Segmentation Algorithms Based on U-Net Variants
    Ke, Cui
    Qichuan, Tian
    Lu, Lian
    Computer Engineering and Applications, 2024, 60 (11) : 32 - 49
  • [33] AttResDU-Net: Medical Image Segmentation Using Attention-based Residual Double U-Net
    Khan, Akib Mohammed
    Ashrafee, Alif
    Khan, Fahim Shahriar
    Hasan, Md. Bakhtiar
    Kabir, Md. Hasanul
    2023 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, IJCNN, 2023,
  • [34] MAEF-Net: MLP Attention for Feature Enhancement in U-Net based Medical Image Segmentation Networks
    Zhang, Yunchu
    Dong, Jianfei
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2024, 28 (02) : 846 - 857
  • [35] Segmentation of skin lesions image based on U-Net + +
    Chen Zhao
    Renjun Shuai
    Li Ma
    Wenjia Liu
    Menglin Wu
    Multimedia Tools and Applications, 2022, 81 : 8691 - 8717
  • [36] Cardiac Image Segmentation Based on Improved U-Net
    Qiao, Guang Xiao
    Song, Ji Hong
    2022 INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, COMPUTER VISION AND MACHINE LEARNING (ICICML), 2022, : 133 - 137
  • [37] MFA U-Net: a U-Net like multi-stage feature analysis network for medical image segmentation
    Wang, Yupeng
    Wang, Suyu
    He, Jian
    PATTERN ANALYSIS AND APPLICATIONS, 2024, 27 (04)
  • [38] Improved U-Net Fundus Image Segmentation Algorithm Integrating Effective Channel Attention
    Liang, Junhua
    Ding, Lihua
    Tong, Xuming
    Zhao, Zhisheng
    Li, Jie
    Liang, Junqiang
    Dong, Beibei
    Yuan, Yanhong
    JOURNAL OF IMAGING SCIENCE AND TECHNOLOGY, 2022, 66 (04)
  • [39] ELU-Net: An Efficient and Lightweight U-Net for Medical Image Segmentation
    Deng, Yunjiao
    Hou, Yulei
    Yan, Jiangtao
    Zeng, Daxing
    IEEE Access, 2022, 10 : 35932 - 35941
  • [40] ELU-Net: An Efficient and Lightweight U-Net for Medical Image Segmentation
    Deng, Yunjiao
    Hou, Yulei
    Yan, Jiangtao
    Zeng, Daxing
    IEEE ACCESS, 2022, 10 : 35932 - 35941