计算机科学 ›› 2026, Vol. 53 ›› Issue (6): 232-241.doi: 10.11896/jsjkx.250400147

• 计算机图形学&多媒体 • 上一篇    下一篇

MambaCS:基于Mamba的图像压缩感知算法

李秀滢, 陈雪松, 李浩泽, 廖鸿苇, 韩佳萌, 段晓毅   

  1. 北京电子科技学院 北京 100070
  • 收稿日期:2025-04-30 修回日期:2025-08-19 出版日期:2026-06-15 发布日期:2026-06-09
  • 通讯作者: 段晓毅(xiaoyi_duan@sina.com)
  • 作者简介:(lixiuying@besti.edu.cn)
  • 基金资助:
    国家重点研发计划(2022YFB3104402);国家档案局科技项目(2025-Z-009);中央高校基本科研业务费专项资金(3282025045)

MambaCS:Mamba-based Image Compressed Sensing Algorithm

LI Xiuying, CHEN Xuesong, LI Haoze, LIAO Hongwei, HAN Jiameng, DUAN Xiaoyi   

  1. Beijing Electronic Science and Technology Institute,Beijing 100070,China
  • Received:2025-04-30 Revised:2025-08-19 Published:2026-06-15 Online:2026-06-09
  • About author:LI Xiuying,born in 1975,master,associate professor.Her main research interests include intelligent systems security and cryptography.
    DUAN Xiaoyi,born in 1979,Ph.D,associate professor.His main research in-terest is information security.
  • Supported by:
    National Key R & D Program of China(2022YFB3104402),Science and Technology Project of the National Archives Administration of China(2025-Z-009) and Fundamental Research Funds for the Central Universities(3282025045).

摘要: 压缩感知技术在图像采集与重建领域具有广泛应用,其核心目标是通过少量测量数据实现高质量图像重建。相较于传统方法,基于深度学习的图像压缩感知能够实现更高的重构质量与更低的计算成本,但如何提升低采样率下的重建精度仍是亟待解决的关键问题。为此,提出了一种基于Mamba的深度展开框架(MambaCS),通过创新性的设计显著提升了图像压缩感知的重建质量。在采样阶段,采用分块测量策略,以平衡计算复杂度和采样效率。在重建阶段,引入残差状态空间模块,利用Mamba在长序列建模上的优势,增强网络对复杂图像结构的建模能力。此外,为了充分挖掘测量数据的潜在信息,在算法中添加了多通道重用块,通过多尺度特征融合和测量值复用,增强网络对关键特征的提取能力。实验证明,在多个公开数据集上,MambaCS在重建质量方面均超越了现有先进方法。

关键词: 压缩感知, 深度学习, Mamba, 图像重建, 状态空间模型

Abstract: Compressed sensing technology has been widely applied in the field of image acquisition and reconstruction,with its core objective being the accurate and high-quality reconstruction of images from a significantly reduced number of measurements.Compared to conventional signal processing approaches,deep learning-based image compressed sensing demonstrates superior performance in terms of reconstruction quality while also achieving reduced computational costs.Nevertheless,enhancing reconstruction accuracy under extremely low sampling rates remains a critical challenge.To address this issue,this paper proposes a novel deep unfolding framework named MambaCS,which significantly improves the performance of image compressed sensing through several innovative architectural components.During the sampling phase,a block-based measurement strategy is adopted to effectively balance computational complexity and sampling efficiency.In the reconstruction phase,the residual state space mo-dule is introduced,leveraging the advantages of Mamba in capturing long-range dependencies to better model complex spatial structures within images.Furthermore,in order to fully utilize the latent information contained in the measurement data,a multi-channel reuse block is incorporated into the reconstruction process.This module integrates multi-scale feature fusion with measurement reuse techniques,enhancing the network's ability to extract and represent key image features.Extensive experiments de-monstrate that MambaCS consistently outperforms existing state-of-the-art methods in terms of reconstruction accuracy and vi-sual quality.

Key words: Compressive sensing, Deep learning, Mamba, Image reconstruction, State space model

中图分类号: 

  • TP391
