A Lightweight Attention-Based CNN with Saliency-Map Interpretability for Skin Cancer Detection
Organizations: Dept. of Computer Science and Engineering, International Standard University, Dhaka, Bangladesh · Dept. of Public Health and Life Sciences, Canadian University of Bangladesh, Dhaka, Bangladesh · Faculty of Science & Engineering, Abo Akademi University, Turku, Finland
Abstract
Skin cancer is among the most prevalent and life threatening forms of cancer worldwide. Early and accurate diagnosis significantly improves patient survival rates. However, manual dermoscopic assessment remains challenging due to visual similarity. This study proposes an Explainable AI-driven framework for automated binary classification of skin lesions using a lightweight attention-based convolutional neural network (CNN-Attention). The model integrates channel attention to focus on informative regions of dermoscopic images while maintaining low computational complexity. A publicly available dataset comprising 2,637 training and 660 test images was used, with balanced class weights. The proposed model achieved 87 percent accuracy with 1.95 million parameters and 7.45 MB of storage, matching the best baseline (MobileNet) while using approximately half as many parameters and half as much storage. ROC AUC and PR AUC scores reached 0.9475 and 0.9430, respectively. Additionally, saliency maps were incorporated as an explainability technique to visualize regions that influence model decisions. Comparative analysis with VGG16,VGG19,ResNet101,and MobileNet demonstrates the proposed models effectiveness in balancing classification results, computational efficiency, and interpretability.
Figures & tables
| Class | Precision | Recall | F1-Score |
|---|---|---|---|
| Benign | 0.91 | 0.85 | 0.88 |
| Malignant | 0.83 | 0.89 | 0.86 |
| Accuracy | 0.87 | ||
| Class | PR-AUC Score |
|---|---|
| Benign | 0.9563 |
| Malignant | 0.9296 |
| Average | 0.9430 |
| Model | Accuracy | Parameters (M) | Size (MB) |
|---|---|---|---|
| VGG16 | 0.84 | 15.11 | 57.64 |
| VGG19 | 0.84 | 20.42 | 77.90 |
| ResNet101 | 0.83 | 43.84 | 167.25 |
| MobileNet | 0.87 | 3.89 | 14.84 |
| CNN-Attention | 0.87 | 1.95 | 7.45 |
Skin Lesion Classification
Skin lesion classification uses medical images and clinical context to identify lesion types and estimate malignancy, supporting diagnosis and triage.
See all 55 papers