Adaptive Sampling and Memory-augmented Compressed Sensing Algorithm Based on Deep Learning

The deep learning technology has significantly improved the speed and accuracy of compressed sensing reconstruction. However, the existing deep compressive sensing algorithms usually use the same sampling rate to process different blocks of an image, ignoring the fact that different image blocks have different reconstruction difficulties. In this paper, a compressive sensing algorithm with adaptive sampling and memory enhancement is proposed. Firstly, the reconstruction difficulty of different blocks is estimated based on the reconstruction error of the measurement domain. Then, the rules are designed to adaptively assign the sampling rates, and the sampling matrix is used to sample each image block at a specific sampling rate with the help of a sampling rate mask. Furthermore, the two-branch aggregation module is added to the reconstruction network to enhance the interaction of context memory, and the reconstruction ability of the network is improved by adjusting the channel weight of different memory branches. The experimental results show that the proposed algorithm increases the average SSIM by approximately 0.0269 and the average PSNR by approximately 1.66 dB over other methods on several common datasets.