SkinSwinViT: A Lightweight Transformer-Based Method for Multiclass Skin Lesion Classification with Enhanced Generalization Capabilities

被引:0
|
作者
Tang, Kun [1 ]
Su, Jing [1 ]
Chen, Ruihan [1 ,2 ]
Huang, Rui [1 ]
Dai, Ming [1 ]
Li, Yongjiang [1 ]
机构
[1] Guangdong Ocean Univ, Sch Math & Comp, Zhanjiang 524008, Peoples R China
[2] Int Macau Inst Acad Res, Artificial Intelligence Res Inst, Taipa 999078, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2024年 / 14卷 / 10期
关键词
skin lesions; ISIC2018; transformer; data enhancement; multiclassification; MACHINE; NETWORK;
D O I
10.3390/app14104005
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
In recent decades, skin cancer has emerged as a significant global health concern, demanding timely detection and effective therapeutic interventions. Automated image classification via computational algorithms holds substantial promise in significantly improving the efficacy of clinical diagnoses. This study is committed to mitigating the challenge of diagnostic accuracy in the classification of multiclass skin lesions. This endeavor is inherently formidable owing to the resemblances among various lesions and the constraints associated with extracting precise global and local image features within diverse dimensional spaces using conventional convolutional neural network methodologies. Consequently, this study introduces the SkinSwinViT methodology for skin lesion classification, a pioneering model grounded in the Swin Transformer framework featuring a global attention mechanism. Leveraging the inherent cross-window attention mechanism within the Swin Transformer architecture, the model adeptly captures local features and interdependencies within skin lesion images while additionally incorporating a global self-attention mechanism to discern overarching features and contextual information effectively. The evaluation of the model's performance involved the ISIC2018 challenge dataset. Furthermore, data augmentation techniques augmented training dataset size and enhanced model performance. Experimental results highlight the superiority of the SkinSwinViT method, achieving notable metrics of accuracy, recall, precision, specificity, and F1 score at 97.88%, 97.55%, 97.83%, 99.36%, and 97.79%, respectively.
引用
收藏
页数:20
相关论文
共 50 条
  • [1] Assist-Dermo: A Lightweight Separable Vision Transformer Model for Multiclass Skin Lesion Classification
    Abbas, Qaisar
    Daadaa, Yassine
    Rashid, Umer
    Ibrahim, Mostafa E. A.
    DIAGNOSTICS, 2023, 13 (15)
  • [2] SkinDistilViT: Lightweight Vision Transformer for Skin Lesion Classification
    Lungu-Stan, Vlad-Constantin
    Cercel, Dumitru-Clementin
    Pop, Florin
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2023, PT I, 2023, 14254 : 268 - 280
  • [3] DeLTran15: A Deep Lightweight Transformer-Based Framework for Multiclass Classification of Disaster Posts on X
    Saleem, Saima
    Hasan, Nabeela
    Khattar, Anuradha
    Jain, Priti Rai
    Gupta, Tarun Kumar
    Mehrotra, Monica
    IEEE ACCESS, 2024, 12 : 153676 - 153693
  • [4] LTUNet: A Lightweight Transformer-Based UNet with Multi-scale Mechanism for Skin Lesion Segmentation
    Guo, Huike
    Zhang, Han
    Li, Minghe
    Quan, Xiongwen
    ARTIFICIAL INTELLIGENCE, CICAI 2023, PT II, 2024, 14474 : 147 - 158
  • [5] Multiclass skin lesion classification in dermoscopic images using swin transformer model
    Ayas, Selen
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (09): : 6713 - 6722
  • [6] Multiclass skin lesion classification in dermoscopic images using swin transformer model
    Selen Ayas
    Neural Computing and Applications, 2023, 35 : 6713 - 6722
  • [7] LiwTERM: A Lightweight Transformer-based Model for Dermatological Multimodal Lesion Detection
    Souza, Luis A., Jr.
    Pacheco, Andre G. C.
    de Angelo, Gabriel G.
    Oliveira-Santos, Thiago
    Palm, Christoph
    Papa, Joao P.
    2024 37TH SIBGRAPI CONFERENCE ON GRAPHICS, PATTERNS AND IMAGES, SIBGRAPI 2024, 2024, : 229 - 234
  • [8] Enhanced deep bottleneck transformer model for skin lesion classification*
    Nakai, Katsuhiro
    Chen, Yen-Wei
    Han, Xian-Hua
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 78
  • [9] A transformer-based lightweight method for multiple-object tracking
    Wan, Qin
    Ge, Zhu
    Yang, Yang
    Shen, Xuejun
    Zhong, Hang
    Zhang, Hui
    Wang, Yaonan
    Wu, Di
    IET IMAGE PROCESSING, 2024, 18 (09) : 2329 - 2345
  • [10] Multiclass Skin Lesion Classification Using a Novel Lightweight Deep Learning Framework for Smart Healthcare
    Hoang, Long
    Lee, Suk-Hwan
    Lee, Eung-Joo
    Kwon, Ki-Ryong
    APPLIED SCIENCES-BASEL, 2022, 12 (05):