Skin lesion classification using CNNs with grouping of multi-scale attention and class-specific loss weighting

Comput Methods Programs Biomed. 2022 Nov:226:107166. doi: 10.1016/j.cmpb.2022.107166. Epub 2022 Sep 30.

Abstract

As one of the most common cancers globally, the incidence of skin cancer has been rising. Dermoscopy-based classification has become the most effective method for the diagnosis of skin lesion types due to its accuracy and non-invasive characteristics, which plays a significant role in reducing mortality. Although a great breakthrough of the task of skin lesion classification has been made with the application of convolutional neural network, the inter-class similarity and intra-class variation in skin lesions images, the high class imbalance of the dataset and the lack of ability to focus on the lesion area all affect the classification results of the model. In order to solve these problems, on the one hand, we use the grouping of multi-scale attention blocks (GMAB) to extract multi-scale fine-grained features so as to improve the model's ability to focus on the lesion area. On the other hand, we adopt the method of class-specific loss weighting for the problem of category imbalance. In this paper, we propose a deep convolution neural network dermatoscopic image classification method based on the grouping of multi-scale attention blocks and class-specific loss weighting. We evaluated our model on the HAM10000 dataset, and the results showed that the ACC and AUC of the proposed method were 91.6% and 97.1% respectively, which can achieve good results in dermatoscopic classification tasks.

Keywords: Attention mechanism; Class imbalance; Deep learning; Dermoscopy; Skin lesion classification.

MeSH terms

  • Attention
  • Dermoscopy / methods
  • Humans
  • Neural Networks, Computer
  • Skin Diseases* / diagnosis
  • Skin Neoplasms* / diagnostic imaging
  • Skin Neoplasms* / pathology