Timezone: »
We introduce a region-specific diffeomorphic metric mapping (RDMM) registration approach. RDMM is non-parametric, estimating spatio-temporal velocity fields which parameterize the sought-for spatial transformation. Regularization of these velocity fields is necessary. In contrast to existing non-parametric registration approaches using a fixed spatially-invariant regularization, for example, the large displacement diffeomorphic metric mapping (LDDMM) model, our approach allows for spatially-varying regularization which is advected via the estimated spatio-temporal velocity field. Hence, not only can our model capture large displacements, it does so with a spatio-temporal regularizer that keeps track of how regions deform, which is a more natural mathematical formulation. We explore a family of RDMM registration approaches: 1) a registration model where regions with separate regularizations are pre-defined (e.g., in an atlas space or for distinct foreground and background regions), 2) a registration model where a general spatially-varying regularizer is estimated, and 3) a registration model where the spatially-varying regularizer is obtained via an end-to-end trained deep learning (DL) model. We provide a variational derivation of RDMM, showing that the model can assure diffeomorphic transformations in the continuum, and that LDDMM is a particular instance of RDMM. To evaluate RDMM performance we experiment 1) on synthetic 2D data and 2) on two 3D datasets: knee magnetic resonance images (MRIs) of the Osteoarthritis Initiative (OAI) and computed tomography images (CT) of the lung. Results show that our framework achieves comparable performance to state-of-the-art image registration approaches, while providing additional information via a learned spatio-temporal regularizer. Further, our deep learning approach allows for very fast RDMM and LDDMM estimations. Code is available at https://github.com/uncbiag/registration.
Author Information
Zhengyang Shen (University of North Carolina at Chapel Hill)
Francois-Xavier Vialard (University Paris-Est)
Marc Niethammer (UNC Chapel Hill)
More from the Same Authors
-
2021 : Faster Unbalanced Optimal Transport: Translation invariant Sinkhorn and 1-D Frank-Wolfe »
Thibault Sejourne · Francois-Xavier Vialard · Gabriel Peyré -
2021 : Faster Unbalanced Optimal Transport: Translation invariant Sinkhorn and 1-D Frank-Wolfe »
Thibault Sejourne · Francois-Xavier Vialard · Gabriel Peyré -
2022 Poster: Parameter tuning and model selection in Optimal Transport with semi-dual Brenier formulation »
Adrien Vacher · Francois-Xavier Vialard -
2022 Poster: Compositional Generalization in Unsupervised Compositional Representation Learning: A Study on Disentanglement and Emergent Language »
Zhenlin Xu · Marc Niethammer · Colin Raffel -
2022 Poster: On global convergence of ResNets: From finite to infinite width using linear parameterization »
Raphaël Barboni · Gabriel Peyré · Francois-Xavier Vialard -
2022 Poster: On Measuring Excess Capacity in Neural Networks »
Florian Graf · Sebastian Zeng · Bastian Rieck · Marc Niethammer · Roland Kwitt -
2021 Poster: Accurate Point Cloud Registration with Robust Optimal Transport »
Zhengyang Shen · Jean Feydy · Peirong Liu · Ariel H Curiale · Ruben San Jose Estepar · Raul San Jose Estepar · Marc Niethammer -
2021 Poster: The Unbalanced Gromov Wasserstein Distance: Conic Formulation and Relaxation »
Thibault Sejourne · Francois-Xavier Vialard · Gabriel Peyré -
2020 Poster: A shooting formulation of deep learning »
François-Xavier Vialard · Roland Kwitt · Susan Wei · Marc Niethammer -
2020 Oral: A shooting formulation of deep learning »
François-Xavier Vialard · Roland Kwitt · Susan Wei · Marc Niethammer -
2017 Poster: Deep Learning with Topological Signatures »
Christoph Hofer · Roland Kwitt · Marc Niethammer · Andreas Uhl -
2015 Poster: Statistical Topological Data Analysis - A Kernel Perspective »
Roland Kwitt · Stefan Huber · Marc Niethammer · Weili Lin · Ulrich Bauer