Single Pixel Imaging Based on Multiple Prior Deep Unfolding Network

Abstract

Single-pixel imaging (SPI), an imaging technique based on the theory of compressed sensing, is limited in real-time imaging and high-resolution images due to its relatively slow imaging speed. In recent years, deep unfolding network compressed sensing reconstruction algorithms based on deep learning have proven to be an effective solution for faster and higher quality image reconstruction. However, existing deep unfolding networks mainly rely on a single piece of a priori information and may ignore other intrinsic structures of the image. Therefore, in this paper, we propose a deep unfolding network (MPDU-Net) that incorporates multiple prior information. To effectively fuse multiple prior information, we propose three different fusion strategies in the deep reconstruction sub-network. An unbiased convolutional layer is used to simulate the sampling reconstruction process to achieve joint reconstruction for effective removal of block artifacts. The sampling matrix is input into the deep reconstruction sub-network as a learnable parameter to achieve joint optimization of sampling reconstruction. Simulation and practical experimental results show that the proposed network outperforms existing compressed sensing reconstruction algorithms based on deep unfolding networks.