Abstract:Conventional deformable registration methods aim at solving an optimization model carefully designed on image pairs and their computational costs are exceptionally high. In contrast, recent deep learning-based approaches can provide fast deformation estimation. These heuristic network architectures are fully data-driven and thus lack explicit geometric constraints which are indispensable to generate plausible deformations, e.g., topology-preserving. Moreover, these learning-based approaches typically pose hyper-parameter learning as a black-box problem and require considerable computational and human effort to perform many training runs. To tackle the aforementioned problems, we propose a new learning-based framework to optimize a diffeomorphic model via multi-scale propagation. Specifically, we introduce a generic optimization model to formulate diffeomorphic registration and develop a series of learnable architectures to obtain propagative updating in the coarse-to-fine feature space. Further, we propose a new bilevel self-tuned training strategy, allowing efficient search of task-specific hyper-parameters. This training strategy increases the flexibility to various types of data while reduces computational and human burdens. We conduct two groups of image registration experiments on 3D volume datasets including image-to-atlas registration on brain MRI data and image-to-image registration on liver CT data. Extensive results demonstrate the state-of-the-art performance of the proposed method with diffeomorphic guarantee and extreme efficiency. We also apply our framework to challenging multi-modal image registration, and investigate how our registration to support the down-streaming tasks for medical image analysis including multi-modal fusion and image segmentation.

Enhanced Deformation Vector Field Generation With Diffusion Models and Mamba‐Based Network for Registration Performance Enhancement

Diffeomorphic Image Registration with Neural Velocity Field

RegMamba: An Improved Mamba for Medical Image Registration

DiffuseReg: Denoising Diffusion Model for Obtaining Deformation Fields in Unsupervised Deformable Image Registration

Deformer: Towards Displacement Field Learning for Unsupervised Medical Image Registration

A Multi-scale Optimization Learning Framework for Diffeomorphic Deformable Registration.

Learning Deformable Image Registration from Optimization: Perspective, Modules, Bilevel Training and Beyond

VMambaMorph: a Multi-Modality Deformable Image Registration Framework based on Visual State Space Model with Cross-Scan Module

Self-Distilled Hierarchical Network for Unsupervised Deformable Image Registration

MambaMorph: a Mamba-based Framework for Medical MR-CT Deformable Registration

A Hybrid Scheme for Efficient Deformable Image Registration.

Deep Diffusion MRI Registration (DDMReg): A Deep Learning Method for Diffusion MRI Registration

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

Multiscale unsupervised network for deformable image registration

A diffeomorphic unsupervised method for deformable soft tissue image registration

Geometric Deep Learning for Unsupervised Registration of Diffusion Magnetic Resonance Images

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

Vector Field Attention for Deformable Image Registration

IDC-Net: Multi-stage Registration Network Using Intensity Adjustment, Dual-Stream and Cost Volume

Learning a Model-Driven Variational Network for Deformable Image Registration