A Variational Bayesian Method for Similarity Learning in Non-Rigid Image Registration

Daniel Grzech, Mohammad Farid Azampour, Ben Glocker, Julia Schnabel, Nassir Navab, Bernhard Kainz, Loïc Le Folgoc

CVPR 2022 pp. 119-128

doi:10.1109/CVPR52688.2022.00022 /cvpr/2022/grzech2022cvpr-variational/

Abstract

We propose a novel variational Bayesian formulation for diffeomorphic non-rigid registration of medical images, which learns in an unsupervised way a data-specific similarity metric. The proposed framework is general and may be used together with many existing image registration models. We evaluate it on brain MRI scans from the UK Biobank and show that use of the learnt similarity metric, which is parametrised as a neural network, leads to more accurate results than use of traditional functions, e.g. SSD and LCC, to which we initialise the model, without a negative impact on image registration speed or transformation smoothness. In addition, the method estimates the uncertainty associated with the transformation. The code and the trained models are available in a public repository: https://github.com/dgrzech/learnsim.

PDF CVPR Semantic Scholar

Cite

Text

Grzech et al. "A Variational Bayesian Method for Similarity Learning in Non-Rigid Image Registration." Conference on Computer Vision and Pattern Recognition, 2022. doi:10.1109/CVPR52688.2022.00022

Markdown

[Grzech et al. "A Variational Bayesian Method for Similarity Learning in Non-Rigid Image Registration." Conference on Computer Vision and Pattern Recognition, 2022.](https://mlanthology.org/cvpr/2022/grzech2022cvpr-variational/) doi:10.1109/CVPR52688.2022.00022

BibTeX

@inproceedings{grzech2022cvpr-variational,
  title     = {{A Variational Bayesian Method for Similarity Learning in Non-Rigid Image Registration}},
  author    = {Grzech, Daniel and Azampour, Mohammad Farid and Glocker, Ben and Schnabel, Julia and Navab, Nassir and Kainz, Bernhard and Le Folgoc, Loïc},
  booktitle = {Conference on Computer Vision and Pattern Recognition},
  year      = {2022},
  pages     = {119-128},
  doi       = {10.1109/CVPR52688.2022.00022},
  url       = {https://mlanthology.org/cvpr/2022/grzech2022cvpr-variational/}
}