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

被引:1
|
作者
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 条
  • [1] MIXED TRANSFORMER U-NET FOR MEDICAL IMAGE SEGMENTATION
    Wang, Hongyi
    Xie, Shiao
    Lin, Lanfen
    Iwamoto, Yutaro
    Han, Xian-Hua
    Chen, Yen-Wei
    Tong, Ruofeng
    2022 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2022, : 2390 - 2394
  • [2] Diffusion Transformer U-Net for Medical Image Segmentation
    Chowdary, G. Jignesh
    Yin, Zhaozheng
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT IV, 2023, 14223 : 622 - 631
  • [3] U-Net Transformer: Self and Cross Attention for Medical Image Segmentation
    Petit, Olivier
    Thome, Nicolas
    Rambour, Clement
    Themyr, Loic
    Collins, Toby
    Soler, Luc
    MACHINE LEARNING IN MEDICAL IMAGING, MLMI 2021, 2021, 12966 : 267 - 276
  • [4] U-Net vs Transformer: Is U-Net Outdated in Medical Image Registration?
    Jia, Xi
    Bartlett, Joseph
    Zhang, Tianyang
    Lu, Wenqi
    Qiu, Zhaowen
    Duan, Jinming
    MACHINE LEARNING IN MEDICAL IMAGING, MLMI 2022, 2022, 13583 : 151 - 160
  • [5] Cross Pyramid Transformer makes U-net stronger in medical image segmentation
    Zhu, Jinghua
    Sheng, Yue
    Cui, Hui
    Ma, Jiquan
    Wang, Jijian
    Xi, Heran
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 86
  • [6] Multiscale transunet +  + : dense hybrid U-Net with transformer for medical image segmentation
    Bo Wang
    ·Fan Wang
    Pengwei Dong
    ·Chongyi Li
    Signal, Image and Video Processing, 2022, 16 : 1607 - 1614
  • [7] Wavelet U-Net for Medical Image Segmentation
    Ying Li
    Yu Wang
    Tuo Leng
    Wen Zhijie
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2020, PT I, 2020, 12396 : 800 - 810
  • [8] TransAttUnet: Multi-Level Attention-Guided U-Net With Transformer for Medical Image Segmentation
    Chen, Bingzhi
    Liu, Yishu
    Zhang, Zheng
    Lu, Guangming
    Kong, Adams Wai Kin
    IEEE TRANSACTIONS ON EMERGING TOPICS IN COMPUTATIONAL INTELLIGENCE, 2024, 8 (01): : 55 - 68
  • [9] HmsU-Net: A hybrid multi-scale U-net based on a CNN and transformer for medical image segmentation
    Fu, Bangkang
    Peng, Yunsong
    He, Junjie
    Tian, Chong
    Sun, Xinhuan
    Wang, Rongpin
    COMPUTERS IN BIOLOGY AND MEDICINE, 2024, 170
  • [10] DS-TransUNet: Dual Swin Transformer U-Net for Medical Image Segmentation
    Lin, Ailiang
    Chen, Bingzhi
    Xu, Jiayu
    Zhang, Zheng
    Lu, Guangming
    Zhang, David
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2022, 71