Abstract:We propose numerical algorithms for solving large deformation diffeomorphic image registration problems. We formulate the nonrigid image registration problem as a problem of optimal control. This leads to an infinite-dimensional partial differential equation (PDE) constrained optimization problem. The PDE constraint consists, in its simplest form, of a hyperbolic transport equation for the evolution of the image intensity. The control variable is the velocity field. Tikhonov regularization on the control ensures well-posedness. We consider standard smoothness regularization based on H1- or H2-seminorms. We augment this regularization scheme with a constraint on the divergence of the velocity field (control variable) rendering the deformation incompressible (Stokes regularization scheme) and thus ensuring that the determinant of the deformation gradient is equal to one, up to the numerical error. We use a Fourier pseudospectral discretization in space and a Chebyshev pseudospectral discretization in time. The latter allows us to reduce the number of unknowns and enables the time-adaptive inversion for nonstationary velocity fields. We use a preconditioned, globalized, matrix-free, inexact Newton-Krylov method for numerical optimization. A parameter continuation is designed to estimate an optimal regularization parameter. Regularity is ensured by controlling the geometric properties of the deformation field. Overall, we arrive at a black-box solver that exploits computational tools that are precisely tailored for solving the optimality system. We study spectral properties of the Hessian, grid convergence, numerical accuracy, computational efficiency, and deformation regularity of our scheme. We compare the designed Newton-Krylov methods with a globalized Picard method (preconditioned gradient descent). We study the influence of a varying number of unknowns in time. The reported results demonstrate excellent numerical accuracy, guaranteed local deformation regularity, and computational efficiency with an optional control on local mass conservation. The Newton-Krylov methods clearly outperform the Picard method if high accuracy of the inversion is required. Our method provides equally good results for stationary and nonstationary velocity fields for two-image registration problems.

ADAPTIVE GRADIENT DESCENT OPTIMIZATION OF INITIAL MOMENTA FOR GEODESIC SHOOTING IN DIFFEOMORPHISMS

FireANTs: Adaptive Riemannian Optimization for Multi-Scale Diffeomorphic Registration

Improving Instance Optimization in Deformable Image Registration with Gradient Projection

Adaptive Diffeomorphic Multiresolution Demons and Their Application to Same Modality Medical Image Registration with Large Deformation

Influence of Implementation Parameters on Registration of Mr and Spect Brain Images by Maximization of Mutual Information

Frequency Diffeomorphisms for Efficient Image Registration

Primitive Simultaneous Optimization of Similarity Metrics for Image Registration

Optimizing the Dice Score and Jaccard Index for Medical Image Segmentation: Theory & Practice

An approach for medical image registration using a Double Exponential-Dynamic Group Based Cooperative Optimization algorithm

An Inexact Newton-Krylov Algorithm for Constrained Diffeomorphic Image Registration

Multi-Modality Image Registration with Gradient Orientation Information Based on Entropic Spanning Graph

The same modality medical image registration with large deformation and clinical application based on adaptive diffeomorphic multi-resolution demons

Diffeomorphic Image Registration with an Optimal Control Relaxation and Its Implementation

GSMorph: Gradient Surgery for cine-MRI Cardiac Deformable Registration

Fast and Robust Multimodal Image Registration Using A Local Derivative Pattern

ConvexAdam: Self-Configuring Dual-Optimisation-Based 3D Multitask Medical Image Registration

A Majorization-Minimization Algorithm for Neuroimage Registration

Edge Preserving Multi-Modal Registration Based On Gradient Intensity Self-Similarity

Applications of Automatic Differentiation in Image Registration

A Novel Unsupervised Learning Model for Diffeomorphic Image Registration