Compressed Sensing via a Deep Convolutional Auto-encoder

The nonlinear recovery is not promising in accuracy and speed, which limits the practical usage of compressed sensing (CS). This paper proposes a deep learning-based CS framework which leverages a deep convolutional auto-encoder for image sensing and recovery. The utilized auto-encoder architecture consists of three components: the fully convolutional network acts as an adaptive measurement matrix generator in the encoder; while in the decoder, the deconvolution network and refined reconstruction network are learned for intermediate and final recovery, respectively. Different from most previous work focusing on the block-wise manner to reduce implementation cost but result in blocky artifacts, our adaptive measurement matrix is applicable to any size of scene image and the decoder network reconstructs the whole image efficiently without any blocky artifacts. Moreover, dense connectivity is leveraged to combine multi-level features and alleviate the vanishing-gradient problem in the refined reconstruction network which boosts the performance on image recovery. Compared to the state-of-the-art methods, our algorithm improves more than 0.8 dB in average PSNR.

[1]  Ting Sun,et al.  Single-pixel imaging via compressive sampling , 2008, IEEE Signal Process. Mag..

[2]  Francesco Visin,et al.  A guide to convolution arithmetic for deep learning , 2016, ArXiv.

[3]  Jian Sun,et al.  Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

[4]  David L Donoho,et al.  Compressed sensing , 2006, IEEE Transactions on Information Theory.

[5]  Pavan K. Turaga,et al.  ReconNet: Non-Iterative Reconstruction of Images from Compressively Sensed Measurements , 2016, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[6]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[7]  Trevor Darrell,et al.  Caffe: Convolutional Architecture for Fast Feature Embedding , 2014, ACM Multimedia.

[8]  Michael Zibulevsky,et al.  Block-based compressed sensing of images via deep learning , 2017, 2017 IEEE 19th International Workshop on Multimedia Signal Processing (MMSP).

[9]  Kilian Q. Weinberger,et al.  Densely Connected Convolutional Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[10]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[11]  Lu Gan Block Compressed Sensing of Natural Images , 2007, 2007 15th International Conference on Digital Signal Processing.

[12]  James E. Fowler,et al.  Multiscale block compressed sensing with smoothed projected Landweber reconstruction , 2011, 2011 19th European Signal Processing Conference.

[13]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.