用注意力机制提升39类皮肤病变分类准确率,达93.46%。
An Attention-Guided Deep Learning Approach for Classifying 39 Skin Lesion Types
- 融合ECA与CBAM注意力模块增强模型特征提取能力。
- 视觉变换器+CBAM在39类病变上准确率达93.46%。
- 数据集涵盖5个公开源,适合医疗图像诊断研究者参考。
皮肤作为人体最大器官,易受多种皮肤病变影响,其中部分类型可作为严重疾病(如皮肤癌)的早期信号,亟需精准诊断。尽管多数病变不致命,但细微差异常难以肉眼识别。本研究构建了一个包含39类皮肤病变的综合性数据集,整合自五个公开数据集,并评估了MobileNetV2、Xception、InceptionV3、EfficientNetB1和Vision Transformer五种先进深度学习模型的性能。为提升模型精度与鲁棒性,引入Efficient Channel Attention(ECA)和Convolutional Block Attention Module(CBAM)注意力机制。多指标综合评估显示,集成CBAM的Vision Transformer表现最优,准确率93.46%、精确率94%、召回率93%、F1分数93%、特异度93.67%。该系统为临床医生提供高效准确的皮肤病变辅助诊断工具。相关数据与代码见https://github.com/akabircs/Skin-Lesions-Classification。
原文摘要 · Abstract (English)
The skin, as the largest organ of the human body, is vulnerable to a diverse array of conditions collectively known as skin lesions, which encompass various dermatoses. Diagnosing these lesions presents significant challenges for medical practitioners due to the subtle visual differences that are often imperceptible to the naked eye. While not all skin lesions are life-threatening, certain types can act as early indicators of severe diseases, including skin cancers, underscoring the critical need for timely and accurate diagnostic methods. Deep learning algorithms have demonstrated remarkable potential in facilitating the early detection and prognosis of skin lesions. This study advances the field by curating a comprehensive and diverse dataset comprising 39 categories of skin lesions, synthesized from five publicly available datasets. Using this dataset, the performance of five state-of-the-art deep learning models -- MobileNetV2, Xception, InceptionV3, EfficientNetB1, and Vision Transformer - is rigorously evaluated. To enhance the accuracy and robustness of these models, attention mechanisms such as the Efficient Channel Attention (ECA) and the Convolutional Block Attention Module (CBAM) are incorporated into their architectures. Comprehensive evaluation across multiple performance metrics reveals that the Vision Transformer model integrated with CBAM outperforms others, achieving an accuracy of 93.46%, precision of 94%, recall of 93%, F1-score of 93%, and specificity of 93.67%. These results underscore the significant potential of the proposed system in supporting medical professionals with accurate and efficient prognostic tools for diagnosing a broad spectrum of skin lesions. The dataset and code used in this study can be found at https://github.com/akabircs/Skin-Lesions-Classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。