DSP-KD: Dual-Stage Progressive Knowledge Distillation for Skin Disease Classification

Xinyi Zeng, Zhanlin Ji, Haiyang Zhang, Rui Chen, Qinping Liao, Jingkun Wang, Tao Lyu*, Li Zhao*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

The increasing global demand for skin disease diagnostics emphasizes the urgent need for advancements in AI-assisted diagnostic technologies for dermatoscopic images. In current practical medical systems, the primary challenge is balancing lightweight models with accurate image analysis to address constraints like limited storage and computational costs. While knowledge distillation methods hold immense potential in healthcare applications, related research on multi-class skin disease tasks is scarce. To bridge this gap, our study introduces an enhanced multi-source knowledge fusion distillation framework, termed DSP-KD, which improves knowledge transfer in a dual-stage progressive distillation approach to maximize mutual information between teacher and student representations. The experimental results highlight the superior performance of our distilled ShuffleNetV2 on both the ISIC2019 dataset and our private skin disorders dataset. Compared to other state-of-the-art distillation methods using diverse knowledge sources, the DSP-KD demonstrates remarkable effectiveness with a smaller computational burden.

Original languageEnglish
Article number70
JournalBioengineering
Volume11
Issue number1
DOIs
Publication statusPublished - Jan 2024

Keywords

  • diverse knowledge
  • ISIC2019
  • knowledge distillation
  • skin disease classification

Fingerprint

Dive into the research topics of 'DSP-KD: Dual-Stage Progressive Knowledge Distillation for Skin Disease Classification'. Together they form a unique fingerprint.

Cite this