Deep Learning for Regularization Prediction in Diffeomorphic Image Registration

Jian Wang10000-0001-8478-0611, Miaomiao Zhang10000-0003-0457-3335
1: University of Virginia, USA
Publication date: 2022/02/13
PDF · arXiv


This paper presents a predictive model for estimating regularization parameters of diffeomorphic image registration. We introduce a novel framework that automatically determines the parameters controlling the smoothness of diffeomorphic transformations. Our method significantly reduces the effort of parameter tuning, which is time and labor-consuming. To achieve the goal, we develop a predictive model based on deep convolutional neural networks (CNN) that learns the mapping between pairwise images and the regularization parameter of image registration. In contrast to previous methods that estimate such parameters in a high-dimensional image space, our model is built in an efficient bandlimited space with much lower dimensions. We demonstrate the effectiveness of our model on both 2D synthetic data and 3D real brain images. Experimental results show that our model not only predicts appropriate regularization parameters for image registration, but also improving the network training in terms of time and memory efficiency.


diffeomorphic image registration · predictive registration regularization · deep learning

Bibtex @article{melba:2021:017:wang, title = "Deep Learning for Regularization Prediction in Diffeomorphic Image Registration", author = "Wang, Jian and Zhang, Miaomiao", journal = "Machine Learning for Biomedical Imaging", volume = "1", issue = "February 2022 issue", year = "2021", pages = "1--20", issn = "2766-905X", doi = "", url = "" }
RISTY - JOUR AU - Wang, Jian AU - Zhang, Miaomiao PY - 2021 TI - Deep Learning for Regularization Prediction in Diffeomorphic Image Registration T2 - Machine Learning for Biomedical Imaging VL - 1 IS - February 2022 issue SP - 1 EP - 20 SN - 2766-905X DO - UR - ER -

2021:017 cover