Abstract:The heart is a relatively complex non-rigid motion organ in the human body. Quantitative motion analysis of the heart takes on a critical significance to help doctors with accurate diagnosis and treatment. Moreover, cardiovascular magnetic resonance imaging (CMRI) can be used to perform a more detailed quantitative analysis evaluation for cardiac diagnosis. Deformable image registration (DIR) has become a vital task in biomedical image analysis since tissue structures have variability in medical images. Recently, the model based on masked autoencoder (MAE) has recently been shown to be effective in computer vision tasks. Vision Transformer has the context aggregation ability to restore the semantic information in the original image regions by using a low proportion of visible image patches to predict the masked image patches. A novel Transformer-ConvNet architecture is proposed in this study based on MAE for medical image registration. The core of the Transformer is designed as a masked autoencoder (MAE) and a lightweight decoder structure, and feature extraction before the downstream registration task is transformed into the self-supervised learning task. This study also rethinks the calculation method of the multi-head self-attention mechanism in the Transformer encoder. We improve the query-key-value-based dot product attention by introducing both depthwise separable convolution (DWSC) and squeeze and excitation (SE) modules into the self-attention module to reduce the amount of parameter computation to highlight image details and maintain high spatial resolution image features. In addition, concurrent spatial and channel squeeze and excitation (scSE) module is embedded into the CNN structure, which also proves to be effective for extracting robust feature representations. The proposed method, called MAE-TransRNet, has better generalization. The proposed model is evaluated on the cardiac short-axis public dataset (with images and labels) at the 2017 Automated Cardiac Diagnosis Challenge (ACDC). The relevant qualitative and quantitative results (e.g., dice performance and Hausdorff distance) suggest that the proposed model can achieve superior results over those achieved by the state-of-the-art methods, thus proving that MAE and improved self-attention are more effective and promising for medical image registration tasks. Codes and models are available at https://github.com/XinXiao101/MAE-TransRNet .

Multi-Resolution Diffeomorphic Image Registration with Convolutional Vision Transformer Network.

TransMorph: Transformer for unsupervised medical image registration

ViT-V-Net: Vision Transformer for Unsupervised Volumetric Medical Image Registration

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

Diffeomorphic Image Registration with Neural Velocity Field

MDH-Net: Advancing 3D Brain MRI Registration with Multi-Stage Transformer and Dual-Stream Feature Refinement Hybrid Network

Affine Medical Image Registration with Coarse-to-Fine Vision Transformer

HCS-Net: Multi-level deformation strategy combined with quadruple attention for image registration

Learning Dual Transformer Network for Diffeomorphic Registration

PIViT: Large Deformation Image Registration with Pyramid-Iterative Vision Transformer

3D Medical Image Registration Based on Simplified Transformer Block and Multi-Scale Iterative Structure

MFCTrans: Multi-scale Feature Connection Transformer for Deformable Medical Image Registration

TD-Net:unsupervised medical image registration network based on Transformer and CNN

Deformable Image Registration Using Vision Transformers for Cardiac Motion Estimation from Cine Cardiac MRI Images

Explicit–implicit symmetric diffeomorphic deformable image registration with convolutional neural network

[Cascaded Multi-Level Medical Image Registration Method Based on Transformer].

TransDIR: Deformable imaging registration network based on transformer to improve the feature extraction ability

MDReg-Net: Multi-resolution diffeomorphic image registration using fully convolutional networks with deep self-supervision

MAE-TransRNet: An improved transformer-ConvNet architecture with masked autoencoder for cardiac MRI registration

CvTMorph: Improving Local Feature Extraction in Medical Image Registration for Respiratory Motion Modeling with Convolutional Vision Transformer