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 条
  • [1] STS-TransUNet: Semi-supervised Tooth Segmentation Transformer U-Net for dental panoramic image
    Sun D.
    Wang J.
    Zuo Z.
    Jia Y.
    Wang Y.
    Mathematical Biosciences and Engineering, 2024, 21 (02) : 2366 - 2384
  • [2] A Comparative Analysis of U-Net and Vision Transformer Architectures in Semi-Supervised Prostate Zonal Segmentation
    Huang, Guantian
    Xia, Bixuan
    Zhuang, Haoming
    Yan, Bohan
    Wei, Cheng
    Qi, Shouliang
    Qian, Wei
    He, Dianning
    BIOENGINEERING-BASEL, 2024, 11 (09):
  • [3] A deep supervised transformer U-shaped full-resolution residual network for the segmentation of breast ultrasound image
    Zhou, Jiale
    Hou, Zuoxun
    Lu, Hongyan
    Wang, Wenhan
    Zhao, Wanchen
    Wang, Zenan
    Zheng, Dezhi
    Wang, Shuai
    Tang, Wenzhong
    Qu, Xiaolei
    MEDICAL PHYSICS, 2023, 50 (12) : 7513 - 7524
  • [4] Collaborative transformer U-shaped network for medical image segmentation
    Gao, Yufei
    Zhang, Shichao
    Shi, Lei
    Zhao, Guohua
    Shi, Yucheng
    APPLIED SOFT COMPUTING, 2025, 173
  • [5] 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
  • [6] Optimization of U-shaped pure transformer medical image segmentation network
    Dan, Yongping
    Jin, Weishou
    Wang, Zhida
    Sun, Changhao
    PEERJ COMPUTER SCIENCE, 2023, 9
  • [7] 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
  • [8] 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
  • [9] Semi-supervised Learning Framework in Segmentation of Retinal Blood Vessel Based on U-Net
    Li, Yaning
    Pei, Zijun
    Li, Jiaguang
    Chen, Dali
    PROCEEDINGS OF THE 33RD CHINESE CONTROL AND DECISION CONFERENCE (CCDC 2021), 2021, : 5972 - 5978
  • [10] GT U-Net: A U-Net Like Group Transformer Network for Tooth Root Segmentation
    Li, Yunxiang
    Wang, Shuai
    Wang, Jun
    Zeng, Guodong
    Liu, Wenjun
    Zhang, Qianni
    Jin, Qun
    Wang, Yaqi
    MACHINE LEARNING IN MEDICAL IMAGING, MLMI 2021, 2021, 12966 : 386 - 395