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 条
  • [41] A Bypass-Based U-Net for Medical Image Segmentation
    Chen, Kaixuan
    Xu, Gengxin
    Qian, Jiaying
    Ren, Chuan-Xian
    INTELLIGENCE SCIENCE AND BIG DATA ENGINEERING: VISUAL DATA ENGINEERING, PT I, 2019, 11935 : 155 - 164
  • [42] Ultrasound image segmentation based on Transformer and U-Net with joint loss
    Cai, Lina
    Li, Qingkai
    Zhang, Junhua
    Zhang, Zhenghua
    Yang, Rui
    Zhang, Lun
    PEERJ COMPUTER SCIENCE, 2023, 9 : 1 - 18
  • [43] Latent fingerprint segmentation using multi-scale attention U-Net
    Akhila, P.
    Koolagudi, Shashidhar G.
    INTERNATIONAL JOURNAL OF BIOMETRICS, 2024, 16 (02) : 195 - 215
  • [44] Double-branch U-Net for multi-scale organ segmentation
    Liu, Yuhao
    Qin, Caijie
    Yu, Zhiqian
    Yang, Ruijie
    Suqing, Tian
    Liu, Xia
    Ma, Xibo
    METHODS, 2022, 205 : 220 - 225
  • [45] A Multi-Scale Feature Fusion Method Based on U-Net for Retinal Vessel Segmentation
    Yang, Dan
    Liu, Guoru
    Ren, Mengcheng
    Xu, Bin
    Wang, Jiao
    ENTROPY, 2020, 22 (08)
  • [46] 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
  • [47] A New Segmentation Method to Preserve the Underlying Image Features: U-Net with Multi-scale Pooling
    Liu, Yuming
    Chen, Gongping
    Dai, Yu
    2021 IEEE INTERNATIONAL CONFERENCE ON ROBOTICS AND BIOMIMETICS (IEEE-ROBIO 2021), 2021, : 1718 - 1722
  • [48] MFA U-Net: a U-Net like multi-stage feature analysis network for medical image segmentation
    Wang, Yupeng
    Wang, Suyu
    He, Jian
    PATTERN ANALYSIS AND APPLICATIONS, 2024, 27 (04)
  • [49] 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
  • [50] MCA-UNet: multi-scale cross co-attentional U-Net for automatic medical image segmentation
    Haonan Wang
    Peng Cao
    Jinzhu Yang
    Osmar Zaiane
    Health Information Science and Systems, 11