Journal Article

Boosting Skin Cancer Classification: A Multi-Scale Attention and Ensemble Approach with Vision Transformers

Luo, Suhuai, Yang, Guang, Greer, Peter

2025

Overview

Skin cancer is a significant global health concern, with melanoma being the most dangerous form, responsible for the majority of skin cancer-related deaths. Early detection of skin cancer is critical, as it can drastically improve survival rates. While deep learning models have achieved impressive results in skin cancer classification, there remain challenges in accurately distinguishing between benign and malignant lesions. In this study, we introduce a novel multi-scale attention-based performance booster inspired by the Vision Transformer (ViT) architecture, which enhances the accuracy of both ViT and convolutional neural network (CNN) models. By leveraging attention maps to identify discriminative regions within skin lesion images, our method improves the models’ focus on diagnostically relevant areas. Additionally, we employ ensemble learning techniques to combine the outputs of several deep learning models using majority voting. Our skin cancer classifier, consisting of ViT and EfficientNet models, achieved a classification accuracy of 95.05% on the ISIC2018 dataset, outperforming individual models. The results demonstrate the effectiveness of integrating attention-based multi-scale learning and ensemble methods in skin cancer classification.

• Publisher: MDPI AG,MDPI
• Subject: Accuracy / Algorithms / Artificial intelligence / Attention / Cancer / Classification / Comparative analysis / Computational linguistics / Datasets / Deep Learning / Dermatology / Diagnosis / Electric transformers / Humans / Identification and classification / image processing / Innovations / Language processing / Machine learning / Machine vision / Medical prognosis / Melanoma / Melanoma - classification / Melanoma - diagnosis / Methods / Natural language interfaces / Neural networks / Neural Networks, Computer / Skin cancer / skin cancer classification / Skin Neoplasms - classification / Skin Neoplasms - diagnosis / Skin Neoplasms - pathology / Support vector machines / transformer