COAST: COntrollable Arbitrary-Sampling NeTwork for Compressive Sensing
arXiv:2107.07225 · doi:10.1109/TIP.2021.3091834
Abstract
Recent deep network-based compressive sensing (CS) methods have achieved great success. However, most of them regard different sampling matrices as different independent tasks and need to train a specific model for each target sampling matrix. Such practices give rise to inefficiency in computing and suffer from poor generalization ability. In this paper, we propose a novel COntrollable Arbitrary-Sampling neTwork, dubbed COAST, to solve CS problems of arbitrary-sampling matrices (including unseen sampling matrices) with one single model. Under the optimization-inspired deep unfolding framework, our COAST exhibits good interpretability. In COAST, a random projection augmentation (RPA) strategy is proposed to promote the training diversity in the sampling space to enable arbitrary sampling, and a controllable proximal mapping module (CPMM) and a plug-and-play deblocking (PnP-D) strategy are further developed to dynamically modulate the network features and effectively eliminate the blocking artifacts, respectively. Extensive experiments on widely used benchmark datasets demonstrate that our proposed COAST is not only able to handle arbitrary sampling matrices with one single model but also to achieve state-of-the-art performance with fast speed. The source code is available on https://github.com/jianzhangcs/COAST.
Published in IEEE Transactions on Image Processing, 2021
References in corpus (6)
- Denoising Prior Driven Deep Neural Network for Image Restoration
- AMP-Inspired Deep Networks for Sparse Linear Inverse Problems
- A Deep Learning Approach to Structured Signal Recovery
- Compressed Sensing for Energy-Efficient Wireless Telemonitoring of Noninvasive Fetal ECG via Block Sparse Bayesian Learning
- Image Compressive Sensing Recovery Using Adaptively Learned Sparsifying Basis via L0 Minimization
- Compressed Domain Image Classification Using a Dynamic-Rate Neural Network
Cited by in corpus (6)
- Dynamic Path-Controllable Deep Unfolding Network for Compressive Sensing
- Content-aware Scalable Deep Compressed Sensing
- Memory-Augmented Deep Unfolding Network for Compressive Sensing
- Fast Hierarchical Deep Unfolding Network for Image Compressed Sensing
- Nest-DGIL: Nesterov-optimized Deep Geometric Incremental Learning for CS Image Reconstruction
- WTDUN: Wavelet Tree-Structured Sampling and Deep Unfolding Network for Image Compressed Sensing