mlsanityData to 7 Oct 2026

U-shaped Vision Mamba for Single Image Dehazing

arXiv 2402.04139 ↗First posted 6 Feb 2024cs.CVcited by 40

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

Abstract (arXiv)

Currently, Transformer is the most popular architecture for image dehazing, but due to its large computational complexity, its ability to handle long-range dependency is limited on resource-constrained devices. To tackle this challenge, we introduce the U-shaped Vision Mamba (UVM-Net), an efficient single-image dehazing network. Inspired by the State Space Sequence Models (SSMs), a new deep sequence model known for its power to handle long sequences, we design a Bi-SSM block that integrates the local feature extraction ability of the convolutional layer with the ability of the SSM to capture long-range dependencies. Extensive experimental results demonstrate the effectiveness of our method. Our method provides a more highly efficient idea of long-range dependency modeling for image dehazing as well as other image restoration tasks. The URL of the code is urlhttps://github.com/zzr-idam/UVM-Net. Our method takes only 0.009 seconds to infer a 325 × 325 resolution image (100FPS) without I/O handling time.

Key facts. Cited by 40 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 33 of its 54 references that are in the map. First posted on arXiv 6 Feb 2024.

Builds on

33 of 54 references.

References · 33

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  4. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
  5. 2016Layer Normalizationcited by 5,579
  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. 2021Multi-Stage Progressive Image Restorationcited by 813
  9. 2018Deep Retinex Decomposition for Low-Light Enhancementcited by 720
  10. 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 607
  11. 2024U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentationcited by 412
  12. 2016Deep Joint Rain Detection and Removal from a Single Imagecited by 351
21 more
  1. 2016DehazeNet: An End-to-End System for Single Image Haze Removalcited by 313
  2. 2019FFA-Net: Feature Fusion Attention Network for Single Image Dehazingcited by 307
  3. 2019Progressive Image Deraining Networks: A Better and Simpler Baselinecited by 271
  4. 2020Multi-Scale Progressive Fusion Network for Single Image Derainingcited by 259
  5. 2022Vision Transformers for Single Image Dehazingcited by 255
  6. 2018Density-aware Single Image De-raining using a Multi-stream Dense Networkcited by 248
  7. 2021HINet: Half Instance Normalization Network for Image Restorationcited by 236
  8. 2020Multi-Scale Boosted Dehazing Network with Dense Feature Fusioncited by 229
  9. 2019GridDehazeNet: Attention-Based Multi-Scale Network for Image Dehazingcited by 227
  10. 2021Contrastive Learning for Compact Single Image Dehazingcited by 219
  11. 2018Recurrent Squeeze-and-Excitation Context Aggregation Net for Single Image Derainingcited by 205
  12. 2022It's Raw! Audio Generation with State-Space Modelscited by 171
  13. 2018Densely Connected Pyramid Dehazing Networkcited by 145
  14. 2016Clearing the Skies: A deep network architecture for single-image rain removalcited by 136
  15. 2018Gated Fusion Network for Single Image Dehazingcited by 133
  16. 2022Deep Generalized Unfolding Networks for Image Restorationcited by 119
  17. 2019Uncertainty Guided Multi-Scale Residual Learning-using a Cycle Spinning CNN for Single Image De-Rainingcited by 117
  18. 2020Domain Adaptation for Image Dehazingcited by 108
  19. 2022Long Movie Clip Classification with State-Space Video Modelscited by 106
  20. 2018Gated Context Aggregation Network for Image Dehazing and Derainingcited by 89
  21. 2019Single Image Deraining: A Comprehensive Benchmark Analysiscited by 84

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 22; 2025: 11; 2026: 7.

40 papers.

Sort:

By area:

  1. 2024VmambaIR: Visual State Space Model for Image Restorationcited by 90
  2. 2024MambaIRv2: Attentive State Space Restorationcited by 90
  3. 2024ZigMa: A DiT-style Zigzag Mamba Diffusion Modelcited by 84
  4. 2024Retinexmamba: Retinex-based Mamba for Low-light Image Enhancementcited by 59
  5. 2024RadioDiff: An Effective Generative Diffusion Model for Sampling-Free Dynamic Radio Map Constructioncited by 55
  6. 2024Vision Mamba: A Comprehensive Survey and Taxonomycited by 53
  7. 2024A Survey on Visual Mambacited by 50
  8. 2024MaIR: A Locality- and Continuity-Preserving Mamba for Image Restorationcited by 43
  9. 2024ReMamber: Referring Image Segmentation with Mamba Twistercited by 38
  10. 2024Matten: Video Generation with Mamba-Attentioncited by 24
  11. 2025RadioDiff-k2: Helmholtz Equation Informed Generative Diffusion Model for Multi-Path Aware Radio Map Constructioncited by 19
  12. 2024Mamba in Vision: A Comprehensive Survey of Techniques and Applicationscited by 18
