arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.11490cs.CV
收藏

SSD-KD:一种基于皮肤镜图像的自监督多元知识蒸馏分类方法

SSD-KD: A Self-supervised Diverse Knowledge Distillation Method for Lightweight Skin Lesion Classification Using Dermoscopic Images

Nanyang Technological University(南洋理工大学);University of British Columbia(不列颠哥伦比亚大学);Vancouver Coast Health Research Institute(温哥华海岸健康研究所);BC Cancer(不列颠哥伦比亚癌症中心)

Yongwei Wang, Yuheng Wang, Tim K. Lee, Chunyan Miao, Z. Jane Wang

摘要:

皮肤癌是最常见的恶性肿瘤之一,影响大量人口,并在世界范围内造成沉重的经济负担。在过去的几年里,由于人工智能的进步,计算机辅助诊断得到了迅速发展,并在医疗保健和医疗实践中取得了很大进展。然而,大多数皮肤癌检测的研究都在追求高预测精度,而没有考虑便携式设备上计算资源的限制。在这种情况下,知识蒸馏(KD)已被证明是一个有效的工具,帮助提高轻量级模型在有限的资源下的适应性,同时保持高层次的表示能力。为了弥合这一差距,这项研究特别提出了一种新的方法,称为SSD-KD,它将不同的知识统一到一个通用的KD框架中,用于皮肤病分类。我们的方法模型的实例内关系特征表示,并将其与现有的KD研究。一个双关系知识蒸馏架构是自我监督训练,而加权软化输出也被利用,使学生模型,以捕捉更丰富的知识,从教师模型。为了证明我们的方法的有效性,我们在ISIC 2019上进行了实验,ISIC 2019是一个大规模开放访问的皮肤病皮肤镜图像基准。实验结果表明,我们的蒸馏轻量级模型可以达到高达85%的准确率为8种不同的皮肤疾病的分类任务,最小的参数和计算要求。消融研究证实了我们的内部和实例间的关系知识整合策略的有效性。与现有的知识蒸馏技术相比,该方法在大规模皮肤镜数据库上的多疾病分类性能得到了提高。

英文摘要:

Skin cancer is one of the most common types of malignancy, affecting a large population and causing a heavy economic burden worldwide. Over the last few years, computer-aided diagnosis has been rapidly developed and make great progress in healthcare and medical practices due to the advances in artificial intelligence. However, most studies in skin cancer detection keep pursuing high prediction accuracies without considering the limitation of computing resources on portable devices. In this case, knowledge distillation (KD) has been proven as an efficient tool to help improve the adaptability of lightweight models under limited resources, meanwhile keeping a high-level representation capability. To bridge the gap, this study specifically proposes a novel method, termed SSD-KD, that unifies diverse knowledge into a generic KD framework for skin diseases classification. Our method models an intra-instance relational feature representation and integrates it with existing KD research. A dual relational knowledge distillation architecture is self-supervisedly trained while the weighted softened outputs are also exploited to enable the student model to capture richer knowledge from the teacher model. To demonstrate the effectiveness of our method, we conduct experiments on ISIC 2019, a large-scale open-accessed benchmark of skin diseases dermoscopic images. Experiments show that our distilled lightweight model can achieve an accuracy as high as 85% for the classification tasks of 8 different skin diseases with minimal parameters and computing requirements. Ablation studies confirm the effectiveness of our intra- and inter-instance relational knowledge integration strategy. Compared with state-of-the-art knowledge distillation techniques, the proposed method demonstrates improved performances for multi-diseases classification on the large-scale dermoscopy database.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