スムーズな変換とトポロジ保存を提供する微分画像レジストレーションは、多くの医療画像解析タスクで必要とされます。従来の方法では、許容可能な変換のスペースに特定のモデリング制約を課し、最適化を使用して 2 つの画像間の最適な変換を見つけます。許容可能な変換の適切なスペースを指定することは困難です。スペースが狭すぎると登録品質が低下する可能性があり、スペースが一般的すぎると最適化を解決するのが難しくなる可能性があります。ディープ ニューラル ネットワークを利用して変換を直接学習する最近の学習ベースの方法は、高速な推論を実現しますが、小さな局所的な変形と一般化能力を捉えるのが難しいため、精度の課題に直面しています。ここでは、深層ニューラル ネットワークを利用して許容可能な変換の空間をモデル化する DNVF (ニューラル ベロシティ フィールドを使用した Diffeomorphic Image Registration) という名前の新しい最適化ベースの方法を提案します。正弦波活性化関数を備えた多層パーセプトロン (MLP) を使用して連続速度場を表し、空間内のすべての点に速度ベクトルを割り当て、複雑な変形をモデル化する柔軟性と最適化の利便性を提供します。さらに、最適化と学習ベースの方法の両方の利点を組み合わせることにより、カスケード画像登録フレームワーク (Cas-DNVF) を提案します。ここでは、完全な畳み込みニューラル ネットワーク (FCN) をトレーニングして初期変形を予測し、その後 DNVF を使用してさらに改良します。 2 つの大規模な 3D MR 脳スキャン データセットに関する実験は、提案された方法が最先端の登録方法よりも大幅に優れていることを示しています。
Diffeomorphic image registration, offering smooth transformation and topology preservation, is required in many medical image analysis tasks.Traditional methods impose certain modeling constraints on the space of admissible transformations and use optimization to find the optimal transformation between two images. Specifying the right space of admissible transformations is challenging: the registration quality can be poor if the space is too restrictive, while the optimization can be hard to solve if the space is too general. Recent learning-based methods, utilizing deep neural networks to learn the transformation directly, achieve fast inference, but face challenges in accuracy due to the difficulties in capturing the small local deformations and generalization ability. Here we propose a new optimization-based method named DNVF (Diffeomorphic Image Registration with Neural Velocity Field) which utilizes deep neural network to model the space of admissible transformations. A multilayer perceptron (MLP) with sinusoidal activation function is used to represent the continuous velocity field and assigns a velocity vector to every point in space, providing the flexibility of modeling complex deformations as well as the convenience of optimization. Moreover, we propose a cascaded image registration framework (Cas-DNVF) by combining the benefits of both optimization and learning based methods, where a fully convolutional neural network (FCN) is trained to predict the initial deformation, followed by DNVF for further refinement. Experiments on two large-scale 3D MR brain scan datasets demonstrate that our proposed methods significantly outperform the state-of-the-art registration methods.