A Multiclass Skin Lesion classification approach using Transfer learning based convolutional Neural Network

K. Cauvery, P. C. Siddalingaswamy, Sameena Pathan, Noel D'Souza

Research output: Chapter in Book/Report/Conference proceedingConference contribution

12 Citations (Scopus)

Abstract

The rapid rise in skin diseases over the past decade has been a growing concern worldwide. Early detection, correct categorization, and accurate identification can result in the successful treatment of melanoma, thereby decreasing the morbidity and mortality rate. Thus, there is a significant need for a system that is capable of identifying skin diseases and precisely classifying them. The proposed work aims to develop a multi class classification system using transfer learning-based convolutional neural networks (CNN). In particular, the proposed solution classifies the dermoscopic images to 8 different categories namely Melanoma (MEL), Basal Cell Carcinoma (BCC), Actinic Keratosis (AK), Benign Keratosis (BKL), Dermatofibroma (DF), Vascular lesions (VASC) and Squamous Cell Carcinoma (SCC). Four state-of-art pre-trained models are used for this task. A functional model-based network is leveraged to embed these sub-models in a larger multi-headed neural network. This will allow the embedded model to be treated as a single large model. An ensemble approach, termed as blending, is employed to combine the predictions efficiently made by the sub-models. Additionally, a robust cropping strategy is implemented to deal with the uncropped images and their impact on the performance of the classifiers is investigated. The impact of applying blending technique to ensemble the pre-trained CNNs are investigated against the performance of the individual classifier. The proposed work is carried out on International Skin Imaging Collaboration (ISIC) 2019 dataset. In this work, the solution for task 1 of the challenge is presented and we obtained balanced multi-class accuracy of 81.2% on the dataset compiled from the original dataset.

Original languageEnglish
Title of host publicationProceedings of 2021 IEEE 7th International Conference on Bio Signals, Images and Instrumentation, ICBSII 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781665441261
DOIs
Publication statusPublished - 25-03-2021
Event7th IEEE International Conference on Bio Signals, Images and Instrumentation, ICBSII 2021 - Chennai, India
Duration: 25-03-202127-03-2021

Publication series

NameProceedings of 2021 IEEE 7th International Conference on Bio Signals, Images and Instrumentation, ICBSII 2021

Conference

Conference7th IEEE International Conference on Bio Signals, Images and Instrumentation, ICBSII 2021
Country/TerritoryIndia
CityChennai
Period25-03-2127-03-21

All Science Journal Classification (ASJC) codes

  • Artificial Intelligence
  • Signal Processing
  • Biomedical Engineering
  • Instrumentation

Fingerprint

Dive into the research topics of 'A Multiclass Skin Lesion classification approach using Transfer learning based convolutional Neural Network'. Together they form a unique fingerprint.

Cite this