arXiv reaDer
高周波ガイダンスによる超解像度ベースのパッチフリー 3D 画像セグメンテーション
Super-Resolution Based Patch-Free 3D Image Segmentation with High-Frequency Guidance
高解像度 (HR) 3D 画像は、磁気共鳴画像法 (MRI) やコンピューター断層撮影法 (CT) などの医療画像など、現在広く使用されています。ただし、現在限られている GPU メモリとは対照的に、これらの 3D 画像の空間解像度と次元性が高いため、これらの 3D 画像のセグメント化は依然として課題です。したがって、既存の 3D 画像セグメンテーション手法のほとんどは、推論効率が低く、グローバルなコンテキスト情報を無視するパッチベースのモデルを使用しています。これらの問題に対処するために、グローバルな低解像度 (LR) 入力から HR セグメンテーションを実現できる、超解像度 (SR) ベースのパッチフリー 3D 画像セグメンテーション フレームワークを提案します。フレームワークには 2 つのサブタスクが含まれており、セマンティック セグメンテーションが主なタスクであり、超解像度は LR 入力からの高周波情報の再構築を支援する補助タスクです。 LR 入力による情報損失のバランスをさらに高めるために、高周波ガイダンス モジュール (HGM) を提案し、復元ガイダンスとして元の画像から HR パッチを切り出すための効率的な選択的クロッピング アルゴリズムを設計します。さらに、セグメンテーションと SR タスク間の相互接続を活用し、2 つのタスクの共同最適化を実現するタスク融合モジュール (TFM) も提案します。予測する場合、メインのセグメンテーション タスクのみが必要ですが、他のモジュールは高速化のために削除できます。 2 つの異なるデータセットでの実験結果は、私たちのフレームワークが従来のパッチベースの方法と比較して推論速度が 4 倍速く、そのパフォーマンスも他のパッチベースおよびパッチフリーのモデルを上回っていることを示しています。
High resolution (HR) 3D images are widely used nowadays, such as medical images like Magnetic Resonance Imaging (MRI) and Computed Tomography (CT). However, segmentation of these 3D images remains a challenge due to their high spatial resolution and dimensionality in contrast to currently limited GPU memory. Therefore, most existing 3D image segmentation methods use patch-based models, which have low inference efficiency and ignore global contextual information. To address these problems, we propose a super-resolution (SR) based patch-free 3D image segmentation framework that can realize HR segmentation from a global-wise low-resolution (LR) input. The framework contains two sub-tasks, of which semantic segmentation is the main task and super resolution is an auxiliary task aiding in rebuilding the high frequency information from the LR input. To furthermore balance the information loss with the LR input, we propose a High-Frequency Guidance Module (HGM), and design an efficient selective cropping algorithm to crop an HR patch from the original image as restoration guidance for it. In addition, we also propose a Task-Fusion Module (TFM) to exploit the inter connections between segmentation and SR task, realizing joint optimization of the two tasks. When predicting, only the main segmentation task is needed, while other modules can be removed for acceleration. The experimental results on two different datasets show that our framework has a four times higher inference speed compared to traditional patch-based methods, while its performance also surpasses other patch-based and patch-free models.
updated: Mon Jul 10 2023 07:53:28 GMT+0000 (UTC)
published: Wed Oct 26 2022 11:46:08 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アソシエイト