Stable Deep MRI Reconstruction using Generative Priors
- URL: http://arxiv.org/abs/2210.13834v3
- Date: Thu, 15 Jun 2023 17:10:10 GMT
- Title: Stable Deep MRI Reconstruction using Generative Priors
- Authors: Martin Zach and Florian Knoll and Thomas Pock
- Abstract summary: We propose a novel deep neural network based regularizer which is trained in a generative setting on reference magnitude images only.
The results demonstrate competitive performance, on par with state-of-the-art end-to-end deep learning methods.
- Score: 13.400444194036101
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Data-driven approaches recently achieved remarkable success in magnetic
resonance imaging (MRI) reconstruction, but integration into clinical routine
remains challenging due to a lack of generalizability and interpretability. In
this paper, we address these challenges in a unified framework based on
generative image priors. We propose a novel deep neural network based
regularizer which is trained in a generative setting on reference magnitude
images only. After training, the regularizer encodes higher-level domain
statistics which we demonstrate by synthesizing images without data. Embedding
the trained model in a classical variational approach yields high-quality
reconstructions irrespective of the sub-sampling pattern. In addition, the
model shows stable behavior when confronted with out-of-distribution data in
the form of contrast variation. Furthermore, a probabilistic interpretation
provides a distribution of reconstructions and hence allows uncertainty
quantification. To reconstruct parallel MRI, we propose a fast algorithm to
jointly estimate the image and the sensitivity maps. The results demonstrate
competitive performance, on par with state-of-the-art end-to-end deep learning
methods, while preserving the flexibility with respect to sub-sampling patterns
and allowing for uncertainty quantification.
Related papers
- Steerable Conditional Diffusion for Out-of-Distribution Adaptation in Medical Image Reconstruction [75.91471250967703]
We introduce a novel sampling framework called Steerable Conditional Diffusion.
This framework adapts the diffusion model, concurrently with image reconstruction, based solely on the information provided by the available measurement.
We achieve substantial enhancements in out-of-distribution performance across diverse imaging modalities.
arXiv Detail & Related papers (2023-08-28T08:47:06Z) - Compressed Sensing MRI Reconstruction Regularized by VAEs with
Structured Image Covariance [7.544757765701024]
This paper investigates how generative models, trained on ground-truth images, can be used changesas priors for inverse problems.
We utilize variational autoencoders (VAEs) that generate not only an image but also a covariance uncertainty matrix for each image.
We compare our proposed learned regularization against other unlearned regularization approaches and unsupervised and supervised deep learning methods.
arXiv Detail & Related papers (2022-10-26T09:51:49Z) - PUERT: Probabilistic Under-sampling and Explicable Reconstruction
Network for CS-MRI [47.24613772568027]
Compressed Sensing MRI aims at reconstructing de-aliased images from sub-Nyquist sampling k-space data to accelerate MR Imaging.
We propose a novel end-to-end Probabilistic Under-sampling and Explicable Reconstruction neTwork, dubbed PUERT, to jointly optimize the sampling pattern and the reconstruction network.
Experiments on two widely used MRI datasets demonstrate that our proposed PUERT achieves state-of-the-art results in terms of both quantitative metrics and visual quality.
arXiv Detail & Related papers (2022-04-24T04:23:57Z) - Towards performant and reliable undersampled MR reconstruction via
diffusion model sampling [67.73698021297022]
DiffuseRecon is a novel diffusion model-based MR reconstruction method.
It guides the generation process based on the observed signals.
It does not require additional training on specific acceleration factors.
arXiv Detail & Related papers (2022-03-08T02:25:38Z) - Federated Learning of Generative Image Priors for MRI Reconstruction [5.3963856146595095]
Multi-institutional efforts can facilitate training of deep MRI reconstruction models, albeit privacy risks arise during cross-site sharing of imaging data.
We introduce a novel method for MRI reconstruction based on Federated learning of Generative IMage Priors (FedGIMP)
FedGIMP leverages a two-stage approach: cross-site learning of a generative MRI prior, and subject-specific injection of the imaging operator.
arXiv Detail & Related papers (2022-02-08T22:17:57Z) - Conditional Variational Autoencoder for Learned Image Reconstruction [5.487951901731039]
We develop a novel framework that approximates the posterior distribution of the unknown image at each query observation.
It handles implicit noise models and priors, it incorporates the data formation process (i.e., the forward operator), and the learned reconstructive properties are transferable between different datasets.
arXiv Detail & Related papers (2021-10-22T10:02:48Z) - Adaptive Gradient Balancing for UndersampledMRI Reconstruction and
Image-to-Image Translation [60.663499381212425]
We enhance the image quality by using a Wasserstein Generative Adversarial Network combined with a novel Adaptive Gradient Balancing technique.
In MRI, our method minimizes artifacts, while maintaining a high-quality reconstruction that produces sharper images than other techniques.
arXiv Detail & Related papers (2021-04-05T13:05:22Z) - Using Deep Image Priors to Generate Counterfactual Explanations [38.62513524757573]
A deep image prior (DIP) can be used to obtain pre-images from latent representation encodings.
We propose a novel regularization strategy based on an auxiliary loss estimator jointly trained with the predictor.
arXiv Detail & Related papers (2020-10-22T20:40:44Z) - Deep Variational Network Toward Blind Image Restoration [60.45350399661175]
Blind image restoration is a common yet challenging problem in computer vision.
We propose a novel blind image restoration method, aiming to integrate both the advantages of them.
Experiments on two typical blind IR tasks, namely image denoising and super-resolution, demonstrate that the proposed method achieves superior performance over current state-of-the-arts.
arXiv Detail & Related papers (2020-08-25T03:30:53Z) - Joint reconstruction and bias field correction for undersampled MR
imaging [7.409376558513677]
Undersampling the k-space in MRI allows saving precious acquisition time, yet results in an ill-posed inversion problem.
Deep learning schemes are susceptible to differences between the training data and the image to be reconstructed at test time.
In this work, we address the sensitivity of the reconstruction problem to the bias field and propose to model it explicitly in the reconstruction.
arXiv Detail & Related papers (2020-07-26T12:58:34Z) - Improved Slice-wise Tumour Detection in Brain MRIs by Computing
Dissimilarities between Latent Representations [68.8204255655161]
Anomaly detection for Magnetic Resonance Images (MRIs) can be solved with unsupervised methods.
We have proposed a slice-wise semi-supervised method for tumour detection based on the computation of a dissimilarity function in the latent space of a Variational AutoEncoder.
We show that by training the models on higher resolution images and by improving the quality of the reconstructions, we obtain results which are comparable with different baselines.
arXiv Detail & Related papers (2020-07-24T14:02:09Z)
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.