Enhancing Automated Skin Cancer Detection Through Ensemble Learning and Multi-Head Attention Mechanisms

Skin cancer is one of the deadliest but most prevalent types of cancer; as such, early diagnosis is urgently required to improve patient outcomes. This work presents a collaborative deep learning model that classifies skin cancer with respect to three different networks: EfficientnetB۱, EfficientnetB۲, and EfficientnetV۲s on dermoscopic images. The proposed collaborative model has a multi-head attention mechanism, ensuring that this model has a better attention capability for improving its accuracy in the task of classification. The HAM۱۰k dataset provided the proposed model with a platform for fine tuning with transfer learning, along with some augmentation techniques to handle class imbalance challenges and feature variations of lesions. The results for the ensemble model combined with Multi-Head Attention were very high: an accuracy of ۹۷.۱۱%, and precision, recall, and F۱-score are also very high. These findings prove that our approach can dramatically improve automation in skin cancer detection. Therefore, it will be helpful in clinical dermatology for early diagnosis in medicine.

کلیدواژه ها:

Skin lesion classification ، Multi-head attention ، Ensemble learning ، Medical imaging