Overcoming Measurement Inconsistency in Deep Learning for Linear Inverse
Problems: Applications in Medical Imaging
- URL: http://arxiv.org/abs/2011.14387v2
- Date: Mon, 31 May 2021 10:28:45 GMT
- Title: Overcoming Measurement Inconsistency in Deep Learning for Linear Inverse
Problems: Applications in Medical Imaging
- Authors: Marija Vella, Jo\~ao F. C. Mota
- Abstract summary: Deep neural networks (DNNs) are the method of choice for solving linear inverse problems.
We propose a framework that post-processes the output of DNNs with an optimization algorithm that enforces measurement consistency.
Experiments on MR images show that enforcing measurement consistency via our method can lead to large gains in reconstruction performance.
- Score: 0.9137554315375922
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: The remarkable performance of deep neural networks (DNNs) currently makes
them the method of choice for solving linear inverse problems. They have been
applied to super-resolve and restore images, as well as to reconstruct MR and
CT images. In these applications, DNNs invert a forward operator by finding,
via training data, a map between the measurements and the input images. It is
then expected that the map is still valid for the test data. This framework,
however, introduces measurement inconsistency during testing. We show that such
inconsistency, which can be critical in domains like medical imaging or
defense, is intimately related to the generalization error. We then propose a
framework that post-processes the output of DNNs with an optimization algorithm
that enforces measurement consistency. Experiments on MR images show that
enforcing measurement consistency via our method can lead to large gains in
reconstruction performance.
Related papers
- Chasing Better Deep Image Priors between Over- and Under-parameterization [63.8954152220162]
We study a novel "lottery image prior" (LIP) by exploiting DNN inherent sparsity.
LIPworks significantly outperform deep decoders under comparably compact model sizes.
We also extend LIP to compressive sensing image reconstruction, where a pre-trained GAN generator is used as the prior.
arXiv Detail & Related papers (2024-10-31T17:49:44Z) - Measurement-Consistent Networks via a Deep Implicit Layer for Solving
Inverse Problems [0.0]
End-to-end deep neural networks (DNNs) have become state-of-the-art (SOTA) for solving inverse problems.
These networks are sensitive to minor variations in the training pipeline and often fail to reconstruct small but important details.
We propose a framework that transforms any DNN for inverse problems into a measurement-consistent one.
arXiv Detail & Related papers (2022-11-06T17:05:04Z) - Denoising Diffusion Restoration Models [110.1244240726802]
Denoising Diffusion Restoration Models (DDRM) is an efficient, unsupervised posterior sampling method.
We demonstrate DDRM's versatility on several image datasets for super-resolution, deblurring, inpainting, and colorization.
arXiv Detail & Related papers (2022-01-27T20:19:07Z) - Solving Inverse Problems in Medical Imaging with Score-Based Generative
Models [87.48867245544106]
Reconstructing medical images from partial measurements is an important inverse problem in Computed Tomography (CT) and Magnetic Resonance Imaging (MRI)
Existing solutions based on machine learning typically train a model to directly map measurements to medical images.
We propose a fully unsupervised technique for inverse problem solving, leveraging the recently introduced score-based generative models.
arXiv Detail & Related papers (2021-11-15T05:41:12Z) - Image Restoration by Deep Projected GSURE [115.57142046076164]
Ill-posed inverse problems appear in many image processing applications, such as deblurring and super-resolution.
We propose a new image restoration framework that is based on minimizing a loss function that includes a "projected-version" of the Generalized SteinUnbiased Risk Estimator (GSURE) and parameterization of the latent image by a CNN.
arXiv Detail & Related papers (2021-02-04T08:52:46Z) - Deep Probabilistic Imaging: Uncertainty Quantification and Multi-modal
Solution Characterization for Computational Imaging [11.677576854233394]
We propose a variational deep probabilistic imaging approach to quantify reconstruction uncertainty.
Deep Probabilistic Imaging employs an untrained deep generative model to estimate a posterior distribution of an unobserved image.
arXiv Detail & Related papers (2020-10-27T17:23:09Z) - 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) - Data Consistent CT Reconstruction from Insufficient Data with Learned
Prior Images [70.13735569016752]
We investigate the robustness of deep learning in CT image reconstruction by showing false negative and false positive lesion cases.
We propose a data consistent reconstruction (DCR) method to improve their image quality, which combines the advantages of compressed sensing and deep learning.
The efficacy of the proposed method is demonstrated in cone-beam CT with truncated data, limited-angle data and sparse-view data, respectively.
arXiv Detail & Related papers (2020-05-20T13:30:49Z) - Improving the Interpretability of fMRI Decoding using Deep Neural
Networks and Adversarial Robustness [1.254120224317171]
A saliency map is a common approach for producing interpretable visualizations of the relative importance of input features for a prediction.
In this paper, we review a variety of methods for producing gradient-based saliency maps, and present a new adversarial training method we developed to make DNNs robust to input noise.
arXiv Detail & Related papers (2020-04-23T12:56:24Z) - Improving Robustness of Deep-Learning-Based Image Reconstruction [24.882806652224854]
We show that for inverse problem solvers, one should analyze and study the effect of adversaries in the measurement-space.
We introduce an auxiliary network to generate adversarial examples, which is used in a min-max formulation to build robust image reconstruction networks.
We find that a linear network using the proposed min-max learning scheme indeed converges to the same solution.
arXiv Detail & Related papers (2020-02-26T22:12:36Z)
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.