PULPo: Probabilistic Unsupervised Laplacian Pyramid Registration
- URL: http://arxiv.org/abs/2407.10567v1
- Date: Mon, 15 Jul 2024 09:30:31 GMT
- Title: PULPo: Probabilistic Unsupervised Laplacian Pyramid Registration
- Authors: Leonard Siegert, Paul Fischer, Mattias P. Heinrich, Christian F. Baumgartner,
- Abstract summary: Deformable image registration is fundamental to many medical imaging applications.
We present PULPo, a method for probabilistic deformable registration capable of uncertainty quantification.
- Score: 3.2868275835047243
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Deformable image registration is fundamental to many medical imaging applications. Registration is an inherently ambiguous task often admitting many viable solutions. While neural network-based registration techniques enable fast and accurate registration, the majority of existing approaches are not able to estimate uncertainty. Here, we present PULPo, a method for probabilistic deformable registration capable of uncertainty quantification. PULPo probabilistically models the distribution of deformation fields on different hierarchical levels combining them using Laplacian pyramids. This allows our method to model global as well as local aspects of the deformation field. We evaluate our method on two widely used neuroimaging datasets and find that it achieves high registration performance as well as substantially better calibrated uncertainty quantification compared to the current state-of-the-art.
Related papers
- Hierarchical uncertainty estimation for learning-based registration in neuroimaging [10.964653898591413]
We propose a principled way to propagate uncertainties (epistemic or aleatoric) estimated at the level of spatial location.
Experiments show that uncertainty-aware fitting of transformations improves the registration accuracy of brain MRI scans.
arXiv Detail & Related papers (2024-10-11T23:12:16Z) - From Registration Uncertainty to Segmentation Uncertainty [11.294691606431526]
We propose a novel framework to concurrently estimate both the epistemic and aleatoric segmentation uncertainties for image registration.
By introducing segmentation uncertainty along with existing methods for estimating registration uncertainty, we offer vital insights into the potential uncertainties at different stages of image registration.
arXiv Detail & Related papers (2024-03-08T07:16:14Z) - GSMorph: Gradient Surgery for cine-MRI Cardiac Deformable Registration [62.41725951450803]
Learning-based deformable registration relies on weighted objective functions trading off registration accuracy and smoothness of the field.
We construct a registration model based on the gradient surgery mechanism, named GSMorph, to achieve a hyper parameter-free balance on multiple losses.
Our method is model-agnostic and can be merged into any deep registration network without introducing extra parameters or slowing down inference.
arXiv Detail & Related papers (2023-06-26T13:32:09Z) - DiffuseMorph: Unsupervised Deformable Image Registration Along
Continuous Trajectory Using Diffusion Models [31.826844124173984]
We present a novel approach of diffusion model-based probabilistic image registration, called DiffuseMorph.
Our model learns the score function of the deformation between moving and fixed images.
Our method can provide flexible and accurate deformation with a capability of topology preservation.
arXiv Detail & Related papers (2021-12-09T08:41:23Z) - Deblurring via Stochastic Refinement [85.42730934561101]
We present an alternative framework for blind deblurring based on conditional diffusion models.
Our method is competitive in terms of distortion metrics such as PSNR.
arXiv Detail & Related papers (2021-12-05T04:36:09Z) - A Deep Discontinuity-Preserving Image Registration Network [73.03885837923599]
Most deep learning-based registration methods assume that the desired deformation fields are globally smooth and continuous.
We propose a weakly-supervised Deep Discontinuity-preserving Image Registration network (DDIR) to obtain better registration performance and realistic deformation fields.
We demonstrate that our method achieves significant improvements in registration accuracy and predicts more realistic deformations, in registration experiments on cardiac magnetic resonance (MR) images.
arXiv Detail & Related papers (2021-07-09T13:35:59Z) - Scalable Marginal Likelihood Estimation for Model Selection in Deep
Learning [78.83598532168256]
Marginal-likelihood based model-selection is rarely used in deep learning due to estimation difficulties.
Our work shows that marginal likelihoods can improve generalization and be useful when validation data is unavailable.
arXiv Detail & Related papers (2021-04-11T09:50:24Z) - Improving Uncertainty Calibration via Prior Augmented Data [56.88185136509654]
Neural networks have proven successful at learning from complex data distributions by acting as universal function approximators.
They are often overconfident in their predictions, which leads to inaccurate and miscalibrated probabilistic predictions.
We propose a solution by seeking out regions of feature space where the model is unjustifiably overconfident, and conditionally raising the entropy of those predictions towards that of the prior distribution of the labels.
arXiv Detail & Related papers (2021-02-22T07:02:37Z) - Constraining Volume Change in Learned Image Registration for Lung CTs [4.37795447716986]
In this paper, we identify important strategies of conventional registration methods for lung registration and successfully developed the deep-learning counterpart.
We employ a Gaussian-pyramid-based multilevel framework that can solve the image registration optimization in a coarse-to-fine fashion.
We show that it archives state-of-the-art results on the COPDGene dataset compared to the challenge winning conventional registration method with much shorter execution time.
arXiv Detail & Related papers (2020-11-29T14:09:31Z) - Flexible Bayesian Modelling for Nonlinear Image Registration [0.3914676152740142]
We describe a diffeomorphic registration algorithm that allows groups of images to be accurately aligned to a common space.
The idea is to perform inference in a probabilistic graphical model that accounts for variability in both shape and appearance.
The model is evaluated at inter-subject registration of 3D human brain scans.
arXiv Detail & Related papers (2020-06-03T15:33:14Z) - Gaussian Process States: A data-driven representation of quantum
many-body physics [59.7232780552418]
We present a novel, non-parametric form for compactly representing entangled many-body quantum states.
The state is found to be highly compact, systematically improvable and efficient to sample.
It is also proven to be a universal approximator' for quantum states, able to capture any entangled many-body state with increasing data set size.
arXiv Detail & Related papers (2020-02-27T15:54:44Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.