Image registration is a geometric deep learning task

Vasiliki Sideri-Lampretsa, Nil Stolt-Ansó, Julian McGinnis, Huaqi Qiu, Martin J. Menten, Daniel Rueckert

Keywords: Image registration, Geometric deep learning

TL;DR: Image registration is a geometric deep learning task

Abstract:

Data-driven deformable image registration methods predominantly rely on operations that process grid-like inputs.
However, applying deformable transformations to an image results in a warped space that deviates from a rigid grid structure.
Consequently, data-driven approaches with sequential deformations have to apply grid resampling operations between each deformation step.
While artifacts caused by resampling are negligible in high-resolution images, the resampling of sparse, high-dimensional feature grids introduces errors that affect the deformation modeling process.
Taking inspiration from Lagrangian reference frames of deformation fields, our work introduces a novel paradigm for data-driven deformable image registration that utilizes geometric deep-learning principles to model deformations without grid requirements.
Specifically, we model image features as a set of nodes that freely move in Euclidean space, update their coordinates under graph operations, and dynamically readjust their local neighborhoods.
We employ this formulation to construct a multi-resolution deformable registration model, where deformation layers iteratively refine the overall transformation at each resolution without intermediate resampling operations on the feature grids.
We investigate our method's ability to fully deformably capture large deformations across a number of medical imaging registration tasks.
In particular, we apply our approach (GeoReg) to the registration of inter-subject brain MR images and inhale-exhale lung CT images, showing on par performance with the current state-of-the-art methods.
We believe our contribution open up avenues of research to reduce the black-box nature of current learned registration paradigms by explicitly modeling the transformation within the architecture.

Primary Area: learning on graphs and other geometries & topologies

Code Of Ethics: I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics.

Submission Guidelines: I certify that this submission complies with the submission instructions as described on https://iclr.cc/Conferences/2025/AuthorGuide.

Anonymous Url: I certify that there is no URL (e.g., github page) that could be used to find authors’ identity.

No Acknowledgement Section: I certify that there is no acknowledgement section in this submission for double blind review.

Submission Number: 3542