arXiv reaDer
神経速度場による微分同相画像レジストレーション
Diffeomorphic Image Registration with Neural Velocity Field
微分同相画像レジストレーションは、医療画像分析における重要なタスクです。最近の学習ベースの画像レジストレーション方法は、畳み込みニューラルネットワーク(CNN)を利用して、画像ペア間の空間変換を学習し、高速な推論速度を実現します。ただし、これらの方法では、一般化能力を向上させるために、多くの場合、多数のトレーニングデータが必要になります。テスト時間中、学習ベースの方法では良好な登録結果が得られない可能性があります。これは、モデルがトレーニングデータセットに過剰適合していることが原因である可能性があります。この論文では、2つの画像にわたる変形を記述するために連続速度場(NeVF)の神経表現を提案します。具体的には、この神経速度フィールドは、空間内の各ポイントに速度ベクトルを割り当てます。これにより、複雑な変形フィールドのモデリングの柔軟性が高まります。さらに、微分同相写像のメモリ消費を削減するための単純なスパースサンプリング戦略を提案します。提案されたNeVFは、事前にトレーニングされた学習ベースのモデルを組み込むこともできます。このモデルの予測された変形は、最適化の初期状態として使用されます。 2つの大規模な3DMR脳スキャンデータセットで実施された広範な実験は、提案された方法が最先端の登録方法を大幅に上回っていることを示しています。
Diffeomorphic image registration is a crucial task in medical image analysis. Recent learning-based image registration methods utilize convolutional neural networks (CNN) to learn the spatial transformation between image pairs and achieve a fast inference speed. However, these methods often require a large number of training data to improve their generalization abilities. During the test time, learning-based methods might fail to provide a good registration result, which is likely because of the model overfitting on the training dataset. In this paper, we propose a neural representation of continuous velocity field (NeVF) to describe the deformations across two images. Specifically, this neural velocity field assigns a velocity vector to each point in the space, which has higher flexibility in modeling the complex deformation field. Furthermore, we propose a simple sparse-sampling strategy to reduce the memory consumption for the diffeomorphic registration. The proposed NeVF can also incorporate with a pre-trained learning-based model whose predicted deformation is taken as an initial state for optimization. Extensive experiments conducted on two large-scale 3D MR brain scan datasets demonstrate that our proposed method outperforms the state-of-the-art registration methods by a large margin.
updated: Fri Aug 19 2022 03:09:30 GMT+0000 (UTC)
published: Fri Feb 25 2022 05:04:29 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アソシエイト