DL-CSNet: Dictionary Learning based Compressed Sensing Neural Network

Abstract

In this paper, we propose a novel neural network for Compressed Sensing (CS) application: the Dictionary Learning based Compressed Sensing neural Network (DL-CSNet). It is fairly simple but highly effective, which consists of only three layers: 1) a DL layer for latent sparse features extraction; 2) a smoothing layer via Total Variation (TV) like constraint; and 3) a CS acquisition layer for neural network training. In particular, the TV-like smoothing layer is a perfect complement to the sparsity-oriented DL layer to achieve smooth images. The trained DL-CSNet can learn the optimal dictionary matrix so that images can be reconstructed in high quality. At last, extensive experiments have been carried out on binary images and compared to most classical CS algorithms, which shows the superior performance of the proposed DL-CSNet.