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 .

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

TF-Unet:An Automatic Cardiac MRI Image Segmentation Method

Mixed Transformer U-Net for Medical Image Segmentation

Symmetric Transformer-based Network for Unsupervised Image Registration

UTSRMorph: A Unified Transformer and Superresolution Network for Unsupervised Medical Image Registration

A Transformer-based Network for Deformable Medical Image Registration

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

TransMorph: Transformer for unsupervised medical image registration

A lightweight residual network for unsupervised deformable image registration

TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation

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

An Adaptive Region-Based Transformer for Nonrigid Medical Image Registration with a Self-Constructing Latent Graph

UCTNet: Uncertainty-guided CNN-Transformer hybrid networks for medical image segmentation

UDRSNet: An unsupervised deformable registration module based on image structure similarity

MSCT-UNET: multi-scale contrastive transformer within U-shaped network for medical image segmentation

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

TEC-Net: Vision Transformer Embrace Convolutional Neural Networks for Medical Image Segmentation

3D TransUNet: Advancing Medical Image Segmentation through Vision Transformers

HCT-Unet: multi-target medical image segmentation via a hybrid CNN-transformer Unet incorporating multi-axis gated multi-layer perceptron

A novel full-convolution UNet-transformer for medical image segmentation

TSCA-Net: Transformer based spatial-channel attention segmentation network for medical images