arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2105.03358eess.IVcs.CVcs.LG
收藏

软注意力机制提升皮肤癌分类性能

Soft-Attention Improves Skin Cancer Classification Performance

State University of New York(纽约州立大学)

Soumyya Kanti Datta, Seyed Mohammad Abuzar Hashemi, Sargur N. Srihari, Mingchen Gao

AI总结:

该研究将软注意力机制与VGG、ResNet等主流深度神经网络结合,在HAM10000和ISIC-2017数据集上提升了皮肤癌分类的准确率与灵敏度,代码已开源。

摘要:

在临床应用中,神经网络必须聚焦并突出输入图像的最重要部分,软注意力(Soft-Attention)机制可使神经网络实现这一目标。本文研究了软注意力在深度神经网络架构中的有效性,其核心目标是提升重要特征的价值并抑制易产生噪声的特征。我们在对皮肤病变进行分类时,对比了VGG、ResNet、InceptionResNetv2和DenseNet架构在有无软注意力机制时的性能。结合软注意力的原始网络在HAM10000数据集[25]上的准确率达到93.7%,较基准[16]提升了4.7%;在ISIC-2017数据集[2]上,结合软注意力的模型灵敏度得分达到91.6%,较基准[31]提升了3.8%。代码已在github上公开。

英文摘要:

In clinical applications, neural networks must focus on and highlight the most important parts of an input image. Soft-Attention mechanism enables a neural network toachieve this goal. This paper investigates the effectiveness of Soft-Attention in deep neural architectures. The central aim of Soft-Attention is to boost the value of important features and suppress the noise-inducing features. We compare the performance of VGG, ResNet, InceptionResNetv2 and DenseNet architectures with and without the Soft-Attention mechanism, while classifying skin lesions. The original network when coupled with Soft-Attention outperforms the baseline[16] by 4.7% while achieving a precision of 93.7% on HAM10000 dataset [25]. Additionally, Soft-Attention coupling improves the sensitivity score by 3.8% compared to baseline[31] and achieves 91.6% on ISIC-2017 dataset [2]. The code is publicly available at github.

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

↑