Light-Dermo: A Lightweight Pretrained Convolution Neural Network for the Diagnosis of Multiclass Skin Lesions

被引:12
作者
Baig, Abdul Rauf [1 ]
Abbas, Qaisar [1 ]
Almakki, Riyad [1 ]
Ibrahim, Mostafa E. A. [1 ]
AlSuwaidan, Lulwah [1 ]
Ahmed, Alaa E. S. [1 ]
机构
[1] Imam Mohammad Ibn Saud Islamic Univ IMSIU, Coll Comp & Informat Sci, Riyadh 11432, Saudi Arabia
关键词
pigmented skin lesions; deep learning; convolutional neural network; transfer learning; pretrained models; ShuffleNet; depthwise separable CNN; CLASSIFICATION;
D O I
10.3390/diagnostics13030385
中图分类号
R5 [内科学];
学科分类号
1002 ; 100201 ;
摘要
Skin cancer develops due to the unusual growth of skin cells. Early detection is critical for the recognition of multiclass pigmented skin lesions (PSLs). At an early stage, the manual work by ophthalmologists takes time to recognize the PSLs. Therefore, several "computer-aided diagnosis (CAD)" systems are developed by using image processing, machine learning (ML), and deep learning (DL) techniques. Deep-CNN models outperformed traditional ML approaches in extracting complex features from PSLs. In this study, a special transfer learning (TL)-based CNN model is suggested for the diagnosis of seven classes of PSLs. A novel approach (Light-Dermo) is developed that is based on a lightweight CNN model and applies the channelwise attention (CA) mechanism with a focus on computational efficiency. The ShuffleNet architecture is chosen as the backbone, and squeeze-and-excitation (SE) blocks are incorporated as the technique to enhance the original ShuffleNet architecture. Initially, an accessible dataset with 14,000 images of PSLs from seven classes is used to validate the Light-Dermo model. To increase the size of the dataset and control its imbalance, we have applied data augmentation techniques to seven classes of PSLs. By applying this technique, we collected 28,000 images from the HAM10000, ISIS-2019, and ISIC-2020 datasets. The outcomes of the experiments show that the suggested approach outperforms compared techniques in many cases. The most accurately trained model has an accuracy of 99.14%, a specificity of 98.20%, a sensitivity of 97.45%, and an F1-score of 98.1%, with fewer parameters compared to state-of-the-art DL models. The experimental results show that Light-Dermo assists the dermatologist in the better diagnosis of PSLs. The Light-Dermo code is available to the public on GitHub so that researchers can use it and improve it.
引用
收藏
页数:35
相关论文
共 48 条
[1]   Machine Learning Methods for Diagnosis of Eye-Related Diseases: A Systematic Review Study Based on Ophthalmic Imaging Modalities [J].
Abbas, Qaisar ;
Qureshi, Imran ;
Yan, Junhua ;
Shaheed, Kashif .
ARCHIVES OF COMPUTATIONAL METHODS IN ENGINEERING, 2022, 29 (06) :3861-3918
[2]   DermoDeep-A classification of melanoma-nevus skin lesions using multi-feature fusion of visual features and deep neural network [J].
Abbas, Qaisar ;
Celebi, M. Emre .
MULTIMEDIA TOOLS AND APPLICATIONS, 2019, 78 (16) :23559-23580
[3]   Multi-Class Skin Lesion Classification Using a Lightweight Dynamic Kernel Deep-Learning-Based Convolutional Neural Network [J].
Aldhyani, Theyazn H. H. ;
Verma, Amit ;
Al-Adhaileh, Mosleh Hmoud ;
Koundal, Deepika .
DIAGNOSTICS, 2022, 12 (09)
[4]   A Non-Invasive Interpretable Diagnosis of Melanoma Skin Cancer Using Deep Learning and Ensemble Stacking of Machine Learning Models [J].
Alfi, Iftiaz A. ;
Rahman, Md Mahfuzur ;
Shorfuzzaman, Mohammad ;
Nazir, Amril .
DIAGNOSTICS, 2022, 12 (03)
[5]   An enhanced technique of skin cancer classification using deep convolutional neural network with transfer learning models [J].
Ali, Md Shahin ;
Miah, Md Sipon ;
Haque, Jahurul ;
Rahman, Md Mahbubur ;
Islam, Md Khairul .
MACHINE LEARNING WITH APPLICATIONS, 2021, 5
[6]   ECG heartbeat arrhythmias classification: a comparison study between different types of spectrum representation and convolutional neural networks architectures [J].
Alqudah, Ali Mohammad ;
Qazan, Shoroq ;
Al-Ebbini, Lina ;
Alquran, Hiam ;
Abu Qasmieh, Isam .
JOURNAL OF AMBIENT INTELLIGENCE AND HUMANIZED COMPUTING, 2021, 13 (10) :4877-4907
[7]   An Enhanced Transfer Learning Based Classification for Diagnosis of Skin Cancer [J].
Anand, Vatsala ;
Gupta, Sheifali ;
Altameem, Ayman ;
Nayak, Soumya Ranjan ;
Poonia, Ramesh Chandra ;
Saudagar, Abdul Khader Jilani .
DIAGNOSTICS, 2022, 12 (07)
[8]   A comparative study of fourteen deep learning networks for multi skin lesion classification (MSLC) on unbalanced data [J].
Arora, Ginni ;
Dubey, Ashwani Kumar ;
Jaffery, Zainul Abdin ;
Rocha, Alvaro .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (11) :7989-8015
[9]  
Chattopadhyay A, 2018, Arxiv, DOI arXiv:1710.11063
[10]   The advantages of the Matthews correlation coefficient (MCC) over F1 score and accuracy in binary classification evaluation [J].
Chicco, Davide ;
Jurman, Giuseppe .
BMC GENOMICS, 2020, 21 (01)