arXiv reaDer
UNETR++: 効率的で正確な 3D 医用画像セグメンテーションの詳細
UNETR++: Delving into Efficient and Accurate 3D Medical Image Segmentation
変圧器モデルの成功により、最近の研究では 3D 医療セグメンテーション タスクへの適用性が研究されています。トランスフォーマー モデル内で、セルフアテンション メカニズムは、長期的な依存関係を把握しようとする主要な構成要素の 1 つです。ただし、セルフアテンション操作には二次的な複雑さがあり、特に入力が多数のスライスを含む 3D であるボリューム メディカル イメージングでは、計算上のボトルネックであることが証明されています。このホワイト ペーパーでは、UNETR++ という名前の 3D 医用画像セグメンテーション アプローチを提案します。これは、高品質のセグメンテーション マスクと、パラメーター、計算コスト、および推論速度に関する効率の両方を提供します。私たちの設計の核心は、空間およびチャネルの注意に基づく相互依存ブランチのペアを使用して、空間およびチャネルごとの識別機能を効率的に学習する、新しい効率的な一対の注意 (EPA) ブロックの導入です。私たちの空間的注意の定式化は、入力シーケンスの長さに関して線形の複雑さを持つ効率的です。空間的およびチャネルに焦点を当てたブランチ間の通信を可能にするために、全体的なネットワーク パラメーターを削減しながら、補完的な利点 (ペア アテンション) を提供するクエリおよびキー マッピング関数の重みを共有します。 Synapse、BTCV、ACDC、BRaTs、および Decathlon-Lung の 5 つのベンチマークに関する広範な評価により、効率と精度の両面での貢献の有効性が明らかになりました。 Synapse では、UNETR++ は 87.2% の Dice Score で新しい最先端を設定し、文献の最良の方法と比較して、パラメーターと FLOP の両方で 71% 以上の削減で大幅に効率的です。 .コード: https://github.com/Amshaker/unetr_plus_plus。
Owing to the success of transformer models, recent works study their applicability in 3D medical segmentation tasks. Within the transformer models, the self-attention mechanism is one of the main building blocks that strives to capture long-range dependencies. However, the self-attention operation has quadratic complexity which proves to be a computational bottleneck, especially in volumetric medical imaging, where the inputs are 3D with numerous slices. In this paper, we propose a 3D medical image segmentation approach, named UNETR++, that offers both high-quality segmentation masks as well as efficiency in terms of parameters, compute cost, and inference speed. The core of our design is the introduction of a novel efficient paired attention (EPA) block that efficiently learns spatial and channel-wise discriminative features using a pair of inter-dependent branches based on spatial and channel attention. Our spatial attention formulation is efficient having linear complexity with respect to the input sequence length. To enable communication between spatial and channel-focused branches, we share the weights of query and key mapping functions that provide a complimentary benefit (paired attention), while also reducing the overall network parameters. Our extensive evaluations on five benchmarks, Synapse, BTCV, ACDC, BRaTs, and Decathlon-Lung, reveal the effectiveness of our contributions in terms of both efficiency and accuracy. On Synapse, our UNETR++ sets a new state-of-the-art with a Dice Score of 87.2%, while being significantly efficient with a reduction of over 71% in terms of both parameters and FLOPs, compared to the best method in the literature. Code: https://github.com/Amshaker/unetr_plus_plus.
updated: Wed Mar 22 2023 20:30:10 GMT+0000 (UTC)
published: Thu Dec 08 2022 18:59:57 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アソシエイト