Table of Contents
Fetching ...

Compressive sensing inspired self-supervised single-pixel imaging

Jijun Lu, Yifan Chen, Libang Chen, Yiqiang Zhou, Ye Zheng, Mingliang Chen, Zhe Sun, Xuelong Li

Abstract

Single-pixel imaging (SPI) is a promising imaging modality with distinctive advantages in strongly perturbed environments. Existing SPI methods lack physical sparsity constraints and overlook the integration of local and global features, leading to severe noise vulnerability, structural distortions and blurred details. To address these limitations, we propose SISTA-Net, a compressive sensing-inspired self-supervised method for single-pixel imaging. SISTA-Net unfolds the Iterative Shrinkage-Thresholding Algorithm (ISTA) into an interpretable network consisting of a data fidelity module and a proximal mapping module. The fidelity module adopts a hybrid CNN-Visual State Space Model (VSSM) architecture to integrate local and global feature modeling, enhancing reconstruction integrity and fidelity. We leverage deep nonlinear networks as adaptive sparse transforms combined with a learnable soft-thresholding operator to impose explicit physical sparsity in the latent domain, enabling noise suppression and robustness to interference even at extremely low sampling rates. Extensive experiments on multiple simulation scenarios demonstrate that SISTA-Net outperforms state-of-the-art methods by 2.6 dB in PSNR. Real-world far-field underwater tests yield a 3.4 dB average PSNR improvement, validating its robust anti-interference capability.

Compressive sensing inspired self-supervised single-pixel imaging

Abstract

Single-pixel imaging (SPI) is a promising imaging modality with distinctive advantages in strongly perturbed environments. Existing SPI methods lack physical sparsity constraints and overlook the integration of local and global features, leading to severe noise vulnerability, structural distortions and blurred details. To address these limitations, we propose SISTA-Net, a compressive sensing-inspired self-supervised method for single-pixel imaging. SISTA-Net unfolds the Iterative Shrinkage-Thresholding Algorithm (ISTA) into an interpretable network consisting of a data fidelity module and a proximal mapping module. The fidelity module adopts a hybrid CNN-Visual State Space Model (VSSM) architecture to integrate local and global feature modeling, enhancing reconstruction integrity and fidelity. We leverage deep nonlinear networks as adaptive sparse transforms combined with a learnable soft-thresholding operator to impose explicit physical sparsity in the latent domain, enabling noise suppression and robustness to interference even at extremely low sampling rates. Extensive experiments on multiple simulation scenarios demonstrate that SISTA-Net outperforms state-of-the-art methods by 2.6 dB in PSNR. Real-world far-field underwater tests yield a 3.4 dB average PSNR improvement, validating its robust anti-interference capability.

Paper Structure

This paper contains 18 equations, 9 figures, 2 tables, 2 algorithms.

Figures (9)

  • Figure 1: Overview of the proposed method: (a) Core reconstruction workflow of the SISTA-Net algorithm. (b) Experimental setup for single-pixel imaging.
  • Figure 2: Architecture of the Data Fidelity Module. (a) The architecture of Res2MM-Net. (b) The structure of MamCNN Block. (c) The structure of Res2Mam Multi-Scale Block. (d) The structure of Window Mamba Block. (e) The structure of the Visual State Space Block.
  • Figure 3: Architecture of the Proximal Mapping Module. (a) The structure of the Sparsity Encoder. (b) The structure of the Sparsity Decoder. (c) The structure of the ResBlock.
  • Figure 4: Reconstruction results on binary images.
  • Figure 5: Reconstruction results on complex grayscale natural images.
  • ...and 4 more figures