mlsanityData to 14 Oct 2026

M2Restore: Mixture-of-Experts-based Mamba-CNN Fusion Framework for All-in-One Image Restoration

arXiv 2506.07814 ↗First posted 9 Jun 2025cs.CVcited by 7

On the map: Mixture-of-experts → Vision mixture-of-experts

also in 1 more
Abstract (arXiv)

Natural images are often degraded by complex, composite degradations such as rain, snow, and haze, which adversely impact downstream vision applications. While existing image restoration efforts have achieved notable success, they are still hindered by two critical challenges: limited generalization across dynamically varying degradation scenarios and a suboptimal balance between preserving local details and modeling global dependencies. To overcome these challenges, we propose M2Restore, a novel Mixture-of-Experts (MoE)-based Mamba-CNN fusion framework for efficient and robust all-in-one image restoration. M2Restore introduces three key contributions: First, to boost the model's generalization across diverse degradation conditions, we exploit a CLIP-guided MoE gating mechanism that fuses task-conditioned prompts with CLIP-derived semantic priors. This mechanism is further refined via cross-modal feature calibration, which enables precise expert selection for various degradation types. Second, to jointly capture global contextual dependencies and fine-grained local details, we design a dual-stream architecture that integrates the localized representational strength of CNNs with the long-range modeling efficiency of Mamba. This integration enables collaborative optimization of global semantic relationships and local structural fidelity, preserving global coherence while enhancing detail restoration. Third, we introduce an edge-aware dynamic gating mechanism that adaptively balances global modeling and local enhancement by reallocating computational attention to degradation-sensitive regions. This targeted focus leads to more efficient and precise restoration. Extensive experiments across multiple image restoration benchmarks validate the superiority of M2Restore in both visual quality and quantitative performance.

Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Vision mixture-of-experts, one of 12 directions in Mixture-of-experts. Builds on 50 of its 77 references that are in the map. First posted on arXiv 9 Jun 2025.

Builds on

50 of 77 references.

Closest ancestors · 5

Same direction, newest first

  1. 2024UniRestorer: Universal Image Restoration via Adaptively Estimating Image Degradation at Proper Granularity28 Dec
  2. 2024Complexity Experts are Task-Discriminative Learners for Any Image Restoration27 Nov
  3. 2024LoRA-IR: Taming Low-Rank Experts for Efficient All-in-One Image Restoration20 Oct
  4. 2024MoE-DiffIR: Task-customized Diffusion Priors for Universal Compressed Image Restoration15 Jul
  5. 2020Restoring Spatially-Heterogeneous Distortions using Mixture of Experts Network30 Sep

Other references · 45

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  4. 2021SwinIR: Image Restoration Using Swin Transformercited by 1,887
  5. 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,627
  6. 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,358
  7. 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 866
  8. 2021Multi-Stage Progressive Image Restorationcited by 816
  9. 2022Simple Baselines for Image Restorationcited by 708
  10. 2024U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentationcited by 419
  11. 2019Dynamic Convolution: Attention over Convolution Kernelscited by 356
  12. 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 352
33 more
  1. 2021TransWeather: Transformer-based Restoration of Images Degraded by Adverse Weather Conditionscited by 244
  2. 2022Restoring Vision in Adverse Weather Conditions with Patch-Based Denoising Diffusion Modelscited by 215
  3. 2017Attentive Generative Adversarial Network for Raindrop Removal from a Single Imagecited by 204
  4. 2017DesnowNet: Context-Aware Deep Network for Snow Removalcited by 200
  5. 2019Heavy Rain Image Restoration: Integrating Physics Model and Conditional Adversarial Learningcited by 157
  6. 2023Efficient and Explicit Modelling of Image Hierarchies for Image Restorationcited by 145
  7. 2024AdaIR: Adaptive All-in-One Image Restoration via Frequency Mining and Modulationcited by 111
  8. 2023Controlling Vision-Language Models for Multi-Task Image Restorationcited by 110
  9. 2019Dense Haze: A benchmark for image dehazing with dense-haze and haze-free imagescited by 107
  10. 2023AutoDIR: Automatic All-in-One Image Restoration with Latent Diffusioncited by 100
  11. 2024MambaIRv2: Attentive State Space Restorationcited by 94
  12. 2024OneRestore: A Universal Restoration Framework for Composite Degradationcited by 92
  13. 2021Deep Dense Multi-scale Network for Snow Removal Using Semantic and Geometric Priorscited by 82
  14. 2024Selective Hourglass Mapping for Universal Image Restoration Based on Diffusion Modelcited by 73
  15. 2023GridFormer: Residual Dense Transformer with Grid Structure for Image Restoration in Adverse Weather Conditionscited by 62
  16. 2024Bidirectional Multi-Scale Implicit Neural Representations for Image Derainingcited by 54
  17. 2024MaIR: A Locality- and Continuity-Preserving Mamba for Image Restorationcited by 45
  18. 2019Path-Restore: Learning Network Path Selection for Image Restorationcited by 40
  19. 2025MB-TaylorFormer V2: Improved Multi-branch Linear Transformer Expanded by Taylor Formula for Image Restorationcited by 36
  20. 2022UCL-Dehaze: Towards Real-world Image Dehazing via Unsupervised Contrastive Learningcited by 35
  21. 2024PALP: Prompt Aligned Personalization of Text-to-Image Modelscited by 35
  22. 2021Enhanced Spatio-Temporal Interaction Learning for Video Deraining: A Faster and Better Frameworkcited by 30
  23. 2024CU-Mamba: Selective State Space Models with Channel Learning for Image Restorationcited by 24
  24. 2024MWFormer: Multi-Weather Image Restoration Using Degradation-Aware Transformerscited by 24
  25. 2025UniUIR: Considering Underwater Image Restoration as An All-in-One Learnercited by 22
  26. 2024Multi-Expert Adaptive Selection: Task-Balancing for All-in-One Image Restorationcited by 18
  27. 2020Lightweight image super-resolution with enhanced CNNcited by 15
  28. 2023Exploring the Application of Large-scale Pre-trained Models on Adverse Weather Removalcited by 12
  29. 2025Efficient Degradation-agnostic Image Restoration via Channel-Wise Functional Decomposition and Manifold Regularizationcited by 8
  30. 2025UniCoRN: Latent Diffusion-based Unified Controllable Image Restoration Network across Multiple Degradationscited by 6
  31. 2024Multi-dimensional Visual Prompt Enhanced Image Restoration via Mamba-Transformer Aggregationcited by 2
  32. 2025MatIR: A Hybrid Mamba-Transformer Image Restoration Modelcited by 2
  33. 2024Gradient-Guided Parameter Mask for Multi-Scenario Image Restoration Under Adverse Weathercited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 2; 2026: 5.

7 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Continuous Expert Assembly: Instance-Conditioned Low-Rank Residuals for All-in-One Image Restoration
  2. 2025An Efficient and Mixed Heterogeneous Model for Image Restoration
  3. 2026UnSCAR: Universal, Scalable, Controllable, and Adaptable Image Restoration
  4. 2026SLER-IR: Spherical Layer-wise Expert Routing for All-in-One Image Restoration
  5. 2026Leveraging Multimodal Large Language Models for All-in-One Image Restoration via a Mixture of Frequency Experts
  6. 2024Training-Free Large Model Priors for Multiple-in-One Image Restoration
  7. 2025Towards Unified Image Deblurring using a Mixture-of-Experts Decoder
  8. 2026Loop-Mamba: A Loop Mamba with Degradation-Aware and Shared Memory for Old Photo Restoration