State-Isolated Vision Mamba via Update-Tendency Routing for Underwater Image Enhancement
On the map: State-space sequence models (S4, Mamba) → Vision Mamba backbones also fits: Mamba for image restoration
Abstract (arXiv)
Vision Mamba has recently emerged as an efficient alternative to convolutional and Transformer-based architectures by modeling long-range spatial dependencies through recurrent state evolution. Existing studies primarily improve token serialization, scanning strategies, or expert routing, yet largely overlook a state level question question: how should recurrent states be organized for spatially heterogeneous visual restoration? In this paper, we show that forcing tokens with heterogeneous update behaviors to recurrently modify a shared hidden state leads to persistent cross-content interference, which we term State Contention. We further argue that effective recurrent modeling should organize state evolution according to token compatibility rather than token ordering. Based on this observation, we introduce State Isolation, a new recurrent state organization principle for Vision Mamba,tokens with compatible state-update tendencies evolve within independent recurrent state streams. To instantiate this principle, we propose IsoMamba, a Delta-guided architecture that constructs model-intrinsic update representations, discovers update-compatible token groups through label-free clustering, and routes them to isolated Mamba experts following a cluster--route--isolate paradigm. In addition, we develop a dynamic channel-wise reweighting objective to compensate for wavelength-dependent reconstruction difficulty in underwater images. Extensive experiments demonstrate that IsoMamba consistently achieves state-of-the-art performance on multiple underwater image enhancement benchmarks while providing cleaner hidden state responses, validating the effectiveness of state isolated recurrent modeling.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Vision Mamba backbones, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 51 of its 72 references that are in the map. First posted on arXiv 9 Oct 2026.
Builds on
51 of 72 references.
Closest ancestors · 3
Same direction, newest first
- 2024Spatial-Mamba: Effective Visual State Space Models via Structure-aware State Fusion19 Oct
- 2024LocalMamba: Visual State Space Model with Windowed Selective Scan14 Mar
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model17 Jan
Other references · 48
- 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,627
- 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,358
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2022Exploring CLIP for Assessing the Look and Feel of Imagescited by 991
- 2024VMamba: Visual State Space Modelcited by 969
- 2021Multi-Stage Progressive Image Restorationcited by 816
- 2016Deep Multi-scale Convolutional Neural Network for Dynamic Scene Deblurringcited by 727
- 2018Deep Retinex Decomposition for Low-Light Enhancementcited by 723
- 2022Simple Baselines for Image Restorationcited by 708
- 2017Benchmarking Single Image Dehazing and Beyondcited by 447
- 2017DeblurGAN: Blind Motion Deblurring Using Conditional Adversarial Networkscited by 367
- 2016Deep Joint Rain Detection and Removal from a Single Imagecited by 352
36 more
- 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 352
- 2016DehazeNet: An End-to-End System for Single Image Haze Removalcited by 314
- 2020Multi-Scale Progressive Fusion Network for Single Image Derainingcited by 260
- 2022Vision Transformers for Single Image Dehazingcited by 260
- 2023DiffIR: Efficient Diffusion Model for Image Restorationcited by 255
- 2019An Underwater Image Enhancement Benchmark Dataset and Beyondcited by 253
- 2023Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancementcited by 231
- 2023Learning A Sparse Transformer Network for Effective Image Derainingcited by 205
- 2019Fast Underwater Image Enhancement for Improved Visual Perceptioncited by 183
- 2022Deep Generalized Unfolding Networks for Image Restorationcited by 121
- 2021U-shape Transformer for Underwater Image Enhancementcited by 112
- 2024MambaIRv2: Attentive State Space Restorationcited by 94
- 2018Underwater Single Image Color Restoration Using Haze-Lines and a New Quantitative Datasetcited by 93
- 2023Hierarchical Integration Diffusion Model for Realistic Image Deblurringcited by 88
- 2023Wavelet-based Fourier Information Interaction with Frequency Diffusion Adjustment for Underwater Image Restorationcited by 88
- 2020FD-GAN: Generative Adversarial Networks with Fusion-discriminator for Single Image Dehazingcited by 80
- 2023Low-Light Image Enhancement via Structure Modeling and Guidancecited by 79
- 2023Contrastive Semi-supervised Learning for Underwater Image Restoration via Reliable Bankcited by 71
- 2022Underwater Ranker: Learn Which Is Better and How to Be Bettercited by 67
- 2024Perceive-IR: Learning to Perceive Degradation Better for All-in-One Image Restorationcited by 62
- 2022Uncertainty Inspired Underwater Image Enhancementcited by 54
- 2024MaIR: A Locality- and Continuity-Preserving Mamba for Image Restorationcited by 45
- 2021UIEC^2-Net: CNN-based Underwater Image Enhancement Using Two Color Spacecited by 35
- 2022SGUIE-Net: Semantic Attention Guided Underwater Image Enhancement with Multi-Scale Perceptioncited by 33
- 2023Underwater Image Enhancement by Transformer-based Diffusion Model with Non-uniform Sampling for Skip Strategycited by 33
- 2024FourierMamba: Fourier Learning Integration with State Space Models for Image Derainingcited by 33
- 2024Efficient Visual State Space Model for Image Deblurringcited by 29
- 2022Adaptive Dynamic Filtering Network for Image Denoisingcited by 28
- 2025UniUIR: Considering Underwater Image Restoration as An All-in-One Learnercited by 22
- 2023A Generalized Physical-knowledge-guided Dynamic Model for Underwater Image Enhancementcited by 16
- 2024WaterMamba: Visual State Space Model for Underwater Image Enhancementcited by 7
- 2025M2Restore: Mixture-of-Experts-based Mamba-CNN Fusion Framework for All-in-One Image Restorationcited by 7
- 2025Adaptive Dual-domain Learning for Underwater Image Enhancementcited by 6
- 2025Devil is in the Uniformity: Exploring Diverse Learners within Transformer for Image Restorationcited by 5
- 2024Efficient Concertormer for Image Deblurring and Beyondcited by 2
- 2025Fourier-RWKV: A Multi-State Perception Network for Efficient Image Dehazingcited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025VRS-UIE: Value-Driven Reordering Scanning for Underwater Image Enhancement
- 2025Fast Vision Mamba: Pooling Spatial Dimensions for Accelerated Processing
- 2026SF-Mamba: Rethinking State Space Model for Vision
- 2025Arcee: Differentiable Recurrent State Chain for Generative Vision Modeling with Mamba SSMs
- 2025UIS-Mamba: Exploring Mamba for Underwater Instance Segmentation via Dynamic Tree Scan and Hidden State Weaken
- 2024O-Mamba: O-shape State-Space Model for Underwater Image Enhancement
- 2026Progressive Split Mamba: Effective State Space Modelling for Image Restoration
- 2024PixMamba: Leveraging State Space Models in a Dual-Level Architecture for Underwater Image Enhancement