arXiv reaDer
Fourier-Net+: 帯域制限された表現を活用して効率的な 3D 医用画像レジストレーションを実現
Fourier-Net+: Leveraging Band-Limited Representation for Efficient 3D Medical Image Registration
U-Net スタイルのネットワークは、密な変位場を予測するための教師なし画像レジストレーションでよく利用されますが、高解像度の体積画像データの場合、これはリソースを大量に消費し、時間のかかる作業となります。この課題に取り組むために、私たちはまずフーリエネットを提案します。これは、高価な U-Net スタイルの拡張パスをパラメータフリーのモデル駆動型デコーダに置き換えます。フル解像度の変位場を直接予測する代わりに、フーリエネットは帯域制限されたフーリエ領域で変位場の低次元表現を学習し、モデル駆動型デコーダーが空間領域のフル解像度の変位場に変換します。 。 Fourier-Net を拡張して、Fourier-Net+ を導入します。これはさらに、画像の帯域制限された空間表現を入力として受け取り、U-Net スタイルのネットワークの縮小パスの畳み込み層の数をさらに削減します。最後に、登録パフォーマンスを向上させるために、Fourier-Net+ のカスケード バージョンを提案します。我々は、3 つのデータセットで提案手法を評価しました。その結果、我々が提案したフーリエネットとその変形は、現在の最先端の手法と同等の結果を達成しながら、より高速な推論速度、より低いメモリフットプリント、より少ない乗加算演算を示しました。このような小さな計算コストにより、当社の Fourier-Net+ は、低 VRAM GPU 上で大規模な 3D レジストレーションの効率的なトレーニングを可能にします。私たちのコードは https://github.com/xi-jia/Fourier-Net で公開されています。
U-Net style networks are commonly utilized in unsupervised image registration to predict dense displacement fields, which for high-resolution volumetric image data is a resource-intensive and time-consuming task. To tackle this challenge, we first propose Fourier-Net, which replaces the costly U-Net style expansive path with a parameter-free model-driven decoder. Instead of directly predicting a full-resolution displacement field, our Fourier-Net learns a low-dimensional representation of the displacement field in the band-limited Fourier domain which our model-driven decoder converts to a full-resolution displacement field in the spatial domain. Expanding upon Fourier-Net, we then introduce Fourier-Net+, which additionally takes the band-limited spatial representation of the images as input and further reduces the number of convolutional layers in the U-Net style network's contracting path. Finally, to enhance the registration performance, we propose a cascaded version of Fourier-Net+. We evaluate our proposed methods on three datasets, on which our proposed Fourier-Net and its variants achieve comparable results with current state-of-the art methods, while exhibiting faster inference speeds, lower memory footprint, and fewer multiply-add operations. With such small computational cost, our Fourier-Net+ enables the efficient training of large-scale 3D registration on low-VRAM GPUs. Our code is publicly available at https://github.com/xi-jia/Fourier-Net.
updated: Thu Jul 06 2023 13:57:12 GMT+0000 (UTC)
published: Thu Jul 06 2023 13:57:12 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アソシエイト