arXiv reaDer
高速微分同相画像レジストレーションのためのネステロフ加速ADMM
Nesterov Accelerated ADMM for Fast Diffeomorphic Image Registration
反復最適化を使用した決定論的アプローチは、微分同相画像レジストレーション(DiffIR)で歴史的に成功しています。これらのアプローチは非常に正確ですが、通常、かなりの計算負荷がかかります。ディープラーニングに基づく確率論的アプローチの最近の開発により、DiffIRのランタイムが競争力のある登録精度で1秒未満になり、従来の反復法に代わる高速な方法が提供されています。このホワイトペーパーでは、DiffIRでの反復アプローチのパフォーマンス上の利点を維持しながら、この速度の違いを減らすことを試みます。最初に、結果として生じる変形の微分同相写像を保証しながら、画像の大きな変形を処理するために中間の非定常速度場を機能的に構成する単純な反復スキームを提案します。次に、任意の次数の正則化項を使用してこれらの速度場に滑らかさを課す凸最適化モデルを提案し、ネステロフ勾配降下法と乗数の交互方向法(ADMM)を組み合わせた高速アルゴリズムでこのモデルを解きます。最後に、GPUの計算能力を活用して、この高速化されたADMMソルバーを3D心臓MRIデータセットに実装し、実行時間を2秒未満にさらに短縮します。厳密にdiffeomorphic変形を生成することに加えて、私たちの方法は、ディープラーニングベースの方法の推論時間に近づく速度で、ダイスとハウスドルフスコアの点で最先端のディープラーニングベースのアプローチと反復DiffIRアプローチの両方を上回ります。
Deterministic approaches using iterative optimisation have been historically successful in diffeomorphic image registration (DiffIR). Although these approaches are highly accurate, they typically carry a significant computational burden. Recent developments in stochastic approaches based on deep learning have achieved sub-second runtimes for DiffIR with competitive registration accuracy, offering a fast alternative to conventional iterative methods. In this paper, we attempt to reduce this difference in speed whilst retaining the performance advantage of iterative approaches in DiffIR. We first propose a simple iterative scheme that functionally composes intermediate non-stationary velocity fields to handle large deformations in images whilst guaranteeing diffeomorphisms in the resultant deformation. We then propose a convex optimisation model that uses a regularisation term of arbitrary order to impose smoothness on these velocity fields and solve this model with a fast algorithm that combines Nesterov gradient descent and the alternating direction method of multipliers (ADMM). Finally, we leverage the computational power of GPU to implement this accelerated ADMM solver on a 3D cardiac MRI dataset, further reducing runtime to less than 2 seconds. In addition to producing strictly diffeomorphic deformations, our methods outperform both state-of-the-art deep learning-based and iterative DiffIR approaches in terms of dice and Hausdorff scores, with speed approaching the inference time of deep learning-based methods.
updated: Sun Sep 26 2021 19:56:45 GMT+0000 (UTC)
published: Sun Sep 26 2021 19:56:45 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アソシエイト