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 条
[11]   On the Automatic Detection and Classification of Skin Cancer Using Deep Transfer Learning [J].
Fraiwan, Mohammad ;
Faouri, Esraa .
SENSORS, 2022, 22 (13)
[12]   Skin lesion classification using ensembles of multi -resolution EfficientNets with meta data [J].
Gessert, Nils ;
Nielsen, Maximilian ;
Shaikh, Mohsin ;
Werner, Rene ;
Schlaefer, Alexander .
METHODSX, 2020, 7
[13]   Multi-class multi-level classification algorithm for skin lesions classification using machine learning techniques [J].
Hameed, Nazia ;
Shabut, Antesar M. ;
Ghosh, Miltu K. ;
Hossain, M. A. .
EXPERT SYSTEMS WITH APPLICATIONS, 2020, 141
[14]   Classification of the Clinical Images for Benign and Malignant Cutaneous Tumors Using a Deep Learning Algorithm [J].
Han, Seung Seog ;
Kim, Myoung Shin ;
Lim, Woohyung ;
Park, Gyeong Hun ;
Park, Ilwoo ;
Chang, Sung Eun .
JOURNAL OF INVESTIGATIVE DERMATOLOGY, 2018, 138 (07) :1529-1538
[15]   Assisted deep learning framework for multi-class skin lesion classification considering a binary classification support [J].
Harangi, Balazs ;
Baran, Agnes ;
Hajdu, Andras .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2020, 62
[16]  
Hasan Md Kamrul, 2022, Informatics in Medicine Unlocked, DOI 10.1016/j.imu.2021.100819
[17]   Classification of Skin Lesions into Seven Classes Using Transfer Learning with AlexNet [J].
Hosny, Khalid M. ;
Kassem, Mohamed A. ;
Fouad, Mohamed M. .
JOURNAL OF DIGITAL IMAGING, 2020, 33 (05) :1325-1334
[18]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
[19]   Automated multi-class classification of skin lesions through deep convolutional neural network with dermoscopic images [J].
Iqbal, Imran ;
Younus, Muhammad ;
Walayat, Khuram ;
Kakar, Mohib Ullah ;
Ma, Jinwen .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2021, 88
[20]   Deep learning architecture using transfer learning for classification of skin lesions [J].
Jasil, S. P. Godlin ;
Ulagamuthalvi, V. .
JOURNAL OF AMBIENT INTELLIGENCE AND HUMANIZED COMPUTING, 2021,