Back to Search Start Over

Optimized Convolutional Neural Network Models for Skin Lesion Classification.

Authors :
Villa-Pulgarin, Juan Pablo
Ruales-Torres, Anderson Alberto
Arias-Garzón, Daniel
Bravo-Ortiz, Mario Alejandro
Arteaga-Arteaga, Harold Brayan
Mora-Rubio, Alejandro
Alzate-Grisales, Jesus Alejandro
Mercado-Ruiz, Esteban
Hassaballah, M.
Orozco-Arias, Simon
Cardona-Morales, Oscar
Tabares-Soto, Reinel
Source :
Computers, Materials & Continua; 2022, Vol. 70 Issue 2, p2131-2148, 18p
Publication Year :
2022

Abstract

Skin cancer is one of themost severe diseases, andmedical imaging is among themain tools for cancer diagnosis. The images provide information on the evolutionary stage, size, and location of tumor lesions. This paper focuses on the classification of skin lesion images considering a framework of four experiments to analyze the classification performance of Convolutional Neural Networks (CNNs) in distinguishing different skin lesions. The CNNs are based on transfer learning, taking advantage of ImageNet weights. Accordingly, in each experiment, different workflow stages are tested, including data augmentation and fine-tuning optimization. Three CNN models based on DenseNet-201, Inception-ResNet-V2, and Inception-V3 are proposed and compared using the HAM10000 dataset. The results obtained by the three models demonstrate accuracies of 98%, 97%, and 96%, respectively. Finally, the best model is tested on the ISIC 2019 dataset showing an accuracy of 93%. The proposed methodology using CNN represents a helpful tool to accurately diagnose skin cancer disease. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
15462218
Volume :
70
Issue :
2
Database :
Complementary Index
Journal :
Computers, Materials & Continua
Publication Type :
Academic Journal
Accession number :
152705391
Full Text :
https://doi.org/10.32604/cmc.2022.019529