mlsanityData to 7 Oct 2026

CU-Mamba: Selective State Space Models with Channel Learning for Image Restoration

arXiv 2404.11778 ↗First posted 17 Apr 2024cs.CVcited by 24

On the map: State-space sequence models (S4, Mamba) → Mamba for image restoration

Abstract (arXiv)

Reconstructing degraded images is a critical task in image processing. Although CNN and Transformer-based models are prevalent in this field, they exhibit inherent limitations, such as inadequate long-range dependency modeling and high computational costs. To overcome these issues, we introduce the Channel-Aware U-Shaped Mamba (CU-Mamba) model, which incorporates a dual State Space Model (SSM) framework into the U-Net architecture. CU-Mamba employs a Spatial SSM module for global context encoding and a Channel SSM component to preserve channel correlation features, both in linear computational complexity relative to the feature map size. Extensive experimental results validate CU-Mamba's superiority over existing state-of-the-art methods, underscoring the importance of integrating both spatial and channel contexts in image restoration.

Key facts. Cited by 24 papers in the map (data through 7 Oct 2026). Placed in Mamba for image restoration, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 39 of its 51 references that are in the map. First posted on arXiv 17 Apr 2024.

Builds on

39 of 51 references.

References · 39

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  4. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
  5. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
  6. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  7. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
  8. 2021SwinIR: Image Restoration Using Swin Transformercited by 1,869
  9. 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,618
  10. 2021MUSIQ: Multi-scale Image Quality Transformercited by 1,058
  11. 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,051
  12. 2021Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Datacited by 961
27 more
  1. 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 864
  2. 2021Multi-Stage Progressive Image Restorationcited by 813
  3. 2022Simplified State Space Layers for Sequence Modelingcited by 776
  4. 2016Deep Multi-scale Convolutional Neural Network for Dynamic Scene Deblurringcited by 722
  5. 2022Simple Baselines for Image Restorationcited by 701
  6. 2021Designing a Practical Degradation Model for Deep Blind Image Super-Resolutioncited by 524
  7. 2024U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentationcited by 412
  8. 2021Pay Attention to MLPscited by 367
  9. 2019DeblurGAN-v2: Deblurring (Orders-of-Magnitude) Faster and Bettercited by 361
  10. 2021Rethinking Coarse-to-Fine Approach in Single Image Deblurringcited by 324
  11. 2024VM-UNet: Vision Mamba UNet for Medical Image Segmentationcited by 286
  12. 2024Scaling Up to Excellence: Practicing Model Scaling for Photo-Realistic Image Restoration In the Wildcited by 281
  13. 2021HINet: Half Instance Normalization Network for Image Restorationcited by 236
  14. 2017Benchmarking Denoising Algorithms with Real Photographscited by 209
  15. 2019Deep Stacked Hierarchical Multi-patch Network for Image Deblurringcited by 205
  16. 2023Learning A Sparse Transformer Network for Effective Image Derainingcited by 204
  17. 2020Deblurring by Realistic Blurringcited by 187
  18. 2019Real Image Denoising with Feature Attentioncited by 176
  19. 2020Human-Aware Motion Deblurringcited by 174
  20. 2022DaViT: Dual Attention Vision Transformerscited by 174
  21. 2022Efficient Frequency Domain-based Transformers for High-Quality Image Deblurringcited by 164
  22. 2022Stripformer: Strip Transformer for Fast Image Deblurringcited by 139
  23. 2020Dual Adversarial Network: Toward Real-world Noise Removal and Noise Generationcited by 101
  24. 2021Spatially-Adaptive Image Restoration using Distortion-Guided Networkscited by 77
  25. 2021Dynamic Attentive Graph Learning for Image Restorationcited by 48
  26. 2024MambaMixer: Efficient Selective State Space Models with Dual Token and Channel Selectioncited by 40
  27. 2023Spectral Enhanced Rectangle Transformer for Hyperspectral Image Denoisingcited by 35

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 16; 2025: 7; 2026: 1.

24 papers.

Sort:

By area:

  1. 2024Vision Mamba: A Comprehensive Survey and Taxonomycited by 53
  2. 2024MaIR: A Locality- and Continuity-Preserving Mamba for Image Restorationcited by 43
  3. 2024Visual Mamba: A Survey and New Outlookscited by 41
  4. 2024A Survey of Mambacited by 39
  5. 2024Efficient Visual State Space Model for Image Deblurringcited by 27
  6. 2024Mamba in Vision: A Comprehensive Survey of Techniques and Applicationscited by 18
  7. 2024Hi-Mamba: Hierarchical Mamba for Efficient Image Super-Resolutioncited by 16
  8. 2024DenoMamba: A fused state-space model for low-dose CT denoisingcited by 11
  9. 2024Wave-U-Mamba: An End-To-End Framework For High-Quality And Efficient Speech Super Resolutioncited by 9
  10. 2024LFMamba: Light Field Image Super-Resolution with State Space Modelcited by 7
  11. 2025MODEM: A Morton-Order Degradation Estimation Mechanism for Adverse Weather Image Recoverycited by 6
  12. 2025M2Restore: Mixture-of-Experts-based Mamba-CNN Fusion Framework for All-in-One Image Restorationcited by 6
12 more
  1. 2024Alignment-free Raw Video Demoireingcited by 3
  2. 2025CymbaDiff: Structured Spatial Diffusion for Sketch-based 3D Semantic Urban Scene Generationcited by 3
  3. 2024Enhanced Prediction of Multi-Agent Trajectories via Control Inference and State-Space Dynamicscited by 2
  4. 2025DiNAT-IR: Exploring Dilated Neighborhood Attention for High-Quality Image Restorationcited by 2
  5. 2024DSDFormer: An Innovative Transformer-Mamba Framework for Robust High-Precision Driver Distraction Identificationcited by 1
  6. 2024SkelMamba: A State Space Model for Efficient Skeleton Action Recognition of Neurological Disorderscited by 1
  7. 2026M3SR: Multi-Scale Multi-Perceptual Mamba for Efficient Spectral Reconstructioncited by 1
  8. 2024HandS3C: 3D Hand Mesh Reconstruction with State Space Spatial Channel Attention from RGB imagescited by 0
  9. 2024Cross-Scan Mamba with Masked Training for Robust Spectral Imagingcited by 0
  10. 2025ControlMambaIR: Conditional Controls with State-Space Model for Image Restorationcited by 0
  11. 2025Few-Shot Object Detection via Spatial-Channel State Space Modelcited by 0
  12. 2025WaMaIR: Image Restoration via Multiscale Wavelet Convolutions and Mamba-based Channel Modeling with Texture Enhancementcited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Model
  2. 2024Multi-Scale Representation Learning for Image Restoration with State-Space Model
  3. 2026Progressive Split Mamba: Effective State Space Modelling for Image Restoration
  4. 2024MambaIRv2: Attentive State Space Restoration
  5. 2024VmambaIR: Visual State Space Model for Image Restoration
  6. 2025MatIR: A Hybrid Mamba-Transformer Image Restoration Model
  7. 2025Directing Mamba to Complex Textures: An Efficient Texture-Aware State Space Model for Image Restoration
  8. 2026Learning Adaptive Dynamical Features via Multi-τ Liquid-Mamba for All-in-one Image Restoration