PLoS ONE (Jan 2024)

A model for skin cancer using combination of ensemble learning and deep learning.

  • Mehdi Hosseinzadeh,
  • Dildar Hussain,
  • Firas Muhammad Zeki Mahmood,
  • Farhan A Alenizi,
  • Amirhossein Noroozi Varzeghani,
  • Parvaneh Asghari,
  • Aso Darwesh,
  • Mazhar Hussain Malik,
  • Sang-Woong Lee

DOI
https://doi.org/10.1371/journal.pone.0301275
Journal volume & issue
Vol. 19, no. 5
p. e0301275

Abstract

Read online

Skin cancer has a significant impact on the lives of many individuals annually and is recognized as the most prevalent type of cancer. In the United States, an estimated annual incidence of approximately 3.5 million people receiving a diagnosis of skin cancer underscores its widespread prevalence. Furthermore, the prognosis for individuals afflicted with advancing stages of skin cancer experiences a substantial decline in survival rates. This paper is dedicated to aiding healthcare experts in distinguishing between benign and malignant skin cancer cases by employing a range of machine learning and deep learning techniques and different feature extractors and feature selectors to enhance the evaluation metrics. In this paper, different transfer learning models are employed as feature extractors, and to enhance the evaluation metrics, a feature selection layer is designed, which includes diverse techniques such as Univariate, Mutual Information, ANOVA, PCA, XGB, Lasso, Random Forest, and Variance. Among transfer models, DenseNet-201 was selected as the primary feature extractor to identify features from data. Subsequently, the Lasso method was applied for feature selection, utilizing diverse machine learning approaches such as MLP, XGB, RF, and NB. To optimize accuracy and precision, ensemble methods were employed to identify and enhance the best-performing models. The study provides accuracy and sensitivity rates of 87.72% and 92.15%, respectively.