Authors: Jijun Lu, Yifan Chen, Libang Chen, Yiqiang Zhou, Ye Zheng, Mingliang Chen, Zhe Sun, Xuelong Li
Organizations: Institute of Artificial Intelligence (TeleAI), China Telecom, China · School of Integrated Circuits, Shanghai Jiao Tong University, Shanghai, China · College of Future Information Technology, Fudan University, Shanghai, China · Aerospace Laser Technology and System Department, Shanghai Institute of Optics and Fine Mechanics, Chinese Academy of Sciences, Shanghai, China · School of Artificial Intelligence, Optics and Electronics (iOPEN), Northwestern Polytechnical University, Xi’an, China · Fujian Ocean Innovation Center, Xiamen, China
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.
Figures & tables
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 Vision 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. The markers (A)–(I) denote DGI, ISTA-Net++, TransCS, Restormer, URNet, GILM, UNet, GIDC, and Ours in order.
Method
Bird
Grass
Raven
Tower
PSNR
SSIM
PSNR-HF
PSNR
SSIM
PSNR-HF
PSNR
SSIM
PSNR-HF
PSNR
SSIM
PSNR-HF
DGI
7.67
0.0288
18.40
6.91
0.0442
15.79
7.30
0.0224
13.00
7.38
0.0213
20.37
ISTA-Net++
6.23
0.0497
19.52
7.19
0.0875
16.27
7.08
0.0390
13.40
7.16
0.0273
22.15
TransCS
6.47
0.0400
19.36
6.79
0.0542
15.93
7.02
0.0212
13.19
9.26
0.0257
22.18
Restormer
14.18
0.1375
16.21
12.40
0.2414
14.47
10.45
0.0589
11.68
17.76
0.0823
19.58
URNet
16.44
0.1297
21.36
13.10
0.1775
17.07
12.48
0.0653
13.05
14.14
0.0512
18.59
Table 1: Quantitative comparison for simulated binary and grayscale reconstructions.
Figure 5: Reconstruction results on complex grayscale natural images.
Figure 6: Detail reconstruction results. (a) Reconstructed images and their locally enlarged regions. (b) Vertical grayscale intensity profiles at the mark.
Figure 7: Convergence and stability analysis. (a) PSNR curves of different models under five random seeds. (b) Intermediate SISTA-Net reconstruction results at selected epochs under random seed 0.
Figure 8: Reconstruction results over different sampling ratios. The markers (A)–(I) denote DGI, ISTA-Net++, TransCS, Restormer, URNet, GILM, UNet, GIDC, and Ours in order.
Figure 9: Evolution of sparse latent feature maps.
SR(N)
Epoch
Mean
E1%
E3%
E5%
2.93% (3k)
200
0.0393
0.7821
0.8717
0.9123
400
0.0051
0.9586
0.9812
0.9891
800
0.0014
0.9521
0.9796
0.9884
5.86% (6k)
200
0.0250
0.8648
0.9249
0.9501
400
0.0033
0.9697
0.9869
0.9925
800
0.0010
0.9652
0.9860
0.9922
Table 2: Latent feature metrics.
Figure 10: Reconstruction results under Gaussian, Poisson, and mixed noise. Noise strength increases from left to right.
Gaussian noise
Level
60 dB
55 dB
50 dB
45 dB
40 dB
Threshold
0.0345
0.0353
0.0391
0.0412
0.0472
Poisson noise
Level
1×106
5×105
1×105
5×104
1×104
Threshold
0.0368
0.0389
0.0385
0.0403
0.0465
Mixed noise
Table 3: Learnable thresholds under different noise levels.
Figure 11: Experimental setup for underwater SPI.
Figure 12: Reconstruction and ablation results.(I) Reconstruction results. (II) Network module ablation. (III) Loss function ablation.
Single-pixel sensing encodes a scene as a short sequence of coded measurements, and image-free methods infer the task directly from that sequence. We show that removing image reconstruction relocates the central design problem to the lift: how 1D measurements become a 2D task representation. We organize this choice as a lift spectrum from a fixed-physics inverse, through a learned static projection, to content-adaptive retrieval. These are not interchangeable forms of reconstruction: the fixed-physics route reconstructs an image consumed at inference, whereas our spatiotemporal soft-fusion (STSF) network lifts measurements directly into task features, and task-prioritized loss scheduling (TPLS) uses a separate learned reconstruction branch only as scheduled training supervision. A probe-selected recurrent encoder and a parameter-matched lift ablation identify the STSF design. In simulation, STSF+TPLS exceeds the prior image-free baseline on three datasets at 3.13% sampling (+3.2 to +9.9 pp foreground mIoU) and remains competitive down to 0.39%. The strongest clean-trained reconstruct-then-segment baseline wins without measurement noise, but measurement noise reverses the ranking: the reconstructed task input carries a 20-70x larger normalized relative perturbation than the measurements themselves. Stressed to failure, the three lift regions exhibit distinct dominant signatures--collapse, imprinting, and coarsening. STSF+TPLS transfers without fine-tuning to a real single-pixel bench, where the reversal reappears as a proof of concept; inference takes about 14 ms per mask on an RTX 4090. Within the tested fixed-acquisition regime, measurement-to-space adaptivity therefore organizes both the clean-to-noisy operating envelope and the failure a system encounters. Code and pretrained weights: https://github.com/Hanyuyuan6/STSF-TPLS.
Yuyuan Han, Jingwei Li, Xiaoxia Zhang +9
Electronic Materials Research Laboratory, Key Laboratory of the Ministry of Education and International Center for Dielectric Research, School of Electronic Science and Engineering, Xi’an Jiaotong University, Xi’an 710049, China · Micius Laboratory, Henan Academy of Sciences, Zhengzhou 450052, China · Department of Pharmacy, the First Affiliated Hospital of Xi’an Jiaotong University, Xi’an 710061, China +1
Despite offering high sensitivity, a high signal-to-noise ratio, and a broad spectral range, single-pixel imaging (SPI) is limited by low measurement efficiency and long data-acquisition times. To address this, we propose a wavelength-multiplexed, spatially incoherent diffractive optical processor combined with a compact/shallow digital artificial neural network (ANN) to implement compressive SPI. Specifically, we model the bucket detection process in conventional SPI as a linear intensity transformation with spatially and spectrally varying point-spread functions. This transformation matrix is treated as a learnable parameter and jointly optimized with a shallow digital ANN composed of 2 hidden nonlinear layers. The wavelength-multiplexed diffractive processor is then configured via data-free optimization to approximate this pre-trained transformation matrix; after this optimization, the diffractive processor remains static/fixed. Upon multi-wavelength illumination and diffractive modulation, the target spatial information of the input object is spectrally encoded. A single-pixel detector captures the output spectral power at each illumination band, which is then rapidly decoded by the jointly trained digital ANN to reconstruct the input image. In addition to our numerical analyses demonstrating the feasibility of this approach, we experimentally validated its proof-of-concept using an array of light-emitting diodes (LEDs). Overall, this work demonstrates a computational imaging framework for compressive SPI that can be useful in applications such as biomedical imaging, autonomous devices, and remote sensing.
Xiao Wang, Yiyang Wu, Yuntian Wang +8
Electrical and Computer Engineering Department, University of California, Los Angeles, CA, 90095, USA · Bioengineering Department, University of California, Los Angeles, CA, 90095, USA · California NanoSystems Institute (CNSI), University of California, Los Angeles, CA, 90095, USA
Single-pixel imaging (SPI) offers a cost-effective route to hyperspectral acquisition but struggles to recover high-fidelity spatial and spectral details under extremely low sampling rates, a severely ill-posed inverse problem. While deep learning has shown potential, existing data-driven methods demand large-scale pretraining datasets that are often impractical in hyperspectral imaging. To overcome this limitation, we propose an end-to-end physics-informed framework that leverages untrained neural networks and RGB guidance for joint hyperspectral reconstruction and super-resolution without any external training data. The framework comprises three physically grounded stages: (1) a Regularized Least-Squares method with RGB-derived Grayscale Priors (LS-RGP) that initializes the solution by exploiting cross-modal structural correlations; (2) an Untrained Hyperspectral Recovery Network (UHRNet) that refines the reconstruction through measurement consistency and hybrid regularization; and (3) a Transformer-based Untrained Super-Resolution Network (USRNet) that upsamples the spatial resolution via cross-modal attention, transferring high-frequency details from the RGB guide. Extensive experiments on benchmark datasets demonstrate that our approach significantly surpasses state-of-the-art algorithms in both reconstruction accuracy and spectral fidelity. Moreover, a proof-of-concept experiment using a physical single-pixel imaging system validates the framework's practical applicability, successfully reconstructing a 144-band hyperspectral data cube at a mere 6.25% sampling rate. The proposed method thus provides a robust, data-efficient solution for computational hyperspectral imaging.
Hao Zhang, Bilige Xu, Lichen Wei +2
College of Science, Northwest Agriculture & Forestry University, Yangling 712100, China · Key Laboratory of Photoelectronic Imaging Technology and System, School of Optics and Photonics, Beijing Institute of Technology, Beijing, China · State Key Laboratory of Digital Manufacturing Equipment and Technology, Huazhong University of Science and Technology, Wuhan 430074, China