DeepAI AI Chat
Log In Sign Up

Fast geodesic shooting for landmark matching using CUDA

by   Jiancong Wang, et al.

Landmark matching via geodesic shooting is a prerequisite task for numerous registration based applications in biomedicine. Geodesic shooting has been developed as one solution approach and formulates the diffeomorphic registration as an optimal control problem under the Hamiltonian framework. In this framework, with landmark positions q0 fixed, the problem solely depends on the initial momentum p0 and evolves through time steps according to a set of constraint equations. Given an initial p0, the algorithm flows q and p forward through time steps, calculates a loss based on point-set mismatch and kinetic energy, back-propagate through time to calculate gradient on p0 and update it. In the forward and backward pass, a pair-wise kernel on landmark points K and additional intermediate terms have to be calculated and marginalized, leading to O(N2) computational complexity, N being the number of points to be registered. For medical image applications, N maybe in the range of thousands, rendering this operation computationally expensive. In this work we ropose a CUDA implementation based on shared memory reduction. Our implementation achieves nearly 2 orders magnitude speed up compared to a naive CPU-based implementation, in addition to improved numerical accuracy as well as better registration results.


Deep Group-wise Variational Diffeomorphic Image Registration

Deep neural networks are increasingly used for pair-wise image registrat...

Self-supervised Landmark Learning with Deformation Reconstruction and Cross-subject Consistency Objectives

A Point Distribution Model (PDM) is the basis of a Statistical Shape Mod...

A matrix-free approach to parallel and memory-efficient deformable image registration

We present a novel computational approach to fast and memory-efficient d...

GraffMatch: Global Matching of 3D Lines and Planes for Wide Baseline LiDAR Registration

Using geometric landmarks like lines and planes can increase navigation ...

A Generalized Framework for Analytic Regularization of Uniform Cubic B-spline Displacement Fields

Image registration is an inherently ill-posed problem that lacks the con...

A landmark-based algorithm for automatic pattern recognition and abnormality detection

We study a class of mathematical and statistical algorithms with the aim...