WaMaIR: Image Restoration via Multiscale Wavelet Convolutions and Mamba-based Channel Modeling with Texture Enhancement
On the map: State-space sequence models (S4, Mamba) → Mamba for image restoration
Abstract (arXiv)
Image restoration is a fundamental and challenging task in computer vision, where CNN-based frameworks demonstrate significant computational efficiency. However, previous CNN-based methods often face challenges in adequately restoring fine texture details, which are limited by the small receptive field of CNN structures and the lack of channel feature modeling. In this paper, we propose WaMaIR, which is a novel framework with a large receptive field for image perception and improves the reconstruction of texture details in restored images. Specifically, we introduce the Global Multiscale Wavelet Transform Convolutions (GMWTConvs) for expandding the receptive field to extract image features, preserving and enriching texture features in model inputs. Meanwhile, we propose the Mamba-Based Channel-Aware Module (MCAM), explicitly designed to capture long-range dependencies within feature channels, which enhancing the model sensitivity to color, edges, and texture information. Additionally, we propose Multiscale Texture Enhancement Loss (MTELoss) for image restoration to guide the model in preserving detailed texture structures effectively. Extensive experiments confirm that WaMaIR outperforms state-of-the-art methods, achieving better image restoration and efficient computational performance of the model.
Key facts. No citing papers in the map yet (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 53 references that are in the map. First posted on arXiv 19 Oct 2025.
Builds on
33 of 53 references.
Closest ancestors · 1
Same direction, newest first
Other references · 32
- 2017Attention Is All You Needcited by 59,932
- 2017Squeeze-and-Excitation Networkscited by 4,763
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,618
- 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,352
- 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 864
- 2021Multi-Stage Progressive Image Restorationcited by 813
- 2017Benchmarking Single Image Dehazing and Beyondcited by 445
- 2016Deep Joint Rain Detection and Removal from a Single Imagecited by 351
- 2016DehazeNet: An End-to-End System for Single Image Haze Removalcited by 313
- 2022MAXIM: Multi-Axis MLP for Image Processingcited by 301
- 2019Progressive Image Deraining Networks: A Better and Simpler Baselinecited by 271
20 more
- 2020Multi-Scale Progressive Fusion Network for Single Image Derainingcited by 259
- 2022Vision Transformers for Single Image Dehazingcited by 255
- 2021TransWeather: Transformer-based Restoration of Images Degraded by Adverse Weather Conditionscited by 243
- 2018Recurrent Squeeze-and-Excitation Context Aggregation Net for Single Image Derainingcited by 205
- 2017DesnowNet: Context-Aware Deep Network for Snow Removalcited by 200
- 2017Image De-raining Using a Conditional Generative Adversarial Networkcited by 197
- 2019Uncertainty Guided Multi-Scale Residual Learning-using a Cycle Spinning CNN for Single Image De-Rainingcited by 117
- 2022Deep Image Deblurring: A Surveycited by 94
- 2023DEA-Net: Single image dehazing based on detail-enhanced convolution and content-guided attentioncited by 90
- 2024VmambaIR: Visual State Space Model for Image Restorationcited by 90
- 2023MB-TaylorFormer: Multi-branch Efficient Transformer Expanded by Taylor Formula for Image Dehazingcited by 81
- 2024Wavelet Convolutions for Large Receptive Fieldscited by 74
- 2023GridFormer: Residual Dense Transformer with Grid Structure for Image Restoration in Adverse Weather Conditionscited by 62
- 2020Nighttime Dehazing with a Synthetic Benchmarkcited by 41
- 2023Enhancing Visibility in Nighttime Haze Images Using Guided APSF and Gradient Adaptive Convolutioncited by 39
- 2024Correlation Matching Transformation Transformers for UHD Image Restorationcited by 39
- 2025JarvisIR: Elevating Autonomous Driving Perception with Intelligent Image Restorationcited by 32
- 2024All-In-One Medical Image Restoration via Task-Adaptive Routingcited by 30
- 2023Mutual Information-driven Triple Interaction Network for Efficient Image Dehazingcited by 18
- 2022Restoring Vision in Hazy Weather with Hierarchical Contrastive Learningcited by 17
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Directing Mamba to Complex Textures: An Efficient Texture-Aware State Space Model for Image Restoration
- 2024Multi-Scale Representation Learning for Image Restoration with State-Space Model
- 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Model
- 2025MatIR: A Hybrid Mamba-Transformer Image Restoration Model
- 2025WDMamba: When Wavelet Degradation Prior Meets Vision Mamba for Image Dehazing
- 2026Progressive Split Mamba: Effective State Space Modelling for Image Restoration
- 2024Training-Free Large Model Priors for Multiple-in-One Image Restoration
- 2024Wave-Mamba: Wavelet State Space Model for Ultra-High-Definition Low-Light Image Enhancement