[1]KULKARN K,LOHIT S,TURAGA P,et al.ReconNet:Non-iterative reconstruction of images from compressively sensed measurements[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition(CVPR).Las Vegas,NV,USA:IEEE,2016:449-458.
[2]YAO H,DAI F,ZHANG S,et al.DR2-Net:Deep residual reconstruction network for image compressive sensing[J].Neurocomputing,2019,359(1):483-493.
[3]SHI W,JIANG F,LIU S,et al.Image compressed sensing using convolutional neural network[J].IEEE Transactions on Image Processing,2020,29:375-388.
[4]SHEN M,GAN H,NING C,et al.TransCS:A transformer-based hybrid architecture for image compressed sensing[J].IEEE Transactions on Image Processing,2022,31:6991-7005.
[5]YE D,NI Z,WANG H,et al.CSformer:Bridging convolution and transformer for compressive sensing[J].IEEE Transactions on Image Processing,2023,32:2827-2842.
[6]SHANNON C E.Communication in the presence of noise[J].Proceedings of the IRE,1949,37(1):10-21.
[7]NYQUIS H.Certain topics in telegraph transmission theory[J].Transactions of the American Institute of Electrical Engineers,1928,47(2):617-644.
[8]DONOHO D L.Compressed sensing[J].IEEE Transactions on Information Theory,2006,52(4):1289-1306.
[9]CHEN S S,DONOHO D L,SAUNDERS M A.Atomic decomposition by basis pursuit[J].SIAM Review,2001,43(1):129-159.
[10]DO T T,GAN L,NGUYEN N,et al.Sparsity adaptive matching pursuit algorithm for practical compressed sensing[C]//Proceedings of the 42nd Asilomar Conference on Signals,Systems and Computers.Pacific Grove,CA,USA:IEEE,2008:581-587.
[11]GU A,DAO T.Mamba:Linear-time sequence modeling with selective state spaces[J].arXiv:2312.00752,2023.
[12]GU A,GOEL K,Rè C.Efficiently modeling long sequences with structured state spaces[J].arXiv:2111.00396,2021.
[13]GU A,JOHNSON I,GOEL K,et al.Combining recurrent,convolutional,and continuous-time models with linear state space layers[J].Advances in Neural Information Processing Systems,2021,34:572-585.
[14]SMITH J T,WARRINGTON A,LINDERMAN S W.Simplified state space layers for sequence modeling[J].arXiv:2208.04933,2022.
[15]ZHU L,LIAO B,ZHANG Q,et al.Vision mamba:Efficient visual representation learning with bidirectional state space model[J].arXiv:2401.09417,2024.
[16]LIU Y,TIAN Y,ZHAO Y,et al.VMamba:Visual state space model[J].arXiv:2401.10166,2024.
[17]WANG C X,TSEPA O,MA J,et al.Graph-Mamba:Towards long-range graph sequence modeling with selective state spaces[J].arXiv:2402.00789,2024.
[18]MA J,LI F,WANG B.U-mamba:Enhancing long-range de-pendency for biomedical image segmentation[J].arXiv:2401.04722,2024.
[19]LIU J,YANG H,ZHOU H,et al.Swin-UMamba:Mamba-based UNet with ImageNet-based pretraining[J].arXiv:2402.03302,2024.
[20]WANG Z,ZHENG J,ZHANG Y,et al.Mamba-UNet:UNet-like pure visual Mamba for medical image segmentation[J].ar-Xiv:2402.05079,2024.
[21]GUO H,LI J,DAI T,et al.MambaIR:A simple baseline for image restoration with state-space model[J].arXiv:2402.15648,2024.
[22]FAN Z E,LIAN F,QUAN J N.Global sensing and measure-ments reuse for image compressed sensing[C]//2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).New Orleans,LA,USA:IEEE,2022:8954-8963.
[23]BEVILACQUA M,ROUM A,GUILLEMOT C,et al.Low complexity single-image super-resolution based on nonnegative neighbor embedding[C]//2012 19th IEEE International Confe-rence on Image Processing.Orlando,FL,USA:IEEE,2012:1489-1492.
[24]ZEYDE R,ELAD M,PROTTER M.On single image scale-up using sparse-representations[C]//Curves and Surfaces:7th International Conference,Avignon,France,Revised Selected Papers.Berlin:Springer,2012:711-730.
[25]MARTIN D,FOWLKES C,TAL D,et al.A database of human segmented natural images and its application to evaluating segmentation algorithms[C]//2001 IEEE International Conference on Computer Vision(ICCV).Vancouver,Canada:IEEE,2001,2:416-423.
[26]HUANG J B,SINGH A,AHUJA N.Single image super-resolu-tion from transformed self-exemplars[C]//2015 IEEE Confe-rence on Computer Vision and Pattern Recognition(CVPR).Boston,MA,USA:IEEE,2015:5197-5206.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!