ディープラーニングは、さまざまな生物医学画像セグメンテーションタスクで大きな期待を示しています。通常、既存のモデルはU-Netに基づいており、ローカルオペレーターを積み重ねたエンコーダーデコーダーアーキテクチャに依存して、長期的な情報を徐々に集約します。ただし、ローカルオペレータのみを使用すると、効率と効果が制限されます。この作業では、生物医学画像のセグメンテーションのために、柔軟なグローバル集約ブロックを備えた非ローカルU-Netを提案します。これらのブロックは、サイズを維持するプロセスとしてU-Netに挿入できるだけでなく、ダウンサンプリングおよびアップサンプリングレイヤーとしても挿入できます。非局所U-Netを評価するために、3Dマルチモダリティ等強度乳児脳MR画像セグメンテーションタスクで徹底的な実験を行います。結果は、提案されたモデルがより少ないパラメーターとより高速な計算で最高のパフォーマンスを達成することを示しています。
Deep learning has shown its great promise in various biomedical image segmentation tasks. Existing models are typically based on U-Net and rely on an encoder-decoder architecture with stacked local operators to aggregate long-range information gradually. However, only using the local operators limits the efficiency and effectiveness. In this work, we propose the non-local U-Nets, which are equipped with flexible global aggregation blocks, for biomedical image segmentation. These blocks can be inserted into U-Net as size-preserving processes, as well as down-sampling and up-sampling layers. We perform thorough experiments on the 3D multimodality isointense infant brain MR image segmentation task to evaluate the non-local U-Nets. Results show that our proposed models achieve top performances with fewer parameters and faster computation.