Multi skin lesions classification using fine-tuning and data-augmentation applying NASNet

PeerJ Comput Sci. 2021 Jun 3:7:e371. doi: 10.7717/peerj-cs.371. eCollection 2021.

Abstract

Skin lesions are one of the typical symptoms of many diseases in humans and indicative of many types of cancer worldwide. Increased risks caused by the effects of climate change and a high cost of treatment, highlight the importance of skin cancer prevention efforts like this. The methods used to detect these diseases vary from a visual inspection performed by dermatologists to computational methods, and the latter has widely used automatic image classification applying Convolutional Neural Networks (CNNs) in medical image analysis in the last few years. This article presents an approach that uses CNNs with a NASNet architecture to recognize in a more accurate way, without segmentation, eight skin diseases. The model was trained end-to-end on Keras with augmented skin diseases images from the International Skin Imaging Collaboration (ISIC). The CNN architectures were initialized with weight from ImageNet, fine-tuned in order to discriminate well among the different types of skin lesions, and then 10-fold cross-validation was applied. Finally, some evaluation metrics are calculated as accuracy, sensitivity, and specificity and compare with other CNN trained architectures. This comparison shows that the proposed system offers higher accuracy results, with a significant reduction on the training paraments. To the best of our knowledge and based in the state-of-art recompiling in this work, the application of the NASNet architecture training with skin image lesion from ISIC archive for multi-class classification and evaluated by cross-validation, represents a novel skin disease classification system.

Keywords: Convolutional neural network; Image analysis; NASNet; Skin cancer; Skin diseases.

Associated data

  • figshare/10.6084/m9.figshare.13130090.v2

Grants and funding

This work was supported by the Secretaría Nacional de Ciencia, Tecnología e Innovación of Panama Republic. The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.