Title
A variational Bayesian method for similarity learning in non-rigid image registration
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.
Year
DOI
Venue
2022
10.1109/CVPR52688.2022.00022
2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)
Keywords
DocType
Volume
Machine learning, Self-& semi-& meta- & unsupervised learning
Conference
2022
Issue
ISSN
ISBN
1
1063-6919
978-1-6654-6947-0
Citations 
PageRank 
References 
0
0.34
17
Authors
7
Name
Order
Citations
PageRank
Daniel Grzech100.68
Mohammad Farid Azampour200.34
Ben Glocker32157119.81
Julia A Schnabel41978151.49
Nassir Navab56594578.60
Bernhard Kainz621117.54
Loïc Le Folgoc7516.48