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 条
  • [1] Ensemble Learning of Lightweight Deep Learning Models Using Knowledge Distillation for Image Classification
    Kang, Jaeyong
    Gwak, Jeonghwan
    MATHEMATICS, 2020, 8 (10)
  • [2] DeepSkin: A Deep Learning Approach for Skin Cancer Classification
    Gururaj, H. L.
    Manju, N.
    Nagarjun, A.
    Aradhya, V. N. Manjunath
    Flammini, Francesco
    IEEE ACCESS, 2023, 11 : 50205 - 50214
  • [3] Lightweight Network Traffic Classification Model Based on Knowledge Distillation
    Wu, Yanhui
    Zhang, Meng
    WEB INFORMATION SYSTEMS ENGINEERING - WISE 2021, PT II, 2021, 13081 : 107 - 121
  • [4] SkCanNet: A Deep Learning based Skin Cancer Classification Approach
    Onesimu J.A.
    Nair V.U.
    Sagayam M.K.
    Eunice J.
    Wahab M.H.A.
    Sudin N.
    Annals of Emerging Technologies in Computing, 2023, 7 (04) : 35 - 45
  • [5] A lightweight deep convolutional neural network model for skin cancer image classification
    Tuncer, Turker
    Barua, Prabal Datta
    Tuncer, Ilknur
    Dogan, Sengul
    Acharya, U. Rajendra
    APPLIED SOFT COMPUTING, 2024, 162
  • [6] Cutting Weights of Deep Learning Models for Heart Sound Classification: Introducing a Knowledge Distillation Approach
    Song, Zikai
    Zhu, Lixian
    Wang, Yiyan
    Sun, Mengkai
    Qian, Kun
    Hu, Bin
    Yamamoto, Yoshiharu
    Schuller, Bjorn W.
    2023 45TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE & BIOLOGY SOCIETY, EMBC, 2023,
  • [7] A novel approach for skin lesion symmetry classification with a deep learning model
    Talavera-Martinez, Lidia
    Bibiloni, Pedro
    Giacaman, Aniza
    Taberner, Rosa
    Del Pozo Hernando, Luis Javier
    Gonzalez-Hidalgo, Manuel
    COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 145
  • [8] Lightweight Traffic Classification Model Based on Deep Learning
    Sun, Chongxin
    Chen, Bo
    Bu, Youjun
    Zhang, Surong
    Zhang, Desheng
    Jiang, Bingbing
    WIRELESS COMMUNICATIONS & MOBILE COMPUTING, 2022, 2022
  • [9] Deep Learning Model with Atrous Convolutions for Improving Skin Cancer Classification
    Kaur, Ranpreet
    GholamHosseini, Hamid
    IMAGE AND VIDEO TECHNOLOGY, PSIVT 2023, 2024, 14403 : 422 - 435
  • [10] A New Approach using Deep Learning and Reinforcement Learning in HealthCare: Skin Cancer Classification
    Yousra, Dahdouh
    Abdelhakim, Anouar Boudhir
    Mohamed, Ben Ahmed
    INTERNATIONAL JOURNAL OF ELECTRICAL AND COMPUTER ENGINEERING SYSTEMS, 2023, 14 (05) : 557 - 564