A new preprocessing approach to improve the performance of CNN-based skin lesion classification

Med Biol Eng Comput. 2021 May;59(5):1123-1131. doi: 10.1007/s11517-021-02355-5. Epub 2021 Apr 26.

Abstract

Skin lesion is one of the severe diseases which in many cases endanger the lives of patients on a worldwide extent. Early detection of disease in dermoscopy images can significantly increase the survival rate. However, the accurate detection of disease is highly challenging due to the following reasons: e.g., visual similarity between different classes of disease (e.g., melanoma and non-melanoma lesions), low contrast between lesions and skin, background noise, and artifacts. Machine learning models based on convolutional neural networks (CNN) have been widely used for automatic recognition of lesion diseases with high accuracy in comparison to conventional machine learning methods. In this research, we proposed a new preprocessing technique in order to extract the region of interest (RoI) of skin lesion dataset. We compare the performance of the most state-of-the-art CNN classifiers with two datasets which contain (1) raw, and (2) RoI extracted images. Our experiment results show that training CNN models by RoI extracted dataset can improve the accuracy of the prediction (e.g., InceptionResNetV2, 2.18% improvement). Moreover, it significantly decreases the evaluation (inference) and training time of classifiers as well.

Keywords: Convolutional neural network; Region of interest; Segmentation; Skin lesion.

MeSH terms

  • Dermoscopy
  • Humans
  • Machine Learning
  • Melanoma* / diagnostic imaging
  • Neural Networks, Computer
  • Skin Diseases*