arXiv reaDer
ニューラルフィールドによる医用画像登録
Medical Image Registration via Neural Fields
画像レジストレーションは、多くの医療画像解析タスクで不可欠なステップです。画像レジストレーションの従来の方法は、主に最適化主導型であり、2 つの画像間の類似性を最大化する最適な変形を見つけます。 2 つの画像間の変換を直接予測するように訓練された最近の学習ベースの方法は、はるかに高速に実行されますが、モデルの一般化と個々の画像固有の変形の処理の非効率性によるパフォーマンスの低下に悩まされています。ここでは、NIR (Neural Image Registration) と呼ばれる新しいニューラル ネット ベースの画像登録フレームワークを紹介します。これは、最適化に基づいていますが、ディープ ニューラル ネットを利用して画像ペア間の変形をモデル化します。 NIR は、入力として 3D 座標を受け取り、対応する変形ベクトルを出力する、ニューラル フィールドを介して実装された連続関数を使用して 2 つの画像間の変換を表します。 NIR は、変形フィールドを生成する 2 つの方法を提供します。一般的な変形可能レジストレーション用の変位ベクトル フィールドを直接出力する方法と、速度ベクトル フィールドを出力して速度フィールドを統合し、微分画像レジストレーション用の変形フィールドを導出する方法です。最適な登録は、確率的勾配降下を介して神経場のパラメーターを更新することによって発見されます。座標エンコーディング、正弦波活性化、座標サンプリング、強度サンプリングなど、モデルの最適化を容易にするいくつかの設計上の選択肢について説明します。 2 つの 3D MR 脳スキャン データセットに関する実験では、NIR が従来の最適化ベースの方法よりも大幅に高速に実行されながら、登録精度と規則性の両方の点で最先端のパフォーマンスが得られることが実証されています。
Image registration is an essential step in many medical image analysis tasks. Traditional methods for image registration are primarily optimization-driven, finding the optimal deformations that maximize the similarity between two images. Recent learning-based methods, trained to directly predict transformations between two images, run much faster, but suffer from performance deficiencies due to model generalization and the inefficiency in handling individual image specific deformations. Here we present a new neural net based image registration framework, called NIR (Neural Image Registration), which is based on optimization but utilizes deep neural nets to model deformations between image pairs. NIR represents the transformation between two images with a continuous function implemented via neural fields, receiving a 3D coordinate as input and outputting the corresponding deformation vector. NIR provides two ways of generating deformation field: directly output a displacement vector field for general deformable registration, or output a velocity vector field and integrate the velocity field to derive the deformation field for diffeomorphic image registration. The optimal registration is discovered by updating the parameters of the neural field via stochastic gradient descent. We describe several design choices that facilitate model optimization, including coordinate encoding, sinusoidal activation, coordinate sampling, and intensity sampling. Experiments on two 3D MR brain scan datasets demonstrate that NIR yields state-of-the-art performance in terms of both registration accuracy and regularity, while running significantly faster than traditional optimization-based methods.
updated: Mon Aug 22 2022 20:31:41 GMT+0000 (UTC)
published: Tue Jun 07 2022 08:43:31 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アソシエイト