Scalable Deep Compressive Sensing

Abstract

Deep learning has been used to image compressive sensing (CS) for enhanced reconstruction performance. However, most existing deep learning methods train different models for different subsampling ratios, which brings additional hardware burden. In this paper, we develop a general framework named scalable deep compressive sensing (SDCS) for the scalable sampling and reconstruction (SSR) of all existing end-to-end-trained models. In the proposed way, images are measured and initialized linearly. Two sampling masks are introduced to flexibly control the subsampling ratios used in sampling and reconstruction, respectively. To make the reconstruction model adapt to any subsampling ratio, a training strategy dubbed scalable training is developed. In scalable training, the model is trained with the sampling matrix and the initialization matrix at various subsampling ratios by integrating different sampling matrix masks. Experimental results show that models with SDCS can achieve SSR without changing their structure while maintaining good performance, and SDCS outperforms other SSR methods.

References in corpus (12)

• Message Passing Algorithms for Compressed Sensing
• Denoising Prior Driven Deep Neural Network for Image Restoration
• AMP-Inspired Deep Networks for Sparse Linear Inverse Problems
• DR2-Net: Deep Residual Reconstruction Network for Image Compressive Sensing
• Compressed Sensing using Generative Models
• Learned D-AMP: Principled Neural Network based Compressive Image Recovery
• Theoretical Linear Convergence of Unfolded ISTA and its Practical Weights and Thresholds
• One Network to Solve Them All --- Solving Linear Inverse Problems using Deep Projection Models
• A new compressive video sensing framework for mobile broadcast
• Deep Compressed Sensing
• Compressed Domain Image Classification Using a Dynamic-Rate Neural Network
• Rate-Adaptive Neural Networks for Spatial Multiplexers