Vision Transformer for Skin Cancer Identification Based on Contrastive Learning and Adaptive-Scale Fragmentation

Author:

Naeem Muhammad Ahtsam1,Yang Shangming1,Sharif Abida2,Saleem Muhammad Asim3,Sharif Muhammad Imran4

Affiliation:

1. University of Electronic Science and Technology of China

2. COMSATS University Islamabad

3. Chulalongkorn University

4. Kansas State University

Abstract

Abstract

The approach of image processing and deep learning has shown to be a breakthrough in the field of medical image diagnosis such as dermoscopic image analysis for skin cancer recognition and their classification. Skin cancer cases are increasing every year and pose a significant threat for health. In recent studies, convolutional neural network (CNN) has accomplished remarkable success in classifying skin cancer images. CNN is limited to extracting features from minor objects from input dermoscopic image and fails to pinpoint significant regions. Consequently, the researchers of this study have utilized vision transformers (VIT), known for their robust performance in conventional classification assignments. The self-attention mechanism (SAM) aims to enhance the significance of pivotal characteristics while modifying the influence of noise-inducing features. Specifically, an enhanced transformer network architecture has been introduced in this context. To assess its effectiveness, several enhancements have been applied to the model. Initially, a ViT network is implemented to evaluate its efficacy in identifying skin cancer. Subsequently, Adaptive-scale image fragmentation is utilized to sequentially process the image, emphasizing adaptive-scale features through patch embedding. Furthermore, contrastive learning is employed to ensure that similar skin cancer data is encoded differently, aiming for distinct encoding outcomes for different data. Skin cancer dataset namely ISIC 2019 is retrieved in this study, locally accessible at Kaggle’s official website. This dataset consists of dermoscopic images of skin cancer having several types: dermatofibroma, melanoma, actinic keratosis, basal cell carcinoma, nevus, vascular lesion, and pigmented benign keratosis. The ViT model has achieved 99.66% accuracy, 94.85% precision, 93.74% recall, and 94.52% f1-score. Three deep learning models Inception V3, MobileNet, and ResNet-50 were also applied with transfer learning approach as comparison to proposed ViT model for performance evaluation that resulted in 72%, 94.3, and 89% accuracies, respectively. The transformer network has shown remarkable success in natural language processing and in the domain of image analysis. These achievements establish a solid groundwork to classify skin cancer using multimodal data. This paper is confident to captivate the attention of medical researchers, computer engineers, dermatologists, and scholars across various related disciplines. Its insights promise to offer enhanced convenience for patients in their respective fields.

Publisher

Research Square Platform LLC

Reference46 articles.

1. Cancer causes and treatments;࿼A. Saini M;Int J Pharm Sci Res,2020

2. DNA damage, mutagenesis and cancer;Basu ࿼A K;Int J Mol Sci,2018

3. ࿼“Cancer Stat Facts: Common Cancer Sites,” https://seer.cancer.gov/statfacts/html/common.html.

4. Cancer Stat Facts: Melanoma of the Skin;࿼National Cancer

5. Mechanisms of UV-induced mutations and skin cancer;Pfeifer ࿼G P;Genome Instab Dis,2020

同舟云学术

1.学者识别学者识别

2.学术分析学术分析

3.人才评估人才评估

"同舟云学术"是以全球学者为主线,采集、加工和组织学术论文而形成的新型学术文献查询和分析系统,可以对全球学者进行文献检索和人才价值评估。用户可以通过关注某些学科领域的顶尖人物而持续追踪该领域的学科进展和研究前沿。经过近期的数据扩容,当前同舟云学术共收录了国内外主流学术期刊6万余种,收集的期刊论文及会议论文总量共计约1.5亿篇,并以每天添加12000余篇中外论文的速度递增。我们也可以为用户提供个性化、定制化的学者数据。欢迎来电咨询!咨询电话:010-8811{复制后删除}0370

www.globalauthorid.com

TOP

Copyright © 2019-2024 北京同舟云网络信息技术有限公司
京公网安备11010802033243号  京ICP备18003416号-3