CLAIRE: A distributed-memory solver for constrained large deformation diffeomorphic image registration

Abstract

With this work, we release CLAIRE, a distributed-memory implementation of an effective solver for constrained large deformation diffeomorphic image registration problems in three dimensions. We consider an optimal control formulation. We invert for a stationary velocity field that parameterizes the deformation map. Our solver is based on a globalized, preconditioned, inexact reduced space Gauss--Newton--Krylov scheme. We exploit state-of-the-art techniques in scientific computing to develop an effective solver that scales to thousands of distributed memory nodes on high-end clusters. We present the formulation, discuss algorithmic features, describe the software package, and introduce an improved preconditioner for the reduced space Hessian to speed up the convergence of our solver. We test registration performance on synthetic and real data. We demonstrate registration accuracy on several neuroimaging datasets. We compare the performance of our scheme against different flavors of the Demons algorithm for diffeomorphic image registration. We study convergence of our preconditioner and our overall algorithm. We report scalability results on state-of-the-art supercomputing platforms. We demonstrate that we can solve registration problems for clinically relevant data sizes in two to four minutes on a standard compute node with 20 cores, attaining excellent data fidelity. With the present work we achieve a speedup of (on average) 5$\times$ with a peak performance of up to 17$\times$ compared to our former work.

arXiv comment37 pages;

References in corpus (7)

• An inverse problem formulation for parameter estimation of a reaction diffusion model of low grade gliomas
• Constrained $H^1$-regularization schemes for diffeomorphic image registration
• An inexact Newton-Krylov algorithm for constrained diffeomorphic image registration
• PDE-constrained optimization in medical image analysis
• A Lagrangian Gauss-Newton-Krylov Solver for Mass- and Intensity-Preserving Diffeomorphic Image Registration
• Coupling Brain-Tumor Biophysical Models and Diffeomorphic Image Registration
• A Semi-Lagrangian two-level preconditioned Newton-Krylov solver for constrained diffeomorphic image registration

Cited by in corpus (8)

• Integrated Biophysical Modeling and Image Analysis: Application to Neuro-Oncology
• Fast GPU 3D Diffeomorphic Image Registration
• Image-Driven Biophysical Tumor Growth Model Calibration
• Multi-Node Multi-GPU Diffeomorphic Image Registration for Large-Scale Imaging Problems
• Deep Implicit Optimization enables Robust Learnable Features for Deformable Image Registration
• Diffeomorphic Shape Matching by Operator Splitting in 3D Cardiology Imaging
• Stochastic Neural Networks for Automatic Cell Tracking in Microscopy Image Sequences of Bacterial Colonies
• Incompressible image registration using divergence-conforming B-splines