Compressed Sensing via Measurement-Conditional Generative Models

Abstract

A pre-trained generator has been frequently adopted in compressed sensing (CS) due to its ability to effectively estimate signals with the prior of NNs. In order to further refine the NN-based prior, we propose a framework that allows the generator to utilize additional information from a given measurement for prior learning, thereby yielding more accurate prediction for signals. As our framework has a simple form, it is easily applied to existing CS methods using pre-trained generators. We demonstrate through extensive experiments that our framework exhibits uniformly superior performances by large margin and can reduce the reconstruction error up to an order of magnitude for some applications. We also explain the experimental success in theory by showing that our framework can slightly relax the stringent signal presence condition, which is required to guarantee the success of signal recovery.

References in corpus (21)

• Auto-Encoding Variational Bayes
• Conditional Generative Adversarial Nets
• Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks
• Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks
• Generative Adversarial Text to Image Synthesis
• fastMRI: An Open Dataset and Benchmarks for Accelerated MRI
• AMP-Inspired Deep Networks for Sparse Linear Inverse Problems
• Compressed Sensing using Generative Models
• Learned D-AMP: Principled Neural Network based Compressive Image Recovery
• Deep Decoder: Concise Image Representations from Untrained Non-convolutional Networks
• Compressed Sensing with Deep Image Prior and Learned Regularization
• Theoretical Linear Convergence of Unfolded ISTA and its Practical Weights and Thresholds
• Neural Proximal Gradient Descent for Compressive Imaging
• Algorithmic Guarantees for Inverse Imaging with Untrained Network Priors
• Deep Compressed Sensing
• Modeling Sparse Deviations for Compressed Sensing using Generative Models
• Learning a Compressed Sensing Measurement Matrix via Gradient Unrolling
• Uncertainty Autoencoders: Learning Compressed Representations via Variational Information Maximization
• Approximate Message Passing with Restricted Boltzmann Machine Priors
• Task-Aware Compressed Sensing with Generative Adversarial Networks
• From Bayesian Sparsity to Gated Recurrent Nets