Double U-Net: semi-supervised ultrasound image segmentation combining CNN and transformer's U-shaped network

被引:0
|
作者
Zhou, Huabiao [1 ,2 ]
Luo, Yanmin [1 ,2 ]
Guo, Jingjing [3 ]
Chen, Zhikui [3 ]
Gong, Wanyuan [1 ,2 ]
Lin, Zhongwei [1 ,2 ]
Zhuo, Minling [3 ]
Lin, Youjia [3 ]
Lin, Weiwei [4 ]
Shen, Qingling [3 ]
机构
[1] Huaqiao Univ, Coll Comp Sci & Technol, Xiamen 361021, Fujian, Peoples R China
[2] Huaqiao Univ, Xiamen Key Lab Comp Vis & Pattern Recognit, Xiamen 361021, Fujian, Peoples R China
[3] Fujian Med Univ, Union Hosp, Dept Ultrasound, 29 Xinquan Rd, Fuzhou 350001, Fujian, Peoples R China
[4] Fujian Univ Tradit Chinese Med, Affiliated Peoples Hosp 3, Dept Ultrasound, 363 Guobing Ave, Fuzhou 350122, Fujian, Peoples R China
来源
JOURNAL OF SUPERCOMPUTING | 2025年 / 81卷 / 05期
关键词
Semi-supervised learning; Ultrasound image segmentation; Consistency regularization; Contrastive learning; Automatic segmentation; Feature calibration; ARCHITECTURE;
D O I
10.1007/s11227-025-07152-7
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Ultrasound image segmentation remains challenging due to blurred boundaries and morphological heterogeneity, while existing deep learning methods heavily rely on costly expert annotations. To address these issues, this study proposes a semi-supervised learning algorithm called Double U-Net (W-Net), built on consistency regularization and a cross-teaching framework. Specifically, we introduce a Deeper Dual-output Fusion U-Net (DDFU-Net) designed to tackle ultrasound-specific challenges. This architecture enhances multi-scale feature extraction by improving the backbone network, integrating a dual-output refinement (DOR) module and incorporating a spatial feature calibration (SFC) module to optimize multi-scale feature fusion. Furthermore, the proposed network combines DDFU-Net with a lightweight Transformer, enabling CNNs and Transformers to complement each other in local and global feature extraction. Through mutual end-to-end supervision, the method effectively leverages unlabeled data. Our method achieves competitive performance: (1) Compared to other semi-supervised methods, it outperforms the second-best by 7.96% (BUSI, 20% labels) and 17.52% (10% labels), with 5.47% (GCUI, 20%) and 6.08% (GCUI, 10%) improvements; and (2) compared to fully supervised U-Net, it elevates Dice by 6.09%/3.86% (BUSI) and 3.89%/4.42% (GCUI) under 10%/20% labels condition, proving the ability to effectively leverage unlabeled data, extracting rich feature information to enhance model interpretability of complex medical images, particularly in low-data scenarios.
引用
收藏
页数:29
相关论文
共 50 条
  • [31] Enhancing medical image segmentation with a multi-transformer U-Net
    Dan, Yongping
    Jin, Weishou
    Yue, Xuebin
    Wang, Zhida
    PEERJ, 2024, 12
  • [32] 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
  • [33] DU-Net: A new double U-shaped network for single image dehazing
    Zhang, Xiaodong
    Zhang, Long
    Chu, Menghui
    Wang, Shuo
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2024, 100
  • [34] Ultrasound Image Segmentation Method of Thyroid Nodules Based on the Improved U-Net Network
    Wang, Bo
    Li, Mengxiang
    Liu, Xia
    JOURNAL OF ELECTRONICS & INFORMATION TECHNOLOGY, 2022, 44 (02) : 514 - 522
  • [35] 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
  • [36] Transformer-Based Cascade U-shaped Network for Action Segmentation
    Bao, Wenxia
    Lin, An
    Huang, Hua
    Yang, Xianjun
    Chen, Hemu
    2024 3RD INTERNATIONAL CONFERENCE ON IMAGE PROCESSING AND MEDIA COMPUTING, ICIPMC 2024, 2024, : 157 - 161
  • [37] Design of Superpiexl U-Net Network for Medical Image Segmentation
    Wang H.
    Liu H.
    Guo Q.
    Deng K.
    Zhang C.
    Jisuanji Fuzhu Sheji Yu Tuxingxue Xuebao/Journal of Computer-Aided Design and Computer Graphics, 2019, 31 (06): : 1007 - 1017
  • [38] Segmentation Technology of Nucleus Image Based on U-Net Network
    Fang, Jie
    Zhou, QingBiao
    Wang, Shuxia
    SCIENTIFIC PROGRAMMING, 2021, 2021
  • [39] Fuzzy U-Net Neural Network Design for Image Segmentation
    Kirichev, Mark
    Slavov, Todor
    Momcheva, Galina
    CONTEMPORARY METHODS IN BIOINFORMATICS AND BIOMEDICINE AND THEIR APPLICATIONS, 2022, 374 : 177 - 184
  • [40] MRU-NET: A U-Shaped Network for Retinal Vessel Segmentation
    Ding, Hongwei
    Cui, Xiaohui
    Chen, Leiyang
    Zhao, Kun
    APPLIED SCIENCES-BASEL, 2020, 10 (19):