计算机科学 ›› 2026, Vol. 53 ›› Issue (6): 232-241.doi: 10.11896/jsjkx.250400147
李秀滢, 陈雪松, 李浩泽, 廖鸿苇, 韩佳萌, 段晓毅
LI Xiuying, CHEN Xuesong, LI Haoze, LIAO Hongwei, HAN Jiameng, DUAN Xiaoyi
摘要: 压缩感知技术在图像采集与重建领域具有广泛应用,其核心目标是通过少量测量数据实现高质量图像重建。相较于传统方法,基于深度学习的图像压缩感知能够实现更高的重构质量与更低的计算成本,但如何提升低采样率下的重建精度仍是亟待解决的关键问题。为此,提出了一种基于Mamba的深度展开框架(MambaCS),通过创新性的设计显著提升了图像压缩感知的重建质量。在采样阶段,采用分块测量策略,以平衡计算复杂度和采样效率。在重建阶段,引入残差状态空间模块,利用Mamba在长序列建模上的优势,增强网络对复杂图像结构的建模能力。此外,为了充分挖掘测量数据的潜在信息,在算法中添加了多通道重用块,通过多尺度特征融合和测量值复用,增强网络对关键特征的提取能力。实验证明,在多个公开数据集上,MambaCS在重建质量方面均超越了现有先进方法。
中图分类号:
| [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. |
|
||