arXiv reaDer
3D軸方向-肺結節分類への注意
3D Axial-Attention for Lung Nodule Classification
目的:近年、非局所ベースの方法が肺結節の分類にうまく適用されています。ただし、これらの方法では、低解像度の特徴マップに対して2Dアテンションまたは限定的な3Dアテンションが提供されます。さらに、完全な3Dアテンションは計算にコストがかかり、利用できない可能性がある大きなデータセットを必要とするため、畳み込みなどの便利なローカルフィルターに依然依存しています。方法:通常の非ローカルネットワークの計算能力の一部を必要とする3DAxial-Attentionを使用することを提案します。さらに、共有埋め込みに3D位置エンコーディングを追加することを提案することにより、非ローカルネットワークの位置不変問題を解決します。結果:少なくとも3人の放射線科医によって注釈が付けられた結節のみを使用して、厳密な実験設定に従うことにより、LIDC-IDRIデータセットで提案された方法を検証しました。私たちの結果は、3D Axial-Attentionモデルが、AUCやAccuracyを含むすべての評価指標で最先端のパフォーマンスを達成していることを示しています。結論:提案されたモデルは、完全な3Dアテンションを効果的に提供し、ローカルフィルターを必要とせずにすべてのレイヤーで使用できます。実験結果は、肺結節を分類するための完全な3D注意の重要性を示しています。
Purpose: In recent years, Non-Local based methods have been successfully applied to lung nodule classification. However, these methods offer 2D attention or a limited 3D attention to low-resolution feature maps. Moreover, they still depend on a convenient local filter such as convolution as full 3D attention is expensive to compute and requires a big dataset, which might not be available. Methods: We propose to use 3D Axial-Attention, which requires a fraction of the computing power of a regular Non-Local network. Additionally, we solve the position invariant problem of the Non-Local network by proposing adding 3D positional encoding to shared embeddings. Results: We validated the proposed method on the LIDC-IDRI dataset by following a rigorous experimental setup using only nodules annotated by at least three radiologists. Our results show that the 3D Axial-Attention model achieves state-of-the-art performance on all evaluation metrics including AUC and Accuracy. Conclusions: The proposed model provides full 3D attention effectively, which can be used in all layers without the need for local filters. The experimental results show the importance of full 3D attention for classifying lung nodules.
updated: Mon Dec 28 2020 06:49:09 GMT+0000 (UTC)
published: Mon Dec 28 2020 06:49:09 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アソシエイト