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 条
[21]   Semi-supervised Learning with a Teacher-Student Network for Generalized Attribute Prediction [J].
Shin, Minchul .
COMPUTER VISION - ECCV 2020, PT XI, 2020, 12356 :509-525
[22]   A semi-supervised fault diagnosis model based on a teacher-student network [J].
Gao Y. ;
Fu Z. ;
Xie Y. ;
Wang S. .
Zhendong yu Chongji/Journal of Vibration and Shock, 2024, 43 (04) :150-157
[23]   Semi-Supervised Dual Stream Segmentation Network for Fundus Lesion Segmentation [J].
Xiang, Dehui ;
Yan, Shenshen ;
Guan, Ying ;
Cai, Mulin ;
Li, Zheqing ;
Liu, Haiyun ;
Chen, Xinjian ;
Tian, Bei .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (03) :713-725
[24]   Contextual Guided Segmentation Framework for Semi-supervised Video Instance Segmentation [J].
Le, Trung-Nghia ;
Nguyen, Tam, V ;
Tran, Minh-Triet .
MACHINE VISION AND APPLICATIONS, 2022, 33 (02)
[25]   Dual teacher model for semi-supervised ABUS tumor segmentation [J].
Pan, Pan ;
Chen, Houjin ;
Li, Yanfeng ;
Li, Jiaxin ;
Cheng, Zhanyi ;
Wang, Shu .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 87
[26]   Contextual Guided Segmentation Framework for Semi-supervised Video Instance Segmentation [J].
Trung-Nghia Le ;
Tam V. Nguyen ;
Minh-Triet Tran .
Machine Vision and Applications, 2022, 33
[27]   Semi-supervised Brain Lesion Segmentation with an Adapted Mean Teacher Model [J].
Cui, Wenhui ;
Liu, Yanlin ;
Li, Yuxing ;
Guo, Menghao ;
Li, Yiming ;
Li, Xiuli ;
Wang, Tianle ;
Zeng, Xiangzhu ;
Ye, Chuyang .
INFORMATION PROCESSING IN MEDICAL IMAGING, IPMI 2019, 2019, 11492 :554-565
[28]   Teacher-Student Mutual Training for Semi-Supervised Object Detection Based on PPYOLOE [J].
Zhang G. ;
Wei J. .
Beijing Ligong Daxue Xuebao/Transaction of Beijing Institute of Technology, 2024, 57 (04) :415-423
[29]   Confidence-aware cross-supervised model for semi-supervised skin lesion segmentation [J].
Shen, Xuanjing ;
Sun, Zhonglin ;
Sun, Yan ;
Chen, Haipeng .
JOURNAL OF ELECTRONIC IMAGING, 2023, 32 (01)
[30]   A segmentation method for oral CBCT image based on Segment Anything Model and semi-supervised teacher-student model [J].
Gan, Jianhong ;
Kang, Runqing ;
Deng, Xun ;
He, Tongli ;
Yu, Nie ;
Gan, Yuling ;
Wei, Peiyang ;
Chen, Xiangyi ;
Peng, Xiaoli ;
Li, Zhibin .
MEDICAL PHYSICS, 2025,