SITReg: Multi-resolution architecture for symmetric, inverse consistent, and topology preserving image registration

Abstract

Deep learning has emerged as a strong alternative for classical iterative methods for deformable medical image registration, where the goal is to find a mapping between the coordinate systems of two images. Popular classical image registration methods enforce the useful inductive biases of symmetricity, inverse consistency, and topology preservation by construction. However, while many deep learning registration methods encourage these properties via loss functions, no earlier methods enforce all of them by construction. Here, we propose a novel registration architecture based on extracting multi-resolution feature representations which is by construction symmetric, inverse consistent, and topology preserving. We also develop an implicit layer for memory efficient inversion of the deformation fields. Our method achieves state-of-the-art registration accuracy on three datasets. The code is available at https://github.com/honkamj/SITReg.

Accepted for publication at the Journal of Machine Learning for Biomedical Imaging (MELBA) https://melba-journal.org/2024:026

References in corpus (9)

• Geometric deep learning: going beyond Euclidean data
• A Deep Learning Framework for Unsupervised Affine and Deformable Image Registration
• Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gauges
• Unsupervised Learning for Fast Probabilistic Diffeomorphic Registration
• Learning a Probabilistic Model for Diffeomorphic Registration
• Deep Equilibrium Models
• Inverse-Consistent Deep Networks for Unsupervised Deformable Image Registration
• Multiscale Deep Equilibrium Models
• A survey on deep learning in medical image registration: new technologies, uncertainty, evaluation metrics, and beyond