Current browse context:
cs.CV
Change to browse by:
References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Conditional Deformable Image Registration with Convolutional Neural Network
(Submitted on 23 Jun 2021 (v1), last revised 5 Jul 2021 (this version, v2))
Abstract: 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 are 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.
Submission history
From: Tony C. W. Mok [view email][v1] Wed, 23 Jun 2021 22:25:28 GMT (7417kb,D)
[v2] Mon, 5 Jul 2021 12:21:32 GMT (7100kb,D)
Link back to: arXiv, form interface, contact.