Search papers, labs, and topics across Lattice.
The paper introduces Polaffini, a feature-based image registration framework leveraging deep learning-based segmentation to extract anatomical centroids as corresponding feature points. Polaffini uses these centroids to compute global and local affine transformations, resulting in a tunable polyaffine transformation within a log-Euclidean framework to ensure diffeomorphic properties. Experiments demonstrate that Polaffini outperforms intensity-based methods in structural alignment and improves initialization for subsequent non-linear registration.
Anatomically-grounded medical image registration gets a speed and accuracy boost by simply using deep learning segmentations to extract anatomical centroids for affine and polyaffine matching.
In this work we present Polaffini, a robust and versatile framework for anatomically grounded registration. Medical image registration is dominated by intensity-based registration methods that rely on surrogate measures of alignment quality. In contrast, feature-based approaches that operate by identifying explicit anatomical correspondences, while more desirable in theory, have largely fallen out of favor due to the challenges of reliably extracting features. However, such challenges are now significantly overcome thanks to recent advances in deep learning, which provide pre-trained segmentation models capable of instantly delivering reliable, fine-grained anatomical delineations. We aim to demonstrate that these advances can be leveraged to create new anatomically-grounded image registration algorithms. To this end, we propose Polaffini, which obtains, from these segmented regions, anatomically grounded feature points with 1-to-1 correspondence in a particularly simple way: extracting their centroids. These enable efficient global and local affine matching via closed-form solutions. Those are used to produce an overall transformation ranging from affine to polyaffine with tunable smoothness. Polyaffine transformations can have many more degrees of freedom than affine ones allowing for finer alignment, and their embedding in the log-Euclidean framework ensures diffeomorphic properties. Polaffini has applications both for standalone registration and as pre-alignment for subsequent non-linear registration, and we evaluate it against popular intensity-based registration techniques. Results demonstrate that Polaffini outperforms competing methods in terms of structural alignment and provides improved initialisation for downstream non-linear registration. Polaffini is fast, robust, and accurate, making it particularly well-suited for integration into medical image processing pipelines.
