Abstract:Objective: This study aims to enhance medical image registration by addressing the limitations of existing approaches that rely on spatial transformations through U-Net, ConvNets, or Transformers. The objective is to develop a novel architecture that combines ConvNets, Graph Neural Networks (GNNs), and Capsule Networks to improve the accuracy and efficiency of medical image registration, which can also deal with the problem of rotating registration. Approach: We propose an deep learning-based approach which can be utilized in both unsupervised and semi-supervised manners, named as HGCMorph. It leverages a hybrid framework that integrates ConvNets and GNNs to capture lower-level features, specifically short-range attention, while also utilizing Capsule Networks (CapsNets) to model abstract higher-level features, including entity properties such as position, size, orientation, deformation, and texture. This hybrid framework aims to provide a comprehensive representation of anatomical structures and their spatial relationships in medical images. Main results: The results demonstrate the superiority of HGCMorph over existing state-of-the-art deep learning-based methods in both qualitative and quantitative evaluations. In unsupervised training process, our model outperforms the recent SOTA method TransMorph by achieving 7 /38 increase on Dice Score Coefficient (DSC), and 2 /7 improvement on Negative Jacobian Determinant (NJD) for OASIS and LPBA40 datasets, respectively. Furthermore, HGCMorph achieves improved registration accuracy in semi-supervised training process. In addition, when dealing with complex 3D rotations and secondary randomly deformations, our method still achieves the best performance. We also tested our methods on lung datasets, such as JSRT, Montgoermy and Shenzhen. Significance: The significance lies in its innovative design to medical image registration. HGCMorph offers a novel framework that overcomes the limitations of existing methods by efficiently capturing both local and abstract features, leading to enhanced registration accuracy, discontinuity-preserving, and pose-learning abilities. The incorporation of Capsule Networks introduces valuable improvements, making the proposed method a valuable contribution to the field of medical image analysis. HGCMorph not only advances the SOTA methods but also has the potential to improve various medical applications that rely on accurate image registration.

GRAM: A framework for geodesic registration on anatomical manifolds

An Image Registration Approach Based on 3D Geometric Projection Similarity of the Human Head

A method for large diffeomorphic registration via broken geodesics

Multimodal Medical Image Registration Via Common Representations Learning and Differentiable Geometric Constraints

Anatomically Constrained and Attention-Guided Deep Feature Fusion for Joint Segmentation and Deformable Medical Image Registration.

Bayesian Unsupervised Disentanglement of Anatomy and Geometry for Deep Groupwise Image Registration

HGCMorph: joint discontinuity-preserving and pose-learning via GNN and capsule networks for deformable medical images registration

Bayesian Intrinsic Groupwise Image Registration: Unsupervised Disentanglement of Anatomy and Geometry.

Distributed-memory large deformation diffeomorphic 3D image registration

An Integrated Method for Large Deformable Registration of Brain Images.

Deformable templates using large deformation kinematics

Solving the problem and quantifying geometric variation in neuroanatomy using generative diffeomorphic mapping

Unsupervised Learning of Probabilistic Diffeomorphic Registration for Images and Surfaces

X-Metric: An N-Dimensional Information-Theoretic Framework for Groupwise Registration and Deep Combined Computing

Unsupervised Probabilistic Deformation Modeling for Robust Diffeomorphic Registration

GSMorph: Gradient Surgery for cine-MRI Cardiac Deformable Registration

DeepASDM: a Deep Learning Framework for Affine and Deformable Image Registration Incorporating a Statistical Deformation Model

Joint affine and deformable three‐dimensional networks for brain MRI registration

Anatomy-aware and acquisition-agnostic joint registration with SynthMorph

CMAN: Cascaded Multi-scale Spatial Channel Attention-guided Network for large 3D deformable registration of liver CT images

Large-scale Neuroanatomical Visualization Using a Manifold Embedding Approach