SSD-KD
SSD-KD performs self-supervised diverse knowledge distillation to produce lightweight MobileNetV2-based models for multi-disease skin lesion classification from dermoscopic images.
Key Features:
- Knowledge Distillation Framework: Integrates diverse forms of knowledge into a unified knowledge distillation (KD) framework to improve student model performance.
- Intra-Instance Relational Feature Representation: Captures complex relationships within individual instances via an intra-instance relational feature representation.
- Dual Relational Knowledge Distillation Architecture: Employs a dual relational KD architecture trained self-supervised that transfers intra- and inter-instance relational knowledge and uses weighted softened outputs.
- Lightweight Model Efficiency: Uses MobileNetV2 as the student backbone to achieve high classification accuracy with minimal parameters and computation.
Scientific Applications:
- Multi-disease skin lesion classification: Classifies multiple skin diseases from dermoscopic images for dermatological diagnosis tasks.
- Benchmarking on ISIC 2019: Achieved up to 85% accuracy on eight-class skin disease classification using MobileNetV2 on the ISIC 2019 dataset.
- Ablation study validation: Ablation experiments confirmed the contribution of intra- and inter-instance relational knowledge integration to performance improvements.
Methodology:
Self-supervised training of a dual relational knowledge distillation architecture that transfers knowledge from a complex teacher model to a lightweight student (MobileNetV2) by integrating intra-instance and inter-instance relational feature representations and using weighted softened outputs, validated on the ISIC 2019 dermoscopic image dataset.
Topics
Details
- License:
- MIT
- Cost:
- Free of charge
- Tool Type:
- command-line tool
- Operating Systems:
- Mac, Linux, Windows
- Programming Languages:
- Python
- Added:
- 2/13/2023
- Last Updated:
- 11/24/2024
Operations
Publications
Wang Y, Wang Y, Cai J, Lee TK, Miao C, Wang ZJ. SSD-KD: A self-supervised diverse knowledge distillation method for lightweight skin lesion classification using dermoscopic images. Medical Image Analysis. 2023;84:102693. doi:10.1016/j.media.2022.102693. PMID:36462373.