HmsU-Net: A hybrid multi-scale U-net based on a CNN and transformer for medical image segmentation

被引:13
|
作者
Fu, Bangkang [1 ,2 ]
Peng, Yunsong [2 ]
He, Junjie [2 ]
Tian, Chong [2 ]
Sun, Xinhuan [2 ]
Wang, Rongpin [2 ,3 ]
机构
[1] Guizhou Univ, Med Coll, Guiyang 550000, Guizhou, Peoples R China
[2] Guizhou Prov Peoples Hosp, Dept Radiol, Key Lab Intelligent Med Imaging Anal & Accurate D, Int Exemplary Cooperat Base Precis Imaging Diag &, Guiyang 550002, Peoples R China
[3] Guizhou Prov Peoples Hosp, Dept Med Imaging, Int Exemplary Cooperat Base Precis Imaging Diag &, 83 Zhongshan East Rd, Guiyang 550002, Guizhou, Peoples R China
基金
中国国家自然科学基金;
关键词
Multi -scale features; Transformer; U; -net; Medical image segmentation; Convolution neural network;
D O I
10.1016/j.compbiomed.2024.108013
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
Accurate medical image segmentation is of great significance for subsequent diagnosis and analysis. The acquisition of multi-scale information plays an important role in segmenting regions of interest of different sizes. With the emergence of Transformers, numerous networks adopted hybrid structures incorporating Transformers and CNNs to learn multi-scale information. However, the majority of research has focused on the design and composition of CNN and Transformer structures, neglecting the inconsistencies in feature learning between Transformer and CNN. This oversight has resulted in the hybrid network's performance not being fully realized. In this work, we proposed a novel hybrid multi-scale segmentation network named HmsU-Net, which effectively fused multi-scale features. Specifically, HmsU-Net employed a parallel design incorporating both CNN and Transformer architectures. To address the inconsistency in feature learning between CNN and Transformer within the same stage, we proposed the multi-scale feature fusion module. For feature fusion across different stages, we introduced the cross-attention module. Comprehensive experiments conducted on various datasets demonstrate that our approach surpasses current state-of-the-art methods.
引用
收藏
页数:10
相关论文
共 50 条
  • [31] 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
  • [32] MSN-Net: a multi-scale context nested U-Net for liver segmentation
    Fan, Tongle
    Wang, Guanglei
    Wang, Xia
    Li, Yan
    Wang, Hongrui
    SIGNAL IMAGE AND VIDEO PROCESSING, 2021, 15 (06) : 1089 - 1097
  • [33] MSN-Net: a multi-scale context nested U-Net for liver segmentation
    Tongle Fan
    Guanglei Wang
    Xia Wang
    Yan Li
    Hongrui Wang
    Signal, Image and Video Processing, 2021, 15 : 1089 - 1097
  • [34] Lightweight Multi-Scale Dilated U-Net for Crop Disease Leaf Image Segmentation
    Xu, Cong
    Yu, Changqing
    Zhang, Shanwen
    ELECTRONICS, 2022, 11 (23)
  • [35] Multiscale transunet plus plus : dense hybrid U-Net with transformer for medical image segmentation
    Wang, Bo
    Wang, Fan
    Dong, Pengwei
    Li, Chongyi
    SIGNAL IMAGE AND VIDEO PROCESSING, 2022, 16 (06) : 1607 - 1614
  • [36] MSP U-Net: Crack Segmentation for Low-Resolution Images Based on Multi-Scale Parallel Attention U-Net
    Kim, Joon-Hyeok
    Noh, Ju-Hyeon
    Jang, Jun-Young
    Yang, Hee-Deok
    Applied Sciences (Switzerland), 2024, 14 (24):
  • [37] 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
  • [38] Hybrid dilation and attention residual U-Net for medical image segmentation
    Wang, Zekun
    Zou, Yanni
    Liu, Peter X.
    COMPUTERS IN BIOLOGY AND MEDICINE, 2021, 134
  • [39] 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
  • [40] Hybrid Swin Deformable Attention U-Net for Medical Image Segmentation
    Wang, Lichao
    Huang, Jiahao
    Xing, Xiaodan
    Yang, Guang
    2023 19TH INTERNATIONAL SYMPOSIUM ON MEDICAL INFORMATION PROCESSING AND ANALYSIS, SIPAIM, 2023,