Semi-Supervised Learning with Online Knowledge Distillation for Skin Lesion Classification
By: Siyamalan Manivannan
Potential Business Impact:
Helps doctors spot skin problems with less data.
Deep Learning has emerged as a promising approach for skin lesion analysis. However, existing methods mostly rely on fully supervised learning, requiring extensive labeled data, which is challenging and costly to obtain. To alleviate this annotation burden, this study introduces a novel semi-supervised deep learning approach that integrates ensemble learning with online knowledge distillation for enhanced skin lesion classification. Our methodology involves training an ensemble of convolutional neural network models, using online knowledge distillation to transfer insights from the ensemble to its members. This process aims to enhance the performance of each model within the ensemble, thereby elevating the overall performance of the ensemble itself. Post-training, any individual model within the ensemble can be deployed at test time, as each member is trained to deliver comparable performance to the ensemble. This is particularly beneficial in resource-constrained environments. Experimental results demonstrate that the knowledge-distilled individual model performs better than independently trained models. Our approach demonstrates superior performance on both the \emph{International Skin Imaging Collaboration} 2018 and 2019 public benchmark datasets, surpassing current state-of-the-art results. By leveraging ensemble learning and online knowledge distillation, our method reduces the need for extensive labeled data while providing a more resource-efficient solution for skin lesion classification in real-world scenarios.
Similar Papers
Deep Self-knowledge Distillation: A hierarchical supervised learning for coronary artery segmentation
Image and Video Processing
Helps doctors find heart problems faster from X-rays.
Uncertainty-Aware Retinal Vessel Segmentation via Ensemble Distillation
CV and Pattern Recognition
Makes eye scans more accurate with less computer power.
Preventing Shortcut Learning in Medical Image Analysis through Intermediate Layer Knowledge Distillation from Specialist Teachers
CV and Pattern Recognition
Helps AI see real problems, not fake clues.