arXiv reaDer
GraDIRN:変形可能な画像レジストレーションのための反復勾配降下ベースのエネルギー最小化の学習
GraDIRN: Learning Iterative Gradient Descent-based Energy Minimization for Deformable Image Registration
深層学習フレームワークに勾配ベースの反復エネルギー最小化を組み込むことにより、変形可能な画像レジストレーションを学習するための勾配降下ベースの画像レジストレーションネットワーク(GraDIRN)を紹介します。従来の画像レジストレーションアルゴリズムは、通常、反復エネルギー最小化最適化を使用して、画像のペア間の最適な変換を見つけます。これは、多くの反復が必要な場合に時間がかかります。対照的に、最近の学習ベースの方法は、深いニューラルネットワークをトレーニングすることによってこのコストのかかる反復最適化を償却し、トレーニング後の高速ネットワークフォワードパスによって1組の画像の登録を実現できるようにします。深層学習と反復変分エネルギー最適化の数学的構造を組み合わせた画像再構成手法の成功に動機付けられて、マルチ解像度勾配降下エネルギー最小化に基づく新しい登録ネットワークを構築します。ネットワークのフォワードパスは、明示的な画像非類似度勾配ステップと、畳み込みニューラルネットワーク(CNN)によってパラメーター化された一般化された正則化ステップを一定の反復回数で実行します。自動微分を使用して、変換による明示的な画像非類似度勾配の前方計算グラフを導出します。そのため、複雑でエラーが発生しやすい勾配導関数なしで、任意の画像非類似度メトリックと変換モデルを使用できます。このアプローチは、2D心臓MR画像と3D脳MR画像を使用した登録タスクの広範な評価を通じて、学習可能なパラメーターを少なくしながら、最先端の登録パフォーマンスを実現することを示しています。
We present a Gradient Descent-based Image Registration Network (GraDIRN) for learning deformable image registration by embedding gradient-based iterative energy minimization in a deep learning framework. Traditional image registration algorithms typically use iterative energy-minimization optimization to find the optimal transformation between a pair of images, which is time-consuming when many iterations are needed. In contrast, recent learning-based methods amortize this costly iterative optimization by training deep neural networks so that registration of one pair of images can be achieved by fast network forward pass after training. Motivated by successes in image reconstruction techniques that combine deep learning with the mathematical structure of iterative variational energy optimization, we formulate a novel registration network based on multi-resolution gradient descent energy minimization. The forward pass of the network takes explicit image dissimilarity gradient steps and generalized regularization steps parameterized by Convolutional Neural Networks (CNN) for a fixed number of iterations. We use auto-differentiation to derive the forward computational graph for the explicit image dissimilarity gradient w.r.t. the transformation, so arbitrary image dissimilarity metrics and transformation models can be used without complex and error-prone gradient derivations. We demonstrate that this approach achieves state-of-the-art registration performance while using fewer learnable parameters through extensive evaluations on registration tasks using 2D cardiac MR images and 3D brain MR images.
updated: Tue Dec 07 2021 14:48:31 GMT+0000 (UTC)
published: Tue Dec 07 2021 14:48: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アソシエイト