28 more
  1. 2024LLEMamba: Low-Light Enhancement via Relighting-Guided Mamba with Deep Unfolding Networkcited by 14
  2. 2024MambaCSR: Dual-Interleaved Scanning for Compressed Image Super-Resolution With SSMscited by 14
  3. 2025Integrating Extra Modality Helps Segmentor Find Camouflaged Objects Wellcited by 14
  4. 2025Generative Diffusion Models for Wireless Networks: Fundamental, Architecture, and State-of-the-Artcited by 13
  5. 2025NTIRE 2025 Challenge on Image Super-Resolution (x4): Methods and Resultscited by 12
  6. 2024Speech-Mamba: Long-Context Speech Recognition with Selective State Spaces Modelscited by 9
  7. 2024Shuffle Mamba: State Space Models with Random Shuffle for Multi-Modal Image Fusioncited by 7
  8. 2024Retinex-RAWMamba: Bridging Demosaicing and Denoising for Low-Light RAW Image Enhancementcited by 6
  9. 2024HDMba: Hyperspectral Remote Sensing Imagery Dehazing with State Space Modelcited by 5
  10. 2025Content-Aware Mamba for Learned Image Compressioncited by 5
  11. 2024SkinMamba: A Precision Skin Lesion Segmentation Architecture with Cross-Scale Global State Modeling and Frequency Boundary Guidancecited by 4
  12. 2026RadioDiff-Flux: Efficient Radio Map Construction via Generative Denoise Diffusion Model Trajectory Midpoint Reusecited by 4
  13. 2024CLIP-Mamba: CLIP Pretrained Mamba Models with OOD and Hessian Evaluationcited by 3
  14. 2025Mesh Mamba: A Unified State Space Model for Saliency Prediction in Non-Textured and Textured Meshescited by 2
  15. 2024ST-Mamba: Spatial-Temporal Selective State Space Model for Traffic Flow Predictioncited by 1
  16. 2024Venturing into Uncharted Waters: The Navigation Compass from Transformer to Mambacited by 1
  17. 2025U-Net-Like Spiking Neural Networks for Single Image Dehazingcited by 1
  18. 2026Pixel Ignores, Superpixel Sees: Adverse Weather Image Restoration via Semantic-Center SSMcited by 1
  19. 2026HNDiff: Haze-Noise Diffusion for Image Dehazingcited by 1
  20. 2024Towards Optimal Aggregation of Varying Range Dependencies in Haze Removalcited by 0
  21. 2025DehazeMamba: SAR-guided Optical Remote Sensing Image Dehazing with Adaptive State Space Modelcited by 0
  22. 2025LMDepth: Lightweight Mamba-based Monocular Depth Estimation for Real-World Deploymentcited by 0
  23. 2025ControlMambaIR: Conditional Controls with State-Space Model for Image Restorationcited by 0
  24. 2025First-order State Space Model for Lightweight Image Super-resolutioncited by 0
  25. 2026Learning Human Visual Attention on 3D Surfaces through Geometry-Queried Semantic Priorscited by 0
  26. 2026U2Mamba: A Two-level Nested U-structure Mamba for Salient Object Detectioncited by 0
  27. 2026Efficient Real-World Dehazing via Physics-Inspired Global-Local Decouplingcited by 0
  28. 2026HazeSpikeMamba: Coupling Spiking-Inspired and State-Space Features for Self-Supervised Real-World Dehazingcited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2024XYScanNet: A State Space Model for Single Image Deblurring
  2. 2024CU-Mamba: Selective State Space Models with Channel Learning for Image Restoration
  3. 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Model
  4. 2025Laplace-Mamba: Laplace Frequency Prior-Guided Mamba-CNN Fusion Network for Image Dehazing
  5. 2025EAMamba: Efficient All-Around Vision State Space Model for Image Restoration
  6. 2024Visual Mamba: A Survey and New Outlooks
  7. 2024RSDehamba: Lightweight Vision Mamba for Remote Sensing Satellite Image Dehazing
  8. 2025U-Shape Mamba: State Space Model for faster diffusion