Soft-Attention による皮膚がん分類性能の向上
Soft-Attention Improves Skin Cancer Classification Performance
臨床アプリケーションでは、ニューラル ネットワークは入力画像の最も重要な部分に焦点を合わせて強調する必要があります。 Soft-Attention メカニズムにより、ニューラル ネットワークはこの目標を達成できます。このペーパーでは、ディープ ニューラル アーキテクチャにおける Soft-Attention の有効性を調査します。 Soft-Attention の主な目的は、重要な機能の価値を高め、ノイズを誘発する機能を抑制することです。皮膚病変を分類しながら、VGG、ResNet、InceptionResNetv2、および DenseNet アーキテクチャの Soft-Attention メカニズムがある場合とない場合のパフォーマンスを比較します。 Soft-Attention と組み合わせた場合の元のネットワークは、HAM10000 データセット [25] で 93.7% の精度を達成しながら、ベースライン [16] を 4.7% 上回っています。さらに、Soft-Attention カップリングは、ベースラインと比較して感度スコアを 3.8% 向上させ [31]、ISIC-2017 データセット [2] で 91.6% を達成します。コードは 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.
updated: Fri Jun 04 2021 19:24:55 GMT+0000 (UTC)
published: Wed May 05 2021 00:13:23 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト