mlsanityData to 14 Oct 2026

MambaIRv2: Attentive State Space Restoration

arXiv 2411.15269 ↗First posted 22 Nov 2024eess.IV cs.CV cs.LGcited by 94

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

Abstract (arXiv)

The Mamba-based image restoration backbones have recently demonstrated significant potential in balancing global reception and computational efficiency. However, the inherent causal modeling limitation of Mamba, where each token depends solely on its predecessors in the scanned sequence, restricts the full utilization of pixels across the image and thus presents new challenges in image restoration. In this work, we propose MambaIRv2, which equips Mamba with the non-causal modeling ability similar to ViTs to reach the attentive state space restoration model. Specifically, the proposed attentive state-space equation allows to attend beyond the scanned sequence and facilitate image unfolding with just one single scan. Moreover, we further introduce a semantic-guided neighboring mechanism to encourage interaction between distant but similar pixels. Extensive experiments show our MambaIRv2 outperforms SRFormer by even 0.35dB PSNR for lightweight SR even with 9.3% less parameters and suppresses HAT on classic SR by up to 0.29dB. Code is available at https://github.com/csguoh/MambaIR.

Key facts. Cited by 94 papers in the map (data through 14 Oct 2026). Placed in Mamba for image restoration, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 52 of its 74 references that are in the map. First posted on arXiv 22 Nov 2024.

Builds on

52 of 74 references.

Closest ancestors · 5

Same direction, newest first

  1. 2024Retinexmamba: Retinex-based Mamba for Low-light Image Enhancement6 May
  2. 2024FreqMamba: Viewing Mamba from a Frequency Perspective for Image Deraining15 Apr
  3. 2024Learning Enriched Features via Selective State Spaces Model for Efficient Image Deblurring29 Mar
  4. 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Model23 Feb
  5. 2024U-shaped Vision Mamba for Single Image Dehazing6 Feb

Other references · 47

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
  4. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  5. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
  6. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  7. 2021SwinIR: Image Restoration Using Swin Transformercited by 1,887
  8. 2022Visual Prompt Tuningcited by 1,681
  9. 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,627
  10. 2017Enhanced Deep Residual Networks for Single Image Super-Resolutioncited by 1,552
  11. 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,358
  12. 2018Image Super-Resolution Using Very Deep Residual Channel Attention Networkscited by 1,264
35 more
  1. 2015Accurate Image Super-Resolution Using Very Deep Convolutional Networkscited by 903
  2. 2020Pre-Trained Image Processing Transformercited by 893
  3. 2018Residual Dense Network for Image Super-Resolutioncited by 799
  4. 2017Understanding the Effective Receptive Field in Deep Convolutional Neural Networkscited by 553
  5. 2020Plug-and-Play Image Restoration with Deep Denoiser Priorcited by 470
  6. 2022Activating More Pixels in Image Super-Resolution Transformercited by 438
  7. 2017Learning Deep CNN Denoiser Prior for Image Restorationcited by 408
  8. 2021Conditional Positional Encodings for Vision Transformerscited by 406
  9. 2015Sketch-based Manga Retrieval using Manga109 Datasetcited by 329
  10. 2018Fast, Accurate, and Lightweight Super-Resolution with Cascading Residual Networkcited by 264
  11. 2019Residual Non-local Attention Networks for Image Restorationcited by 236
  12. 2018Residual Dense Network for Image Restorationcited by 181
  13. 2022Efficient Long-Range Attention Network for Image Super-resolutioncited by 151
  14. 2023Efficient and Explicit Modelling of Image Hierarchies for Image Restorationcited by 145
  15. 2015Compression Artifacts Reduction by a Deep Convolutional Networkcited by 136
  16. 2020Interpreting Super-Resolution Networks with Local Attribution Mapscited by 136
  17. 2023Dual Aggregation Transformer for Image Super-Resolutioncited by 124
  18. 2024Demystify Mamba in Vision: A Linear Attention Perspectivecited by 102
  19. 2022Cross Aggregation Transformer for Image Restorationcited by 92
  20. 2024VmambaIR: Visual State Space Model for Image Restorationcited by 90
  21. 2022Accurate Image Restoration with Attention Retractable Transformercited by 85
  22. 2024FusionMamba: Dynamic Feature Enhancement for Multimodal Image Fusion with Mambacited by 81
  23. 2024Diffusion Models in Low-Level Vision: A Surveycited by 74
  24. 2021On Efficient Transformer-Based Image Pre-training for Low-Level Visioncited by 64
  25. 2024Transcending the Limit of Local Window: Advanced Super-Resolution Transformer with Adaptive Token Dictionarycited by 64
  26. 2023Omni Aggregation Networks for Lightweight Image Super-Resolutioncited by 62
  27. 2024A Survey on All-in-One Image Restoration: Taxonomy, Evaluation and Future Trendscited by 60
  28. 2023A Comparative Study of Image Restoration Networks for General Backbone Network Designcited by 57
  29. 2024RainMamba: Enhanced Locality Learning with State Space Models for Video Derainingcited by 56
  30. 2024MambaLLIE: Implicit Retinex-Aware Low Light Enhancement with Global-then-Local State Spacecited by 48
  31. 2023Xformer: Hybrid X-Shaped Transformer for Image Denoisingcited by 37
  32. 2023Recursive Generalization Transformer for Image Super-Resolutioncited by 29
  33. 2024MambaVC: Learned Visual Compression with Selective State Spacescited by 28
  34. 2024Hi-Mamba: Hierarchical Mamba for Efficient Image Super-Resolutioncited by 16
  35. 2024PixMamba: Leveraging State Space Models in a Dual-Level Architecture for Underwater Image Enhancementcited by 6

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2024: 2; 2025: 46; 2026: 46.

94 papers.

Sort:

By area:

  1. 2025The Tenth NTIRE 2025 Efficient Super-Resolution Challenge Reportcited by 26
  2. 2026The Eleventh NTIRE 2026 Efficient Super-Resolution Challenge Reportcited by 25
  3. 2025The Tenth NTIRE 2025 Image Denoising Challenge Reportcited by 21
  4. 2025NTIRE 2025 Challenge on Short-form UGC Video Quality Assessment and Enhancement: Methods and Resultscited by 14
  5. 2025NTIRE 2025 Challenge on Image Super-Resolution (x4): Methods and Resultscited by 12
  6. 2025IDF: Iterative Dynamic Filtering Networks for Generalizable Image Denoisingcited by 12
  7. 2025NTIRE 2025 Challenge on Short-form UGC Video Quality Assessment and Enhancement: KwaiSR Dataset and Studycited by 11
  8. 2026The Fourth Challenge on Image Super-Resolution (×4) at NTIRE 2026: Benchmark Results and Method Overviewcited by 10
  9. 2025PMA: Towards Parameter-Efficient Point Cloud Understanding via Point Mamba Adaptercited by 9
  10. 2025AceVFI: A Comprehensive Survey of Advances in Video Frame Interpolationcited by 9
  11. 2025NTIRE 2025 Challenge on RAW Image Restoration and Super-Resolutioncited by 9
  12. 2026Scan Clusters, Not Pixels: A Cluster-Centric Paradigm for Efficient Ultra-high-definition Image Restorationcited by 8
38 more
  1. 2025DSwinIR: Rethinking Window-based Attention for Image Restorationcited by 7
  2. 2025M2Restore: Mixture-of-Experts-based Mamba-CNN Fusion Framework for All-in-One Image Restorationcited by 7
  3. 2026Dual-Branch Remote Sensing Infrared Image Super-Resolutioncited by 7
  4. 2025Spiking Meets Attention: Efficient Remote Sensing Image Super-Resolution with Attention Spiking Neural Networkscited by 6
  5. 2025C2D-ISR: Optimizing Attention-based Image Super-resolution from Continuous to Discrete Scalescited by 6
  6. 2025Vision Mamba in Remote Sensing: A Comprehensive Survey of Techniques, Applications and Outlookcited by 6
  7. 2025Content-Aware Mamba for Learned Image Compressioncited by 6
  8. 2026Training-Free Model Ensemble for Single-Image Super-Resolution via Strong-Branch Compensationcited by 6
  9. 2025Emulating Self-attention with Convolution for Efficient Image Super-Resolutioncited by 5
  10. 2025Q-MambaIR: Accurate Quantized Mamba for Efficient Image Restorationcited by 5
  11. 2026Depth-Synergized Mamba Meets Memory Experts for All-Day Image Reflection Separationcited by 5
  12. 2026Beyond Model Design: Data-Centric Training and Self-Ensemble for Gaussian Color Image Denoisingcited by 5
  13. 2024Unsupervised Multi-Parameter Inverse Solving for Reducing Ring Artifacts in 3D X-Ray CBCTcited by 4
  14. 2026ICPR 2026 Competition on Low-Resolution License Plate Recognitioncited by 4
  15. 2025Mamba-MOC: A Multicategory Remote Object Counting via State Space Modelcited by 3
  16. 2025DH-Mamba: Exploring Dual-domain Hierarchical State Space Models for MRI Reconstructioncited by 3
  17. 2025FourierSR: A Fourier Token-based Plugin for Efficient Image Super-Resolutioncited by 3
  18. 2025DPMambaIR: All-in-One Image Restoration via Degradation-Aware Prompt State Space Modelcited by 3
  19. 2025RetinexDual: Retinex-based Dual Nature Approach for Generalized Ultra-High-Definition Image Restorationcited by 3
  20. 2025SAT-SKYLINES: 3D Building Generation from Satellite Imagery and Coarse Geometric Priorscited by 3
  21. 2026ATD: Improved Transformer with Adaptive Token Dictionary for Image Restorationcited by 3
  22. 2024LossAgent: Towards Any Optimization Objectives for Image Processing with LLM Agentscited by 2
  23. 2025Mamba-FETrack V2: Revisiting State Space Model for Frame-Event based Visual Object Trackingcited by 2
  24. 2025Lightweight Quad Bayer HybridEVS Demosaicing via State Space Augmented Cross-Attentioncited by 2
  25. 2025MBMamba: When Memory Buffer Meets Mamba for Structure-Aware Image Deblurringcited by 2
  26. 2025RDDM: Practicing RAW Domain Diffusion Model for Real-world Image Restorationcited by 2
  27. 2025Beyond the Ground Truth: Enhanced Supervision for Image Restorationcited by 2
  28. 2026Learning Adaptive Dynamical Features via Multi-τ Liquid-Mamba for All-in-one Image Restorationcited by 2
  29. 2025A Preliminary Study on GPT-Image Generation Model for Image Restorationcited by 1
  30. 2025Instance Data Condensation for Image Super-Resolutioncited by 1
  31. 2025FADPNet: Frequency-Aware Dual-Path Network for Face Super-Resolutioncited by 1
  32. 2025Polyline Path Masked Attention for Vision Transformercited by 1
  33. 2025BSMamba: Brightness and Semantic Modeling for Long-Range Interaction in Low-Light Image Enhancementcited by 1
  34. 2025Trajectory-aware Shifted State Space Models for Online Video Super-Resolutioncited by 1
  35. 2025PRISM: Progressive Rain removal with Integrated State-space Modelingcited by 1
  36. 2025Transformer-Progressive Mamba Network for Lightweight Image Super-Resolutioncited by 1
  37. 2026From Local Windows to Adaptive Candidates via Individualized Exploratory: Rethinking Attention for Image Super-Resolutioncited by 1
  38. 2026Rank-Factorized Implicit Neural Bias: Scaling Super-Resolution Transformer with FlashAttentioncited by 1

50 most cited of 94 All 94 (JSON)

Similar, not cited

Close in wording, no citation either way

  1. 2025EAMamba: Efficient All-Around Vision State Space Model for Image Restoration
  2. 2024MaIR: A Locality- and Continuity-Preserving Mamba for Image Restoration
  3. 2024CU-Mamba: Selective State Space Models with Channel Learning for Image Restoration
  4. 2024MambaCSR: Dual-Interleaved Scanning for Compressed Image Super-Resolution With SSMs
  5. 2025Directing Mamba to Complex Textures: An Efficient Texture-Aware State Space Model for Image Restoration
  6. 2026Loop-Mamba: A Loop Mamba with Degradation-Aware and Shared Memory for Old Photo Restoration
  7. 2026SF-Mamba: Rethinking State Space Model for Vision
  8. 2024Multi-Scale Representation Learning for Image Restoration with State-Space Model