Shallow Tuned DenseNet: A Lightweight Convolutional Neural Network Approach for Enhanced Skin Lesion Recognition
Skin lesions are irregularities on the skin, often harmless, though some may progress to skin cancer. Detecting them early is crucial for successful treatment and prevention of cancer development. Traditional detection methods involve significant expenses and testing. This study introduces an auto-mated system for classifying skin lesions using a Shallow Tuned DenseNet (STDN), a compact Convolutional Neural Network (CNN) model. We employ Transfer Learning (TL) by fine-tuning a pretrained DenseNet-169 model with additional lightweight layers, allowing the STDN to adapt to the PH<sup xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">2</sup> dataset's specific characteristics while remaining efficient. Initially, we utilize data augmentation techniques to address class imbalances within the dataset, improving model performance by reducing bias towards the majority class. The evaluation demonstrates a promising hundred percent accuracy with perfect F1-score and specificity. We attribute these results to effective transfer learning utilization, enhanced tuning, and partially to data augmentation, which mitigates overfitting and biases, facilitating robust feature extraction. Furthermore, we employ Grad-CAM (Gradient-weighted Class Activation Mapping) to visualize and interpret the model's predictions, enhancing trust and comprehension. The proposed STDN model makes a significant contribution by achieving high accuracy with a lightweight design, offering a potentially cost-effective and efficient solution for automated skin lesion detection, thereby aiding in early cancer prevention.
Authors
- Md. Rakib Hossain (ORCID: https://orcid.org/0000-0002-5613-4599)
- Anwar Hossain Efat
- Mohammad Faruk (ORCID: https://orcid.org/0000-0002-7115-9196)
- Azmain Yakin Srizon (ORCID: https://orcid.org/0000-0003-4674-0365)
- Md. Al Mamun
- S. M. Mahedy Hasan (ORCID: https://orcid.org/0009-0005-7885-8378)
- Tomal Tahsin Ahmmed
- Md. Nasif Khansur
Institutions
- Rajshahi University of Engineering and Technology (BD)
Publication Details
- Published
- 2024-09-12
- DOI
- https://doi.org/10.1109/peeiacon63629.2024.10800672
- Citations
- 2
- Primary Topic
- Cutaneous Melanoma Detection and Management
- Type
- article
- Field-Weighted Citation Impact
- 0.57