Skin lesion classification using transfer learning

Abstract

This work presents an essential module for the Transfer Learning approach's classification of melanoma skin lesions. Melanoma, a highly lethal form of skin cancer, poses a significant health threat globally. Image analysis plays a crucial role in enhancing the accuracy of malignant skin lesion classification. Although neural networks trained on extensive datasets have emerged as the latest solution, their scalability remains a challenge. This study proposes an efficient method for classifying skin lesions utilizing labelled data from open sources, leveraging EfficientNet as the foundational model to robustly capture discriminative features from diverse visual perspectives. Validation of the proposed algorithms relies on the classifier's capacity to distinguish between classes is measured by the Area Under the Receiver Operating Characteristic (AUC-ROC) curve. AUC-ROC score greater than zero denotes better classification performance. Our proposed model achieves an impressive score of 98.65%. In contrast to existing approaches, our method demonstrates swift and accurate identification and segmentation of melanoma skin lesions, showcasing its efficacy in advancing the field of skin lesion classification.