Embedding Gradient-Based Optimization in Image Registration Networks
Qiu H, Hammernik K, Qin C, Chen C, Rueckert D (2022)
Publication Type: Conference contribution
Publication year: 2022
Journal
Publisher: Springer Science and Business Media Deutschland GmbH
Book Volume: 13436 LNCS
Pages Range: 56-65
Conference Proceedings Title: Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Event location: Singapore, SGP
ISBN: 9783031164453
DOI: 10.1007/978-3-031-16446-0_6
Abstract
Deep learning (DL) image registration methods amortize the costly pair-wise iterative optimization by training deep neural networks to predict the optimal transformation in one fast forward-pass. In this work, we bridge the gap between traditional iterative energy optimization-based registration and network-based registration, and propose Gradient Descent Network for Image Registration (GraDIRN). Our proposed approach trains a DL network that embeds unrolled multi-resolution gradient-based energy optimization in its forward pass, which explicitly enforces image dissimilarity minimization in its update steps. Extensive evaluations were performed on registration tasks using 2D cardiac MR and 3D brain MR images. We demonstrate that our approach achieved state-of-the-art registration performance while using fewer learned parameters, with good data efficiency and domain robustness.
Involved external institutions
United Kingdom (GB)How to cite
APA:
Qiu, H., Hammernik, K., Qin, C., Chen, C., & Rueckert, D. (2022). Embedding Gradient-Based Optimization in Image Registration Networks. In Linwei Wang, Qi Dou, P. Thomas Fletcher, Stefanie Speidel, Shuo Li (Eds.), Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics) (pp. 56-65). Singapore, SGP: Springer Science and Business Media Deutschland GmbH.
MLA:
Qiu, Huaqi, et al. "Embedding Gradient-Based Optimization in Image Registration Networks." Proceedings of the 25th International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2022, Singapore, SGP Ed. Linwei Wang, Qi Dou, P. Thomas Fletcher, Stefanie Speidel, Shuo Li, Springer Science and Business Media Deutschland GmbH, 2022. 56-65.
BibTeX: Download