Details

Autor(en) / Beteiligte

• Qureshi, Mohammad Naved
• Umar, Mohammad Sarosh
• Shahab, Sana

Titel

A Transfer-Learning-Based Novel Convolution Neural Network for Melanoma Classification

Ist Teil von

• Computers (Basel), 2022-05, Vol.11 (5), p.64

Ort / Verlag

Basel: MDPI AG

Erscheinungsjahr

2022

Quelle

EZB Free E-Journals

Beschreibungen/Notizen

• Skin cancer is one of the most common human malignancies, which is generally diagnosed by screening and dermoscopic analysis followed by histopathological assessment and biopsy. Deep-learning-based methods have been proposed for skin lesion classification in the last few years. The major drawback of all methods is that they require a considerable amount of training data, which poses a challenge for classifying medical images as limited datasets are available. The problem can be tackled through transfer learning, in which a model pre-trained on a huge dataset is utilized and fine-tuned as per the problem domain. This paper proposes a new Convolution neural network architecture to classify skin lesions into two classes: benign and malignant. The Google Xception model is used as a base model on top of which new layers are added and then fine-tuned. The model is optimized using various optimizers to achieve the maximum possible performance gain for the classifier output. The results on ISIC archive data for the model achieved the highest training accuracy of 99.78% using Adam and LazyAdam optimizers, validation and test accuracy of 97.94% and 96.8% using RMSProp, and on the HAM10000 dataset utilizing the RMSProp optimizer, the model achieved the highest training and prediction accuracy of 98.81% and 91.54% respectively, when compared to other models.