Alert button

Conditional Deformable Image Registration with Convolutional Neural Network

Jun 23, 2021
Tony C. W. Mok, Albert C. S. Chung

Figure 1 for Conditional Deformable Image Registration with Convolutional Neural Network
Figure 2 for Conditional Deformable Image Registration with Convolutional Neural Network
Figure 3 for Conditional Deformable Image Registration with Convolutional Neural Network
Figure 4 for Conditional Deformable Image Registration with Convolutional Neural Network

Share this with someone who'll enjoy it:

Recent deep learning-based methods have shown promising results and runtime advantages in deformable image registration. However, analyzing the effects of hyperparameters and searching for optimal regularization parameters prove to be too prohibitive in deep learning-based methods. This is because it involves training a substantial number of separate models with distinct hyperparameter values. In this paper, we propose a conditional image registration method and a new self-supervised learning paradigm for deep deformable image registration. By learning the conditional features that correlated with the regularization hyperparameter, we demonstrate that optimal solutions with arbitrary hyperparameters can be captured by a single deep convolutional neural network. In addition, the smoothness of the resulting deformation field can be manipulated with arbitrary strength of smoothness regularization during inference. Extensive experiments on a large-scale brain MRI dataset show that our proposed method enables the precise control of the smoothness of the deformation field without sacrificing the runtime advantage or registration accuracy.

* Early accepted by MICCAI2021  
View paper onarxiv icon

Share this with someone who'll enjoy it: