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

Tanay Dwivedi, Brijesh Kumar Chaurasia () and Man Mohan Shukla
Additional contact information
Tanay Dwivedi: Pranveer Singh Institute of Technology
Brijesh Kumar Chaurasia: Pranveer Singh Institute of Technology
Man Mohan Shukla: Pranveer Singh Institute of Technology

International Journal of System Assurance Engineering and Management, 2024, vol. 15, issue 10, No 22, 5030-5055

Abstract: 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.

Keywords: Skin-cancer; Malignant; Benign; Transfer learning; Vision transformer; ViT-B16 (search for similar items in EconPapers)
Date: 2024
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
http://link.springer.com/10.1007/s13198-024-02521-6 Abstract (text/html)
Access to the full text of the articles in this series is restricted.

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:spr:ijsaem:v:15:y:2024:i:10:d:10.1007_s13198-024-02521-6

Ordering information: This journal article can be ordered from
http://www.springer.com/engineering/journal/13198

DOI: 10.1007/s13198-024-02521-6

Access Statistics for this article

International Journal of System Assurance Engineering and Management is currently edited by P.K. Kapur, A.K. Verma and U. Kumar

More articles in International Journal of System Assurance Engineering and Management from Springer, The Society for Reliability, Engineering Quality and Operations Management (SREQOM),India, and Division of Operation and Maintenance, Lulea University of Technology, Sweden
Bibliographic data for series maintained by Sonal Shukla () and Springer Nature Abstracting and Indexing ().