Abstract:The main objective of anatomically plausible results for deformable image registration is to improve model's registration accuracy by minimizing the difference between a pair of fixed and moving images. Since many anatomical features are closely related to each other, leveraging supervision from auxiliary tasks (such as supervised anatomical segmentation) has the potential to enhance the realism of the warped images after registration. In this work, we employ a Multi-Task Learning framework to formulate registration and segmentation as a joint issue, in which we utilize anatomical constraint from auxiliary supervised segmentation to enhance the realism of the predicted images. First, we propose a Cross-Task Attention Block to fuse the high-level feature from both the registration and segmentation network. With the help of initial anatomical segmentation, the registration network can benefit from learning the task-shared feature correlation and rapidly focusing on the parts that need deformation. On the other hand, the anatomical segmentation discrepancy from ground-truth fixed annotations and predicted segmentation maps of initial warped images are integrated into the loss function to guide the convergence of the registration network. Ideally, a good deformation field should be able to minimize the loss function of registration and segmentation. The voxel-wise anatomical constraint inferred from segmentation helps the registration network to reach a global optimum for both deformable and segmentation learning. Both networks can be employed independently during the testing phase, enabling only the registration output to be predicted when the segmentation labels are unavailable. Qualitative and quantitative results indicate that our proposed methodology significantly outperforms the previous state-of-the-art approaches on inter-patient brain MRI registration and pre- and intra-operative uterus MRI registration tasks within our specific experimental setup, which leads to state-of-the-art registration quality scores of 0.755 and 0.731 (i.e., by 0.8% and 0.5% increases) DSC for both tasks, respectively.

Unsupervised Medical Image Registration Via Dynamic Adaptive Total P-Variation Regularization.

Locally Adaptive Total P-Variation Regularization for Non-Rigid Image Registration with Sliding Motion.

Unsupervised Imbalanced Registration for Enhancing Accuracy and Stability in Medical Image Registration

A New Unsupervised Learning Method for 3D Deformable Medical Image Registration

Multi-phase Deformable Registration for Time-dependent Abdominal Organ Variations

Weakly supervised medical image registration with multi-information guidance

Residual Aligner Network

Residual Aligner-based Network (RAN): Motion-separable structure for coarse-to-fine discontinuous deformable registration

Unsupervised 3D End-to-End Medical Image Registration with Volume Tweening Network

A Novel Unsupervised Learning Model for Diffeomorphic Image Registration

Non-rigid MRI-CT Image Registration with Unsupervised Deep Learning-Based Deformation Prediction

An Unsupervised Learning Model for Deformable Medical Image Registration

Medical Image Registration Algorithm in Image-Guided Noninvasive Surgery

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

Deformable registration of chest CT images using a 3D convolutional neural network based on unsupervised learning

Boundary-aware registration network for 4D-CT lung image with sliding motion

Unsupervised Registration for Liver CT-MR Images Based on the Multiscale Integrated Spatial-Weight Module and Dual Similarity Guidance.

DiffuseMorph: Unsupervised Deformable Image Registration Using Diffusion Model

Unsupervised Probabilistic Deformation Modeling for Robust Diffeomorphic Registration

Unsupervised End-to-end Learning for Deformable Medical Image Registration

LDVoxelMorph: A Precise Loss Function and Cascaded Architecture for Unsupervised Diffeomorphic Large Displacement Registration.