本論文では、3D医療画像セグメンテーションのための効率的かつ高精度なアプローチであるUNETR++を提案している。UNETR++の中核となるのは、効率的な対ペア注意(EPA)ブロックの導入である。EPAブロックは、空間注意と特徴量注意の2つのモジュールから構成され、空間的特徴と特徴量間の相互依存性を効率的に学習する。
空間注意モジュールでは、キーとバリューを低次元空間にプロジェクトすることで、入力系列長に対して線形の計算量で自己注意を実現している。一方、特徴量注意モジュールは、チャンネル間の依存性を捉えるためにクエリーと
キーの内積演算を行う。
さらに、2つのモジュールでクエリーとキーの重みを共有することで、空間的特徴と特徴量間の強い相関を学習できるようにしている。一方、バリューの重みは独立に学習することで、両ブランチで相補的な特徴を抽出できるようにしている。
提案手法UNETR++は、5つのベンチマークデータセットで評価され、既存手法と比較して高精度かつ効率的な結果を示している。特に、Synapseデータセットでは、最新手法に対して8.9%の精度向上を達成しつつ、パラメータ数とFLOPsを71%以上削減している。
إلى لغة أخرى
من محتوى المصدر
arxiv.org
الرؤى الأساسية المستخلصة من
by Abdelrahman ... في arxiv.org 05-07-2024
https://arxiv.org/pdf/2212.04497.pdfاستفسارات أعمق