arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.18554cs.CV
收藏

利用深度学习进行皮肤癌检测:使用 Vision Transformer 对皮肤病变图像进行分类

Skin Cancer Detection utilizing Deep Learning: Classification of Skin Lesion Images using a Vision Transformer

University of Cologne(科隆大学)

Carolin Flosdorf, Justin Engelker, Igor Keller, Nicolas Mohr

AI总结:

针对皮肤癌检测精度有限的问题,采用基于自注意力机制的预训练 Vision Transformer(ViT)模型对皮肤病变图像进行分类,相比 CNN 等基础模型取得更优准确率,为医疗诊断提供了更快更准的潜在工具。

摘要:

皮肤癌检测在医疗保健中仍是一项重大挑战。常见的检测方法耗时长且需要人工协助,这在许多国家存在不足。先前的研究表明,卷积神经网络(CNNs)可以通过自动化和与人类水平相当的准确性来有效提供帮助。然而,尽管在过去几十年中取得了进展,其精确度仍然有限,导致大量错误分类,对人们的健康产生严重影响。因此,我们采用近年来基于自注意力机制思想开发的 Vision Transformer(ViT),特别是预训练 ViT 的两种配置。在与决策树分类器和 k-nearest neighbor(KNN)分类器等基础模型,以及 CNN 和较不复杂的 ViT 进行比较后,我们通常发现其在皮肤病变分类方面具有更优的指标。我们特别重视黑色素瘤的性能,因为它是致死率最高的皮肤癌类型。ViT-L32 模型实现了 91.57% 的准确率和 58.54% 的黑色素瘤召回率,而 ViT-L16 实现了 92.79% 的准确率和 56.10% 的黑色素瘤召回率。这为更快、更准确的诊断提供了潜在工具,并促进了医疗保健领域的整体改善。

英文摘要:

Skin cancer detection still represents a major challenge in healthcare. Common detection methods can be lengthy and require human assistance which falls short in many countries. Previous research demonstrates how convolutional neural networks (CNNs) can help effectively through both automation and an accuracy that is comparable to the human level. However, despite the progress in previous decades, the precision is still limited, leading to substantial misclassifications that have a serious impact on people's health. Hence, we employ a Vision Transformer (ViT) that has been developed in recent years based on the idea of a self-attention mechanism, specifically two configurations of a pre-trained ViT. We generally find superior metrics for classifying skin lesions after comparing them to base models such as decision tree classifier and k-nearest neighbor (KNN) classifier, as well as to CNNs and less complex ViTs. In particular, we attach greater importance to the performance of melanoma, which is the most lethal type of skin cancer. The ViT-L32 model achieves an accuracy of 91.57% and a melanoma recall of 58.54%, while ViT-L16 achieves an accuracy of 92.79% and a melanoma recall of 56.10%. This offers a potential tool for faster and more accurate diagnoses and an overall improvement for the healthcare sector.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