MetaAttention model: a new approach for skin lesion diagnosis using AB features and attention mechanism

Biomed Phys Eng Express. 2023 May 12;9(4). doi: 10.1088/2057-1976/acd1f0.

Abstract

Ozone depletion has always been a hot crisis around the globe. Its consequence is the increase in ultraviolet radiation at the surface in many regions and countries, which then causes danger to the human immune system, eyes, and especially skin - the part that is directly exposed most to the sunlight. According to the World Health Organization, the number of cases of skin cancer is higher than that of breast, prostate, and lung cancer combined. Therefore, there has been a lot of research to apply deep learning models in solving the problem of skin cancer classification. This paper proposes a novel approach, named MetaAttention, aimed at improving the performance of transfer learning models for skin lesion classification. The method combines image features with patients' metadata features using the attention mechanism, incorporating clinical knowledge related to ABCD signals to better distinguish melanoma cell carcinoma disease, which has long been a significant challenge for researchers. The experimental results indicate that the proposed approach outperforms the state-of-the-art method, EfficientNet-B4, achieving an accuracy of 89.9% with Scale-dot product MetaAttention and 90.63% with Additive MetaAttention. The method has the potential to support dermatologists in diagnosing skin lesions effectively and efficiently. Furthermore, with larger datasets, our method could be further fine-tuned to achieve even better performance on a broader range of labels.

Keywords: attention mechanism; deep learning; image processing; skin cancer; transfer learning.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Humans
  • Male
  • Melanoma* / diagnosis
  • Skin / pathology
  • Skin Diseases*
  • Skin Neoplasms* / diagnosis
  • Skin Neoplasms* / pathology
  • Ultraviolet Rays