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.

An Integrated Method for Large Deformable Registration of Brain Images.

Registration of Cardiac Diffusion Tensor Images Using B-spline Free-Form Deformation

From Local to Global: An Anatomy-Aware Deep Learning Approach for Deformable Registration of Brain MR Images

Local Statistical Deformation Models for Deformable Image Registration.

An Unsupervised Registration Method for Brain Images Based on Contour Guidance

INR-LDDMM: Fluid-based Medical Image Registration Integrating Implicit Neural Representation and Large Deformation Diffeomorphic Metric Mapping

Sparse Deformation Prediction Using Markove Decision Processes (MDP) for Non-rigid Registration of MR Image

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

Deformable Registration For Geometric Distortion Correction Of Diffusion Tensor Imaging

Deformable Registration Of Dti And Spgr Images

Distributed-memory large deformation diffeomorphic 3D image registration

Non-rigid Registration Algorithm Based on Modally Controlled Free Form Deformation

Landmark and Intensity Based Registration with Large Deformations via Quasi-conformal Maps

A Nonrigid Registration Method for Correcting Brain Deformation Induced by Tumor Resection.

Large Deformation Diffeomorphic Image Registration with Laplacian Pyramid Networks

The same modality medical image registration with large deformation and clinical application based on adaptive diffeomorphic multi-resolution demons

A diffeomorphic unsupervised method for deformable soft tissue image registration

Longitudinal Deformable MRI Registration Via Dual-Feasible Deep Learning-Based Framework

A Fast Image Registration Algorithm for Diffeomorphic Image with Large Deformation

Multi-Task Learning for Registering Images with Large Deformation.

Unsupervised Deformable Medical Image Registration via Pyramidal Residual Deformation Fields Estimation