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

被引:4
|
作者
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 条
  • [1] MUNet: A Multi-scale U-Net Framework for Medical Image Segmentation
    Zhang, Wentao
    Cheng, Hao
    Gan, Jun
    2020 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2020,
  • [2] Medical image segmentation method based on multi-scale feature and U-net network
    Wang, Jingquan
    INTERNET TECHNOLOGY LETTERS, 2023, 7 (05)
  • [3] MFH-Net: A Hybrid CNN-Transformer Network Based Multi-Scale Fusion for Medical Image Segmentation
    Wang, Ying
    Zhang, Meng
    Liang, Jian'an
    Liang, Meiyan
    International Journal of Imaging Systems and Technology, 2024, 34 (06)
  • [4] MSU-Net: Multi-Scale U-Net for 2D Medical Image Segmentation
    Su, Run
    Zhang, Deyun
    Liu, Jinhuai
    Cheng, Chuandong
    FRONTIERS IN GENETICS, 2021, 12
  • [5] Enhancing medical image segmentation with a multi-transformer U-Net
    Dan, Yongping
    Jin, Weishou
    Yue, Xuebin
    Wang, Zhida
    PEERJ, 2024, 12
  • [6] Multi-scale-ResUNet: an improve u-net with multi-scale attention and hybrid dilation for medical image segmentation
    Jin, Tao
    Wang, Zhen
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (18) : 28473 - 28492
  • [7] Multi-scale-ResUNet: an improve u-net with multi-scale attention and hybrid dilation for medical image segmentation
    Tao Jin
    Zhen Wang
    Multimedia Tools and Applications, 2023, 82 : 28473 - 28492
  • [8] 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
  • [9] 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
  • [10] MSGU-Net: a lightweight multi-scale ghost U-Net for image segmentation
    Cheng, Hua
    Zhang, Yang
    Xu, Huangxin
    Li, Dingliang
    Zhong, Zejian
    Zhao, Yinchuan
    Yan, Zhuo
    Frontiers in Neurorobotics, 2024, 18