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 条
  • [31] DEEP LEARNING APPROACH FOR CLASSIFICATION OF BREAST CANCER
    Togacar, Mesut
    Ergen, Burhan
    2018 INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE AND DATA PROCESSING (IDAP), 2018,
  • [32] A Lightweight Knowledge-Distillation-Based Model for the Detection and Classification of Impacted Mandibular Third Molars
    Lei, Yujie
    Chen, Xiang
    Wang, Yunlong
    Tang, Rong
    Zhang, Baoping
    APPLIED SCIENCES-BASEL, 2023, 13 (17):
  • [33] Deep Residual Learning Image Recognition Model for Skin Cancer Disease Detection and Classification
    Al-Tuwaijari, Jamal Mustafa
    Yousir, Naeem Th.
    Alhammad, Nafea Ali Majeed
    Mostafa, Salama
    ACTA INFORMATICA PRAGENSIA, 2023, 12 (01) : 19 - 31
  • [34] A lightweight deep learning model for classification of synthetic aperture radar images
    Passah, Alicia
    Kandar, Debdatta
    ECOLOGICAL INFORMATICS, 2023, 77
  • [35] Deep Learning Based Lightweight Model for Brain Tumor Classification and Segmentation
    Andleeb, Ifrah
    Hussain, B. Zahid
    Ansari, Salik
    Ansari, Mohammad Samar
    Kanwal, Nadia
    Aslam, Asra
    ADVANCES IN COMPUTATIONAL INTELLIGENCE SYSTEMS, UKCI 2023, 2024, 1453 : 491 - 503
  • [36] A Lightweight Malware Detection Model Based on Knowledge Distillation
    Miao, Chunyu
    Kou, Liang
    Zhang, Jilin
    Dong, Guozhong
    MATHEMATICS, 2024, 12 (24)
  • [37] A Lightweight Deep Learning Model for EEG Classification Across Visual Stimuli
    Liu, Yi
    Goh, Steven
    Low, Tobias
    Quince, Zach
    Teragawa, Shoryu
    PROCEEDINGS OF THE 2024 27 TH INTERNATIONAL CONFERENCE ON COMPUTER SUPPORTED COOPERATIVE WORK IN DESIGN, CSCWD 2024, 2024, : 2900 - 2905
  • [38] Deep Learning-Based Transfer Learning for Classification of Skin Cancer
    Jain, Satin
    Singhania, Udit
    Tripathy, Balakrushna
    Nasr, Emad Abouel
    Aboudaif, Mohamed K.
    Kamrani, Ali K.
    SENSORS, 2021, 21 (23)
  • [39] Knowledge distillation-based deep learning classification network for peripheral blood leukocytes
    Leng, Bing
    Leng, Min
    Ge, Mingfeng
    Dong, Wenfei
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 75
  • [40] Collaborative deep semi-supervised learning with knowledge distillation for surface defect classification
    Manivannan, Siyamalan
    COMPUTERS & INDUSTRIAL ENGINEERING, 2023, 186