arXiv reaDer
焦点注意ネットワーク:生物医学画像セグメンテーションのための注意の最適化
Focal Attention Networks: optimising attention for biomedical image segmentation
近年、生物医学的画像セグメンテーションのための深層学習アーキテクチャに注意を組み込むことへの関心が高まっています。注意メカニズムのモジュラー設計により、U-Netなどの畳み込みニューラルネットワークアーキテクチャへの柔軟な統合が可能になります。注意を使用するのが適切かどうか、どのタイプの注意を使用するか、ネットワークのどこに注意モジュールを組み込むかは、現在見過ごされている重要な考慮事項です。この論文では、注意の変調における焦点パラメータの役割を調査し、損失関数とネットワークにおける注意の間のリンクを明らかにします。焦点距離ペナルティ項を組み込むことにより、境界ベースの損失を含むように統合焦点損失フレームワークを拡張します。さらに、FocalパラメーターをSqueeze-and-ExcitationブロックとAttention Gateに統合するための、シンプルで解釈可能なデータセットとモデル固有のヒューリスティックを開発し、3つの十分に検証された生物医学画像データセットでより少ない数の注意モジュールで最適なパフォーマンスを実現します。注意モジュールの賢明な使用を提案すると、パフォーマンスと効率が向上します。
In recent years, there has been increasing interest to incorporate attention into deep learning architectures for biomedical image segmentation. The modular design of attention mechanisms enables flexible integration into convolutional neural network architectures, such as the U-Net. Whether attention is appropriate to use, what type of attention to use, and where in the network to incorporate attention modules, are all important considerations that are currently overlooked. In this paper, we investigate the role of the Focal parameter in modulating attention, revealing a link between attention in loss functions and networks. By incorporating a Focal distance penalty term, we extend the Unified Focal loss framework to include boundary-based losses. Furthermore, we develop a simple and interpretable, dataset and model-specific heuristic to integrate the Focal parameter into the Squeeze-and-Excitation block and Attention Gate, achieving optimal performance with fewer number of attention modules on three well-validated biomedical imaging datasets, suggesting judicious use of attention modules results in better performance and efficiency.
updated: Sun Oct 31 2021 16:20:22 GMT+0000 (UTC)
published: Sun Oct 31 2021 16:20:22 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アソシエイト