BF2SkNet: best deep learning features fusion-assisted framework for multiclass skin lesion classification

Muhammad Ajmal, Muhammad Attique Khan, Tallha Akram, Abdullah Alqahtani, Majed Alhaisoni, Ammar Armghan, Sara A. Althubiti, Fayadh Alenezi

Research output: Contribution to journalArticlepeer-review

30 Scopus citations

Abstract

The convolutional neural network showed considerable success in medical imaging with explainable AI for cancer detection and recognition. However, the irrelevant and large number of features increases the computational time and decreases the accuracy. This work proposes a deep learning and fuzzy entropy slime mould algorithm-based architecture for multiclass skin lesion classification. In the first step, we employed the data augmentation technique to increase the training data and further utilized it for training two fine-tuned deep learning models such as Inception-ResNetV2 and NasNet Mobile. Then, we used transfer learning on augmented datasets to train both models and obtained two feature vectors from newly fine-tuned models. Later, we applied a fuzzy entropy slime mould algorithm on both vectors to get optimal features that are finally fused using the Serial-Threshold fusion technique and classified using several machine learning classifiers. Eventually, the explainable AI technique named Gradcam opted for the visualization of the lesion region. The experimental process was conducted on two datasets, such as HAM10000 and ISIC 2018, and achieved 97.1 and 90.2% accuracy, better than the other techniques.

Original languageEnglish
Pages (from-to)22115-22131
Number of pages17
JournalNeural Computing and Applications
Volume35
Issue number30
DOIs
StatePublished - Oct 2023

Keywords

  • Artificial intelligence
  • Deep learning
  • Dermoscopy
  • Features fusion
  • Features optimization
  • Machine learning
  • Skin cancer

Fingerprint

Dive into the research topics of 'BF2SkNet: best deep learning features fusion-assisted framework for multiclass skin lesion classification'. Together they form a unique fingerprint.

Cite this