A Dual-Headed Teacher-Student Framework with an Uncertainty-Guided Mechanism for Semi-Supervised Skin Lesion Segmentation

被引:1
作者
Zou, Changman [1 ,2 ]
Jeon, Wang-Su [3 ]
Ju, Hye-Rim [2 ]
Rhee, Sang-Yong [3 ]
机构
[1] Beihua Univ, Coll Comp Sci & Technol, Jilin 132013, Peoples R China
[2] Kyungnam Univ, Dept IT Convergence Engn, Chang Won 51767, South Korea
[3] Kyungnam Univ, Dept Comp Engn, Chang Won 51767, South Korea
来源
ELECTRONICS | 2025年 / 14卷 / 05期
关键词
skin lesion segmentation; semi-supervised learning; teacher-student framework; uncertainty-guided pseudo-labeling; dual-headed architecture; medical image analysis; DIAGNOSIS; IMAGE;
D O I
10.3390/electronics14050984
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Medical image segmentation is a challenging task due to limited annotated data, complex lesion boundaries, and the inherent variability in medical images. These challenges make accurate and robust segmentation crucial for clinical applications. In this study, we propose the Uncertainty-Driven Auxiliary Mean Teacher (UDAMT) model, a novel semi-supervised framework specifically designed for skin lesion segmentation. Our approach employs a dual-headed teacher-student architecture with an uncertainty-guided mechanism, enhancing feature learning and boundary precision. Extensive experiments on the ISIC 2016, ISIC 2017, and ISIC 2018 datasets demonstrate that UDAMT achieves significant improvements over state-of-the-art methods, with increases of 1.17 percentage points in the Dice coefficient and 1.31 percentage points in mean Intersection over Union (mIoU) under low-label settings (5% labeled data). Furthermore, UDAMT requires 12.9 M parameters, which is slightly higher than the baseline model (12.5 M) but significantly lower than MT (14.8 M) and UAMT (15.2 M). It also achieves an inference time of 25.7 ms per image, ensuring computational efficiency. Ablation studies validate the contributions of each component, and cross-dataset evaluations on the PH2 benchmark confirm robustness to small lesions. This work provides a scalable and efficient solution for semi-supervised medical image segmentation, balancing accuracy, efficiency, and clinical applicability.
引用
收藏
页数:24
相关论文
共 50 条
[31]   Semi-Supervised Segmentation Framework for Gastrointestinal Lesion Diagnosis in Endoscopic Images [J].
Lonseko, Zenebe Markos ;
Du, Wenju ;
Adjei, Prince Ebenezer ;
Luo, Chengsi ;
Hu, Dingcan ;
Gan, Tao ;
Zhu, Linlin ;
Rao, Nini .
JOURNAL OF PERSONALIZED MEDICINE, 2023, 13 (01)
[32]   Multi-scale spatial consistency for deep semi-supervised skin lesion segmentation [J].
Nouboukpo, Adama ;
Allaoui, Mohamed Lamine ;
Allili, Mohand Said .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 135
[33]   Uncertainty teacher with dense focal loss for semi-supervised medical image segmentation [J].
Chen, Jialei ;
Fu, Chong ;
Xie, Haoyu ;
Zheng, Xu ;
Geng, Rong ;
Sham, Chiu-Wing .
COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 149
[34]   Uncertainty-Guided Semi-Supervised Few-Shot Class-Incremental Learning With Knowledge Distillation [J].
Cui, Yawen ;
Deng, Wanxia ;
Xu, Xin ;
Liu, Zhen ;
Liu, Zhong ;
Pietikainen, Matti ;
Liu, Li .
IEEE TRANSACTIONS ON MULTIMEDIA, 2023, 25 :6422-6435
[35]   Dual Attention Based Uncertainty-aware Mean Teacher Model for Semi-supervised Cardiac Image Segmentation [J].
Xu, An ;
Wang, Shaoyu ;
Fan, Jingyi ;
Shi, Xiujin ;
Chen, Qiang .
PROCEEDINGS OF THE 2021 IEEE INTERNATIONAL CONFERENCE ON PROGRESS IN INFORMATICS AND COMPUTING (PIC), 2021, :82-86
[36]   Semi-supervised federated learning fault diagnosis method driven by teacher-student model consistency [J].
Wang, Guilong ;
Pu, Chenjie ;
Fu, Dongliang ;
Zhang, Yi ;
Yu, Jiongmin ;
Hou, Yanru .
SIGNAL IMAGE AND VIDEO PROCESSING, 2025, 19 (05)
[37]   Consistency regularization teacher-student semi-supervised learning method for target recognition in SAR images [J].
Tian, Ye ;
Zhang, Liguo ;
Sun, Jianguo ;
Yin, Guisheng ;
Dong, Yuxin .
VISUAL COMPUTER, 2022, 38 (12) :4179-4192
[38]   Semi-TSGAN: Semi-Supervised Learning for Highlight Removal Based on Teacher-Student Generative Adversarial Network [J].
Zheng, Yuanfeng ;
Yan, Yuchen ;
Jiang, Hao .
SENSORS, 2024, 24 (10)
[39]   Uncertainty Global Contrastive Learning Framework for Semi-Supervised Medical Image Segmentation [J].
Liu, Hengyang ;
Ren, Pengcheng ;
Yuan, Yang ;
Song, Chengyun ;
Luo, Fen .
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2025, 29 (01) :433-442
[40]   Self-training with dual uncertainty for semi-supervised MRI image segmentation [J].
Qiu, Zhanhong ;
Gan, Haitao ;
Shi, Ming ;
Huang, Zhongwei ;
Yang, Zhi .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 94