Diffeomorphic Image Registration Using Lipschitz Continuous Residual Networks

Ankita Joshi, Yi Hong
Proceedings of The 5th International Conference on Medical Imaging with Deep Learning, PMLR 172:605-617, 2022.

Abstract

Image registration is an essential task in medical image analysis. We propose two novel unsupervised diffeomorphic image registration networks, which use deep Residual Networks (ResNets) as numerical approximations of the underlying continuous diffeomorphic setting governed by ordinary differential equations (ODEs), viewed as an Eulerian discretization scheme. While considering the ODE-based parameterizations of diffeomorphisms, we consider both stationary and non-stationary (time varying) velocity fields as the driving velocities to solve the ODEs, which gives rise to our two proposed architectures for diffeomorphic registration. We also employ Lipschitz-continuity on the Residual Networks in both architectures to define the admissible Hilbert space of velocity fields as a Reproducing Kernel Hilbert Spaces (RKHS) and regularize the smoothness of the velocity fields. We apply both registration networks to align and segment the OASIS brain MRI dataset. Experimental results demonstrate that our models are computation efficient and achieve comparable registration results with a smoother deformation field.

Cite this Paper


BibTeX
@InProceedings{pmlr-v172-joshi22a, title = {Diffeomorphic Image Registration Using Lipschitz Continuous Residual Networks}, author = {Joshi, Ankita and Hong, Yi}, booktitle = {Proceedings of The 5th International Conference on Medical Imaging with Deep Learning}, pages = {605--617}, year = {2022}, editor = {Konukoglu, Ender and Menze, Bjoern and Venkataraman, Archana and Baumgartner, Christian and Dou, Qi and Albarqouni, Shadi}, volume = {172}, series = {Proceedings of Machine Learning Research}, month = {06--08 Jul}, publisher = {PMLR}, pdf = {https://proceedings.mlr.press/v172/joshi22a/joshi22a.pdf}, url = {https://proceedings.mlr.press/v172/joshi22a.html}, abstract = {Image registration is an essential task in medical image analysis. We propose two novel unsupervised diffeomorphic image registration networks, which use deep Residual Networks (ResNets) as numerical approximations of the underlying continuous diffeomorphic setting governed by ordinary differential equations (ODEs), viewed as an Eulerian discretization scheme. While considering the ODE-based parameterizations of diffeomorphisms, we consider both stationary and non-stationary (time varying) velocity fields as the driving velocities to solve the ODEs, which gives rise to our two proposed architectures for diffeomorphic registration. We also employ Lipschitz-continuity on the Residual Networks in both architectures to define the admissible Hilbert space of velocity fields as a Reproducing Kernel Hilbert Spaces (RKHS) and regularize the smoothness of the velocity fields. We apply both registration networks to align and segment the OASIS brain MRI dataset. Experimental results demonstrate that our models are computation efficient and achieve comparable registration results with a smoother deformation field.} }
Endnote
%0 Conference Paper %T Diffeomorphic Image Registration Using Lipschitz Continuous Residual Networks %A Ankita Joshi %A Yi Hong %B Proceedings of The 5th International Conference on Medical Imaging with Deep Learning %C Proceedings of Machine Learning Research %D 2022 %E Ender Konukoglu %E Bjoern Menze %E Archana Venkataraman %E Christian Baumgartner %E Qi Dou %E Shadi Albarqouni %F pmlr-v172-joshi22a %I PMLR %P 605--617 %U https://proceedings.mlr.press/v172/joshi22a.html %V 172 %X Image registration is an essential task in medical image analysis. We propose two novel unsupervised diffeomorphic image registration networks, which use deep Residual Networks (ResNets) as numerical approximations of the underlying continuous diffeomorphic setting governed by ordinary differential equations (ODEs), viewed as an Eulerian discretization scheme. While considering the ODE-based parameterizations of diffeomorphisms, we consider both stationary and non-stationary (time varying) velocity fields as the driving velocities to solve the ODEs, which gives rise to our two proposed architectures for diffeomorphic registration. We also employ Lipschitz-continuity on the Residual Networks in both architectures to define the admissible Hilbert space of velocity fields as a Reproducing Kernel Hilbert Spaces (RKHS) and regularize the smoothness of the velocity fields. We apply both registration networks to align and segment the OASIS brain MRI dataset. Experimental results demonstrate that our models are computation efficient and achieve comparable registration results with a smoother deformation field.
APA
Joshi, A. & Hong, Y.. (2022). Diffeomorphic Image Registration Using Lipschitz Continuous Residual Networks. Proceedings of The 5th International Conference on Medical Imaging with Deep Learning, in Proceedings of Machine Learning Research 172:605-617 Available from https://proceedings.mlr.press/v172/joshi22a.html.

Related Material