Enhancing medical image segmentation with a multi-transformer U-Net

被引:3
|
作者
Dan, Yongping [1 ]
Jin, Weishou [1 ]
Yue, Xuebin [2 ]
Wang, Zhida [1 ]
机构
[1] Zhongyuan Univ Technol, Sch Elect & Informat, Zhengzhou, Henan, Peoples R China
[2] Ritsumeikan Univ, Res Org Sci & Technol, Kusatsu, Japan
来源
PEERJ | 2024年 / 12卷
关键词
CT or X-ray lung images; Medical image segmentation; Multi-transformer; Unet;
D O I
10.7717/peerj.17005
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Various segmentation networks based on Swin Transformer have shown promise in medical segmentation tasks. Nonetheless, challenges such as lower accuracy and slower training convergence have persisted. To tackle these issues, we introduce a novel approach that combines the Swin Transformer and Deformable Transformer to enhance overall model performance. We leverage the Swin Transformer's window attention mechanism to capture local feature information and employ the Deformable Transformer to adjust sampling positions dynamically, accelerating model convergence and aligning it more closely with object shapes and sizes. By amalgamating both Transformer modules and incorporating additional skip connections to minimize information loss, our proposed model excels at rapidly and accurately segmenting CT or X-ray lung images. Experimental results demonstrate the remarkable, showcasing the significant prowess of our model. It surpasses the performance of the standalone Swin Transformer's Swin Unet and converges more rapidly under identical conditions, yielding accuracy improvements of 0.7% (resulting in 88.18%) and 2.7% (resulting in 98.01%) on the COVID-19 CT scan lesion segmentation dataset and Chest X-ray Masks and Labels dataset, respectively. This advancement has the potential to aid medical practitioners in early diagnosis and treatment decision-making.
引用
收藏
页数:19
相关论文
共 50 条
  • [41] DRA U-Net: An Attention based U-Net Framework for 2D Medical Image Segmentation
    Zhang, Xian
    Feng, Ziyuan
    Zhong, Tianchi
    Shen, Sicheng
    Zhang, Ruolin
    Zhou, Lijie
    Zhang, Bo
    Wang, Wendong
    2021 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2021, : 3936 - 3942
  • [42] Auxiliary Segmentation Method of Osteosarcoma MRI Image Based on Transformer and U-Net
    Liu, Feng
    Zhu, Jun
    Lv, Baolong
    Yang, Lei
    Sun, Wenyan
    Dai, Zhehao
    Gou, Fangfang
    Wu, Jia
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022 : 9990092
  • [43] Adaptive enhanced swin transformer with U-net for remote sensing image segmentation*
    Gu, Xingjian
    Li, Sizhe
    Ren, Shougang
    Zheng, Hengbiao
    Fan, Chengcheng
    Xu, Huanliang
    COMPUTERS & ELECTRICAL ENGINEERING, 2022, 102
  • [44] Efficient Generative-Adversarial U-Net for Multi-Organ Medical Image Segmentation
    Wang, Haoran
    Wu, Gengshen
    Liu, Yi
    JOURNAL OF IMAGING, 2025, 11 (01)
  • [45] Medical image segmentation method based on multi-scale feature and U-net network
    Wang, Jingquan
    INTERNET TECHNOLOGY LETTERS, 2024, 7 (05)
  • [46] MSU-Net: Multi-Scale U-Net for 2D Medical Image Segmentation
    Su, Run
    Zhang, Deyun
    Liu, Jinhuai
    Cheng, Chuandong
    FRONTIERS IN GENETICS, 2021, 12
  • [47] Shape-intensity-guided U-net for medical image segmentation
    Dong, Wenhui
    Du, Bo
    Xu, Yongchao
    NEUROCOMPUTING, 2024, 610
  • [48] Content-adaptive U-Net Architecture for Medical Image Segmentation
    Mostayed, Ahmed
    Wee, William G.
    Zhou, Xuefu
    2019 6TH INTERNATIONAL CONFERENCE ON COMPUTATIONAL SCIENCE AND COMPUTATIONAL INTELLIGENCE (CSCI 2019), 2019, : 698 - 702
  • [49] Medical image segmentation based on state transition algorithm and U-net
    Zhou, Xiaojun
    Geng, Chuanyu
    Yang, Chunhua
    Zhongnan Daxue Xuebao (Ziran Kexue Ban)/Journal of Central South University (Science and Technology), 2023, 54 (04): : 1358 - 1369
  • [50] Application of U-Net and Optimized Clustering in Medical Image Segmentation: A Review
    Shao, Jiaqi
    Chen, Shuwen
    Zhou, Jin
    Zhu, Huisheng
    Wang, Ziyi
    Brown, Mackenzie
    CMES-COMPUTER MODELING IN ENGINEERING & SCIENCES, 2023, 136 (03): : 2173 - 2219