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
关键词
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 条
[41]   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
[42]   Decoupling mean teacher via Dual students: a prototype-based framework for semi-supervised medical segmentation [J].
Zuo, Yang ;
Zhang, Renfeng ;
Guo, Xiurui ;
Kang, Chunmeng ;
Lyu, Lei .
COMPLEX & INTELLIGENT SYSTEMS, 2025, 11 (08)
[43]   Uncertainty guided semi-supervised few-shot segmentation with prototype level fusion [J].
Wang, Hailing ;
Wu, Chunwei ;
Zhang, Hai ;
Cao, Guitao ;
Cao, Wenming .
NEURAL NETWORKS, 2025, 181
[44]   A Semi-Supervised Object Detection Algorithm Based on Teacher-Student Models with Strong-Weak Heads [J].
Cai, Xiaowei ;
Luo, Fuyi ;
Qi, Wei ;
Liu, Hong .
ELECTRONICS, 2022, 11 (23)
[45]   Semi-supervised end-to-end ASR via teacher-student learning with conditional posterior distribution [J].
Zhang, Zi-qiang ;
Song, Yan ;
Zhang, Jian-shu ;
McLoughlin, Ian ;
Dai, Li-Rong .
INTERSPEECH 2020, 2020, :3580-3584
[46]   Semi-Supervised CT Lesion Segmentation Using Uncertainty-Based Data Pairing and SwapMix [J].
Qiao, Pengchong ;
Li, Han ;
Song, Guoli ;
Han, Hu ;
Gao, Zhiqiang ;
Tian, Yonghong ;
Liang, Yongsheng ;
Li, Xi ;
Zhou, S. Kevin ;
Chen, Jie .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (05) :1546-1562
[47]   Dual-Consistency Semi-supervised Learning with Uncertainty Quantification for COVID-19 Lesion Segmentation from CT Images [J].
Li, Yanwen ;
Luo, Luyang ;
Lin, Huangjing ;
Chen, Hao ;
Heng, Pheng-Ann .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT II, 2021, 12902 :199-209
[48]   Hybrid dual mean-teacher network with double-uncertainty guidance for semi-supervised segmentation of magnetic resonance images [J].
Zhu, Jiayi ;
Bolsterlee, Bart ;
Chow, Brian V. Y. ;
Song, Yang ;
Meijering, Erik .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2024, 115
[49]   Semi-supervised single image dehazing based on dual-teacher-student network with knowledge transfer [J].
Liu, Jianlei ;
Hou, Qianwen ;
Wang, Shilong ;
Zhang, Xueqing .
SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (6-7) :5073-5087
[50]   CUAMT: A MRI semi-supervised medical image segmentation framework based on contextual information and mixed uncertainty [J].
Xiao, Hanguang ;
Wang, Yangjian ;
Xiong, Shidong ;
Ren, Yanjun ;
Zhang, Hongmin .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2025, 267