用在线知识蒸馏提升皮肤病变分类,少标注也能高精度。
Semi-Supervised Learning with Online Knowledge Distillation for Skin Lesion Classification
- 用集成学习加在线知识蒸馏,让模型互相学
- 在两个公开数据集上超越现有最好水平
- 单个模型可独立部署,适合资源受限场景
深度学习在皮肤病变分析中展现出巨大潜力。然而,现有方法多依赖全监督学习,需大量标注数据,获取成本高。为减轻标注负担,本研究提出一种新颖的半监督深度学习方法,结合集成学习与在线知识蒸馏,以提升皮肤病变分类性能。方法通过训练一组卷积神经网络集成模型,利用在线知识蒸馏将集成整体的知识传递给各成员模型,从而增强每个模型表现,进而提升整体集成性能。训练完成后,集成中任意单一模型均可在测试时独立部署,且性能与集成相当,特别适用于资源受限环境。实验表明,经知识蒸馏的单个模型优于独立训练的模型。该方法在《国际皮肤影像合作组织》2018 和 2019 公共基准数据集上均取得领先结果,显著优于当前最先进方法。通过融合集成学习与在线知识蒸馏,该方法在减少标注数据需求的同时,提供了更高效的现实应用解决方案。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。