An Explainable Deep Learning Framework for Multi-Class Skin Lesion Classification While Resolving Class Imbalance
Skin lesions are indicative of various skin diseases, including serious conditions like melanoma, which can be life-threatening if not detected early. Automated systems for analyzing dermatoscopic images have shown promise in improving diagnostic accuracy. However, many existing studies rely on pre-trained convolutional neural network (CNN) architectures without customization, leading to potential biases and suboptimal performance. In this study, we propose a customized DenseNet121 architecture tailored for skin lesion classification. We address class imbalance issues through data augmentation and fine-tune the model to extract fine-grained features. Our method surpasses other cutting-edge models in terms of accuracy, precision, recall, specificity, sensitivity, and root mean square (ROC) values. Furthermore, we utilise Gradient-weighted Class Activation Mapping (Grad-CAM) to visually represent the decision-making process of the model. The next study will prioritise the validation of the model using a wide range of datasets and the integration of lesion-related information to enhance its performance.
Authors
- Md. Rakib Hossain (ORCID: https://orcid.org/0000-0002-5613-4599)
- Anwar Hossain Efat
- Md. Farukuzzaman Faruk (ORCID: https://orcid.org/0000-0001-6446-1066)
- Azmain Yakin Srizon (ORCID: https://orcid.org/0000-0003-4674-0365)
- Md. Al Mamun (ORCID: https://orcid.org/0000-0003-0330-9241)
- Israt Jahan (ORCID: https://orcid.org/0000-0002-1384-5657)
- S. M. Mahedy Hasan (ORCID: https://orcid.org/0009-0005-7885-8378)
Institutions
- Rajshahi University of Engineering and Technology (BD)
Publication Details
- Published
- 2024-09-12
- DOI
- https://doi.org/10.1109/peeiacon63629.2024.10800046
- Citations
- 1
- Primary Topic
- Cutaneous Melanoma Detection and Management
- Type
- article
- Field-Weighted Citation Impact
- 0.28