Knowledge distillation approach for skin cancer classification on lightweight deep learning model

被引:0
|
作者
Saha, Suman [1 ]
Hemal, Md. Moniruzzaman [1 ]
Eidmum, Md. Zunead Abedin [1 ]
Mridha, Muhammad Firoz [2 ]
机构
[1] Department of IoT and Robotics Engineering, Bangabandhu Sheikh Mujibur Rahman Digital University, Bangladesh, Gazipur, Bangladesh
[2] Department of Computer Science, American International University-Bangladesh, Dhaka, Bangladesh
关键词
Adversarial machine learning - Contrastive Learning - Deep learning - Diseases - Oncology - Personnel training - Students;
D O I
10.1049/htl2.12120
中图分类号
学科分类号
摘要
Over the past decade, there has been a global increase in the incidence of skin cancers. Skin cancer has serious consequences if left untreated, potentially leading to more advanced cancer stages. In recent years, deep learning based convolutional neural network have emerged as powerful tools for skin cancer detection. Generally, deep learning approaches are computationally expensive and require large storage space. Therefore, deploying such a large complex model on resource-constrained devices is challenging. An ultra-light and accurate deep learning model is highly desirable for better inference time and memory in low-power-consuming devices. Knowledge distillation is an approach for transferring knowledge from a large network to a small network. This small network is easily compatible with resource-constrained embedded devices while maintaining accuracy. The main aim of this study is to develop a deep learning-based lightweight network based on knowledge distillation that identifies the presence of skin cancer. Here, different training strategies are implemented for the modified benchmark (Phase 1) and custom-made model (Phase 2) and demonstrated various distillation configurations on two datasets: HAM10000 and ISIC2019. In Phase 1, the student model using knowledge distillation achieved accuracies ranging from 88.69% to 93.24% for HAM10000 and from 82.14% to 84.13% on ISIC2019. In Phase 2, the accuracies ranged from 88.63% to 88.89% on HAM10000 and from 81.39% to 83.42% on ISIC2019. These results highlight the effectiveness of knowledge distillation in improving the classification performance across diverse datasets and enabling the student model to approach the performance of the teacher model. In addition, the distilled student model can be easily deployed on resource-constrained devices for automated skin cancer detection due to its lower computational complexity. © 2025 The Author(s). Healthcare Technology Letters published by John Wiley & Sons Ltd on behalf of The Institution of Engineering and Technology.
引用
收藏
相关论文
共 50 条
  • [21] LEARNING LIGHTWEIGHT PEDESTRIAN DETECTOR WITH HIERARCHICAL KNOWLEDGE DISTILLATION
    Chen, Rui
    Ai, Haizhou
    Shang, Chong
    Chen, Long
    Zhuang, Zijie
    2019 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2019, : 1645 - 1649
  • [22] Skin Cancer Classification using Deep Learning Models
    Kahia, Marwa
    Echtioui, Amira
    Kallel, Fathi
    Ben Hamida, Ahmed
    ICAART: PROCEEDINGS OF THE 14TH INTERNATIONAL CONFERENCE ON AGENTS AND ARTIFICIAL INTELLIGENCE - VOL 1, 2022, : 554 - 559
  • [23] An Advanced Deep Learning Framework for Skin Cancer Classification
    Khan, Muhammad Amir
    Ali, Muhammad Danish
    Mazhar, Tehseen
    Shahzad, Tariq
    Rehman, Waheed Ur
    Shahid, Mohammad
    Hamam, Habib
    REVIEW OF SOCIONETWORK STRATEGIES, 2025, : 111 - 130
  • [24] Automatic Malignant and Benign Skin Cancer Classification Using a Hybrid Deep Learning Approach
    Bassel, Atheer
    Abdulkareem, Amjed Basil
    Alyasseri, Zaid Abdi Alkareem
    Sani, Nor Samsiah
    Mohammed, Husam Jasim
    DIAGNOSTICS, 2022, 12 (10)
  • [25] Lightweight remote sensing scene classification based on knowledge distillation
    Zhang, Chong-Yang
    Wang, Bin
    JOURNAL OF INFRARED AND MILLIMETER WAVES, 2024, 43 (05) : 684 - 695
  • [26] The Power of Generative AI to Augment for Enhanced Skin Cancer Classification: A Deep Learning Approach
    Saeed, Mudassir
    Naseer, Asma
    Masood, Hassan
    Rehman, Shafiq Ur
    Gruhn, Volker
    IEEE ACCESS, 2023, 11 : 130330 - 130344
  • [27] Skin Cancer Classification With Deep Learning: A Systematic Review
    Wu, Yinhao
    Chen, Bin
    Zeng, An
    Pan, Dan
    Wang, Ruixuan
    Zhao, Shen
    FRONTIERS IN ONCOLOGY, 2022, 12
  • [28] A novel lightweight deep learning framework with knowledge distillation for efficient diabetic foot ulcer detection
    Amjad, Kamran
    Asif, Sohaib
    Waheed, Zafran
    Guo, Ying
    APPLIED SOFT COMPUTING, 2024, 167
  • [29] Toward Lightweight Diabetic Retinopathy Classification: A Knowledge Distillation Approach for Resource-Constrained Settings
    Islam, Niful
    Jony, Md. Mehedi Hasan
    Hasan, Emam
    Sutradhar, Sunny
    Rahman, Atikur
    Islam, Md. Motaharul
    APPLIED SCIENCES-BASEL, 2023, 13 (22):
  • [30] A lightweight approach for network intrusion detection in industrial cyber-physical systems based on knowledge distillation and deep metric learning
    Wang, Zhendong
    Li, Zeyu
    He, Daojing
    Chan, Sammy
    EXPERT SYSTEMS WITH APPLICATIONS, 2022, 206