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 条
  • [31] Performance Characterization of Single and Multi GPU Training of U-Net Architecture for Medical Image Segmentation Tasks
    Patel, Trupesh R.
    Bodduluri, Sandeep
    Anthony, Thomas
    Monroe, William S.
    Kandhare, Pravinkumar G.
    Robinson, John-Paul
    Nakhmani, Arie
    Zhang, Chengcui
    Bhatt, Surya P.
    Bangalore, Purushotham, V
    PEARC '19: PROCEEDINGS OF THE PRACTICE AND EXPERIENCE IN ADVANCED RESEARCH COMPUTING ON RISE OF THE MACHINES (LEARNING), 2019,
  • [32] MS-UNet: Swin Transformer U-Net with Multi-scale Nested Decoder for Medical Image Segmentation with Small Training Data
    Chen, Haoyuan
    Han, Yufei
    Li, Yanyi
    Xu, Pin
    Li, Kuan
    Yin, Jianping
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT XIII, 2024, 14437 : 472 - 483
  • [33] C2 Transformer U-Net: A Medical Image Segmentation Model for Cross-modality and Contextual Semantics
    Zhou, Tao
    Hou, Senbao
    Lu, Huiling
    Liu, Yuncan
    Dang, Pei
    JOURNAL OF ELECTRONICS & INFORMATION TECHNOLOGY, 2023, 45 (05) : 1807 - 1816
  • [34] DRLSU-Net: Level set with U-Net for medical image segmentation
    Wang, Xiaofeng
    Liu, Jiashan
    Yang, Rentao
    Wu, Zhize
    Sun, Lingma
    Zou, Le
    DIGITAL SIGNAL PROCESSING, 2025, 157
  • [35] Suvery of Medical Image Segmentation Technology Based on U-Net Structure Improvement
    Yin X.-H.
    Wang Y.-C.
    Li D.-Y.
    Ruan Jian Xue Bao/Journal of Software, 2021, 32 (02): : 519 - 550
  • [36] A COVID-19 medical image Segmentation method based on U-NET
    Wang, Chao
    Zhu, Jin
    Snu, Kai
    Li, Dayi
    Wang, Zaoji
    Yuan, Huining
    IEEE INTERNATIONAL CONFERENCE ON RECENT ADVANCES IN SYSTEMS SCIENCE AND ENGINEERING (IEEE RASSE 2021), 2021,
  • [37] 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
  • [38] Ensemble Learning for Three-dimensional Medical Image Segmentation of Organ at Risk in Brachytherapy Using Double U-Net, Bi-directional ConvLSTM U-Net, and Transformer Network
    Pal, Soniya
    Singh, Raj Pal
    Kumar, Anuj
    JOURNAL OF MEDICAL PHYSICS, 2024, 49 (04) : 574 - 582
  • [39] 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
  • [40] Medical image segmentation method based on multi-scale feature and U-net network
    Wang, Jingquan
    INTERNET TECHNOLOGY LETTERS, 2024, 7 (05)