M2IR: Proactive All-in-One Image Restoration via Mamba-style Modulation and Mixture-of-Experts
On the map: Mixture-of-experts → Vision mixture-of-experts
also in 1 more
Abstract (arXiv)
While Transformer-based architectures have dominated recent advances in all-in-one image restoration, they remain fundamentally reactive: propagating degradations rather than proactively suppressing them. In the absence of explicit suppression mechanisms, degraded signals interfere with feature learning, compelling the decoder to balance artifact removal and detail preservation, thereby increasing model complexity and limiting adaptability. To address these challenges, we propose M2IR, a novel restoration framework that proactively regulates degradation propagation during the encoding stage and efficiently eliminates residual degradations during decoding. Specifically, the Mamba-Style Transformer (MST) block performs pixel-wise selective state modulation to mitigate degradations while preserving structural integrity. In parallel, the Adaptive Degradation Expert Collaboration (ADEC) module utilizes degradation-specific experts guided by a DA-CLIP-driven router and complemented by a shared expert to eliminate residual degradations through targeted and cooperative restoration. By integrating the MST block and ADEC module, M2IR transitions from passive reaction to active degradation control, effectively harnessing learned representations to achieve superior generalization, enhanced adaptability, and refined recovery of fine-grained details across diverse all-in-one image restoration benchmarks. Our source codes are available at https://github.com/Im34v/M2IR.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Vision mixture-of-experts, one of 12 directions in Mixture-of-experts. Builds on 40 of its 54 references that are in the map. First posted on arXiv 16 Mar 2026.
Builds on
40 of 54 references.
Closest ancestors · 1
Same direction, newest first
Other references · 39
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 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
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,051
- 2024VMamba: Visual State Space Modelcited by 955
- 2016Deep Multi-scale Convolutional Neural Network for Dynamic Scene Deblurringcited by 722
- 2018Deep Retinex Decomposition for Low-Light Enhancementcited by 720
- 2022Simple Baselines for Image Restorationcited by 701
- 2017FFDNet: Toward a Fast and Flexible Solution for CNN based Image Denoisingcited by 509
- 2017Benchmarking Single Image Dehazing and Beyondcited by 445
- 2016Deep Joint Rain Detection and Removal from a Single Imagecited by 351
27 more
- 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 349
- 2016DehazeNet: An End-to-End System for Single Image Haze Removalcited by 313
- 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
- 2022Restoring Vision in Adverse Weather Conditions with Patch-Based Denoising Diffusion Modelscited by 215
- 2023Learning A Sparse Transformer Network for Effective Image Derainingcited by 204
- 2022Learning Enriched Features for Fast Image Restoration and Enhancementcited by 150
- 2024InstructIR: High-Quality Image Restoration Following Human Instructionscited by 139
- 2022Deep Generalized Unfolding Networks for Image Restorationcited by 119
- 2019Uncertainty Guided Multi-Scale Residual Learning-using a Cycle Spinning CNN for Single Image De-Rainingcited by 117
- 2024AdaIR: Adaptive All-in-One Image Restoration via Frequency Mining and Modulationcited by 108
- 2024Demystify Mamba in Vision: A Linear Attention Perspectivecited by 101
- 2023AutoDIR: Automatic All-in-One Image Restoration with Latent Diffusioncited by 98
- 2024OneRestore: A Universal Restoration Framework for Composite Degradationcited by 91
- 2020FD-GAN: Generative Adversarial Networks with Fusion-discriminator for Single Image Dehazingcited by 79
- 2024Selective Hourglass Mapping for Universal Image Restoration Based on Diffusion Modelcited by 73
- 2024Restoring Images in Adverse Weather Conditions via Histogram Transformercited by 73
- 2018Lightweight Pyramid Networks for Image Derainingcited by 64
- 2023GridFormer: Residual Dense Transformer with Grid Structure for Image Restoration in Adverse Weather Conditionscited by 62
- 2024Perceive-IR: Learning to Perceive Degradation Better for All-in-One Image Restorationcited by 60
- 2024Towards Effective Multiple-in-One Image Restoration: A Sequential and Prompt Learning Strategycited by 48
- 2022Adaptive Dynamic Filtering Network for Image Denoisingcited by 27
- 2025WeatherBench: A Real-World Benchmark Dataset for All-in-One Adverse Weather Image Restorationcited by 20
- 2024Multi-Expert Adaptive Selection: Task-Balancing for All-in-One Image Restorationcited by 18
- 2026Learning Continuous Wasserstein Barycenter Space for Generalized All-in-One Image Restorationcited by 14
- 2025Efficient Degradation-agnostic Image Restoration via Channel-Wise Functional Decomposition and Manifold Regularizationcited by 8
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Leveraging Multimodal Large Language Models for All-in-One Image Restoration via a Mixture of Frequency Experts
- 2025ClusIR: Towards Cluster-Guided All-in-One Image Restoration
- 2024Training-Free Large Model Priors for Multiple-in-One Image Restoration
- 2025MatIR: A Hybrid Mamba-Transformer Image Restoration Model
- 2024Chain-of-Restoration: Multi-Task Image Restoration Models are Zero-Shot Step-by-Step Universal Image Restorers
- 2026Continuous Expert Assembly: Instance-Conditioned Low-Rank Residuals for All-in-One Image Restoration
- 2024MambaIRv2: Attentive State Space Restoration
- 2026Loop-Mamba: A Loop Mamba with Degradation-Aware and Shared Memory for Old Photo Restoration