Back to Search Start Over

Lightweight vision image transformer (LViT) model for skin cancer disease classification.

Authors :
Dwivedi, Tanay
Chaurasia, Brijesh Kumar
Shukla, Man Mohan
Source :
International Journal of Systems Assurance Engineering & Management; Oct2024, Vol. 15 Issue 10, p5030-5055, 26p
Publication Year :
2024

Abstract

Skin cancer (SC) is a lethal disease not only in India but also in the world; there are more than a million cases of melanoma per year in India. Early detection of skin cancer through accurate classification of skin lesions is essential for effective treatment. Visual inspection by clinical screening, dermoscopy, or histological tests is strongly emphasised in today's skin cancer diagnosis. It can be challenging to determine the kind of skin cancer, especially in the early stages, due to the resemblance among cancer types. However, the precise classification of skin lesions could be time-consuming and challenging for dermatologists. To address these issues, we propose transfer learning to accurately classify skin lesions into several forms of skin cancer using a lightweight B-16 Vision Image Transformer model (LViT). An extensive dataset is used in the experiment to verify the efficiency of the proposed LViT model. The LViT model can classify skin cancer with high accuracy, sensitivity, and specificity and generalise favourably to new images. The proposed model has a 93.17% accuracy rating for classifying SC images over 25 epochs and a remarkable accuracy of 95.82% over 100 epochs. The proposed LViT model is lightweight, requires minimal processing resources, and achieves good accuracy on small and enormous data sets. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
09756809
Volume :
15
Issue :
10
Database :
Complementary Index
Journal :
International Journal of Systems Assurance Engineering & Management
Publication Type :
Academic Journal
Accession number :
180105019
Full Text :
https://doi.org/10.1007/s13198-024-02521-6