Optimized Convolutional Neural Network Models for Skin Lesion Classification

Villa-Pulgarin, Juan Pablo; Ruales-Torres, Anderson Alberto; Arias-Garzon, 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

Abstract

Skin cancer is one of the most severe diseases, and medical imaging is among the main 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.

Más información

Título según WOS: ID WOS:000705060700003 Not found in local WOS DB
Título de la Revista: CMC-COMPUTERS MATERIALS & CONTINUA
Volumen: 70
Número: 2
Editorial: Tech Science Press
Fecha de publicación: 2022
Página de inicio: 2131
Página final: 2148
DOI:

10.32604/cmc.2022.019529

Notas: ISI