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
相关论文
共 25 条
[21]   Sub-convolutional U-Net with transformer attention network for end-to-end single-channel speech enhancement [J].
Yecchuri, Sivaramakrishna ;
Vanambathina, Sunny Dayal .
EURASIP JOURNAL ON AUDIO SPEECH AND MUSIC PROCESSING, 2024, 2024 (01)
[22]   U-Net-Based Models towards Optimal MR Brain Image Segmentation [J].
Yousef, Rammah ;
Khan, Shakir ;
Gupta, Gaurav ;
Siddiqui, Tamanna ;
Albahlal, Bader M. ;
Alajlan, Saad Abdullah ;
Haq, Mohd Anul .
DIAGNOSTICS, 2023, 13 (09)
[23]   Brain tumor segmentation in MRI with multi-modality spatial information enhancement and boundary shape correction [J].
Zhu, Zhiqin ;
Wang, Ziyu ;
Qi, Guanqiu ;
Mazur, Neal ;
Yang, Pan ;
Liu, Yu .
PATTERN RECOGNITION, 2024, 153
[24]   Sparse Dynamic Volume TransUNet with multi-level edge fusion for brain tumor segmentation [J].
Zhu Z. ;
Sun M. ;
Qi G. ;
Li Y. ;
Gao X. ;
Liu Y. .
Computers in Biology and Medicine, 2024, 172
[25]   Brain tumor segmentation based on the fusion of deep semantics and edge information in multimodal MRI [J].
Zhu, Zhiqin ;
He, Xianyu ;
Qi, Guanqiu ;
Li, Yuanyuan ;
Cong, Baisen ;
Liu, Yu .
INFORMATION FUSION, 2023, 91 :376-387