Boosting the performance of pretrained CNN architecture on dermoscopic pigmented skin lesion classification

Skin Res Technol. 2023 Nov;29(11):e13505. doi: 10.1111/srt.13505.

Abstract

Background: Pigmented skin lesions (PSLs) pose medical and esthetic challenges for those affected. PSLs can cause skin cancers, particularly melanoma, which can be life-threatening. Detecting and treating melanoma early can reduce mortality rates. Dermoscopic imaging offers a noninvasive and cost-effective technique for examining PSLs. However, the lack of standardized colors, image capture settings, and artifacts makes accurate analysis challenging. Computer-aided diagnosis (CAD) using deep learning models, such as convolutional neural networks (CNNs), has shown promise by automatically extracting features from medical images. Nevertheless, enhancing the CNN models' performance remains challenging, notably concerning sensitivity.

Materials and methods: In this study, we aim to enhance the classification performance of selected pretrained CNNs. We use the 2019 ISIC dataset, which presents eight disease classes. To achieve this goal, two methods are applied: resolution of the dataset imbalance challenge through augmentation and optimization of the training hyperparameters via Bayesian tuning.

Results: The performance improvement was observed for all tested pretrained CNNs. The Inception-V3 model achieved the best performance compared to similar results, with an accuracy of 96.40% and an AUC of 0.98.

Conclusion: According to the study, classification performance was significantly enhanced by augmentation and Bayesian hyperparameter tuning.

Keywords: Bayesian tuning; augmentation; convolutional neural network; hyper-parameter optimization; pigmented skin lesion; skin cancer.

MeSH terms

  • Bayes Theorem
  • Diagnosis, Computer-Assisted / methods
  • Humans
  • Melanoma* / pathology
  • Neural Networks, Computer
  • Pigmentation Disorders*
  • Skin Neoplasms* / pathology