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

被引:6
作者
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
关键词
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
相关论文
共 34 条
[1]  
Abedalla A, 2021, PEERJ COMPUT SCI, V7, DOI 10.7717/peerj-cs.607
[2]   SEEDED REGION GROWING [J].
ADAMS, R ;
BISCHOF, L .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 1994, 16 (06) :641-647
[3]   Improved Road Connectivity by Joint Learning of Orientation and Segmentation [J].
Batra, Anil ;
Singh, Suriya ;
Pang, Guan ;
Basu, Saikat ;
Jawahar, C., V ;
Paluri, Manohar .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :10377-10385
[4]   Lung Segmentation in Chest Radiographs Using Anatomical Atlases With Nonrigid Registration [J].
Candemir, Sema ;
Jaeger, Stefan ;
Palaniappan, Kannappan ;
Musco, Jonathan P. ;
Singh, Rahul K. ;
Xue, Zhiyun ;
Karargyris, Alexandros ;
Antani, Sameer ;
Thoma, George ;
McDonald, Clement J. .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2014, 33 (02) :577-590
[5]   Automatic Tuberculosis Screening Using Chest Radiographs [J].
Jaeger, Stefan ;
Karargyris, Alexandros ;
Candemir, Sema ;
Folio, Les ;
Siegelman, Jenifer ;
Callaghan, Fiona ;
Xue, Zhiyun ;
Palaniappan, Kannappan ;
Singh, Rahul K. ;
Antani, Sameer ;
Thoma, George ;
Wang, Yi-Xiang ;
Lu, Pu-Xuan ;
McDonald, Clement J. .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2014, 33 (02) :233-245
[6]   Multiple Resolution Residually Connected Feature Streams for Automatic Lung Tumor Segmentation From CT Images [J].
Jiang, Jue ;
Hu, Yu-Chi ;
Liu, Chia-Ju ;
Halpenny, Darragh ;
Hellmann, Matthew D. ;
Deasy, Joseph O. ;
Mageras, Gig ;
Veeraraghavan, Harini .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (01) :134-144
[7]  
Jun Ma, 2020, Zenodo, DOI 10.5281/ZENODO.3757476
[8]  
Kayalibay B, 2017, Arxiv, DOI arXiv:1701.03056
[9]   Breast tumor segmentation and shape classification in mammograms using generative adversarial and convolutional neural network [J].
Kumar Singh, Vivek ;
Rashwan, Hatem A. ;
Romani, Santiago ;
Akram, Farhan ;
Pandey, Nidhi ;
Kamal Sarker, Md Mostafa ;
Saleh, Adel ;
Arenas, Meritxell ;
Arquez, Miguel ;
Puig, Domenec ;
Torrents-Barrena, Jordina .
EXPERT SYSTEMS WITH APPLICATIONS, 2020, 139
[10]   A review and comparison of breast tumor cell nuclei segmentation performances using deep convolutional neural networks [J].
Lagree, Andrew ;
Mohebpour, Majidreza ;
Meti, Nicholas ;
Saednia, Khadijeh ;
Lu, Fang-, I ;
Slodkowska, Elzbieta ;
Gandhi, Sonal ;
Rakovitch, Eileen ;
Shenfield, Alex ;
Sadeghi-Naini, Ali ;
Tran, William T. .
SCIENTIFIC REPORTS, 2021, 11 (01)