mlsanityData to 7 Oct 2026

Continuous Expert Assembly: Instance-Conditioned Low-Rank Residuals for All-in-One Image Restoration

arXiv 2605.06127 ↗First posted 7 May 2026cs.CV cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Real-world image degradation is often unknown, spatially non-uniform, and compositional, requiring all-in-one restoration models to adapt a single set of weights to diverse local corruption patterns without test-time degradation labels. Existing methods typically modulate a shared backbone with global prompts or degradation descriptors, or route features through predefined expert pools. However, compact global conditioning can bottleneck localized degradation evidence, while static expert routing may produce homogeneous updates or rely on unstable sparse assignments. We propose Continuous Expert Assembly (CEA), a token-wise dynamic parameterization framework for all-in-one image restoration. CEA employs a lightweight Cross-Attention Hyper-Adapter to probe intermediate spatial features and synthesize instance-conditioned low-rank routing bases and residual directions. Each spatial token then assembles its own residual update via dense signed dot-product affinities over the generated rank-wise components, avoiding external prompts, static expert banks, and discrete Top- selection. The resulting assembly rule also admits a linear-attention perspective, making its dense token-wise routing behavior transparent. Experiments on AIO-3, AIO-5, and CDD-11 show that CEA improves average restoration quality over strong prompt-, descriptor-, and expert-based baselines, with the clearest gains on spatially varying and compositional degradations, while maintaining favorable parameter, FLOP, and runtime efficiency.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 49 of its 56 references that are in the map. First posted on arXiv 7 May 2026.

Builds on

49 of 56 references.

References · 49

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
  3. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  4. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  5. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
  6. 2021SwinIR: Image Restoration Using Swin Transformercited by 1,869
  7. 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,618
  8. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
  9. 2020Rethinking Attention with Performerscited by 1,584
  10. 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,352
  11. 2016HyperNetworkscited by 1,003
  12. 2020Pre-Trained Image Processing Transformercited by 891
37 more
  1. 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 864
  2. 2021Multi-Stage Progressive Image Restorationcited by 813
  3. 2021Scaling Vision with Sparse Mixture of Expertscited by 740
  4. 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 740
  5. 2022Simple Baselines for Image Restorationcited by 701
  6. 2018Efficient Attention: Attention with Linear Complexitiescited by 366
  7. 2019Dynamic Convolution: Attention over Convolution Kernelscited by 355
  8. 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 349
  9. 2016Dynamic Filter Networkscited by 318
  10. 2022MAXIM: Multi-Axis MLP for Image Processingcited by 301
  11. 2019CondConv: Conditionally Parameterized Convolutions for Efficient Inferencecited by 273
  12. 2023From Sparse to Soft Mixtures of Expertscited by 245
  13. 2021TransWeather: Transformer-based Restoration of Images Degraded by Adverse Weather Conditionscited by 243
  14. 2021HINet: Half Instance Normalization Network for Image Restorationcited by 236
  15. 2022Restoring Vision in Adverse Weather Conditions with Patch-Based Denoising Diffusion Modelscited by 215
  16. 2020Focal Frequency Loss for Image Reconstruction and Synthesiscited by 204
  17. 2017Burst Denoising with Kernel Prediction Networkscited by 156
  18. 2024InstructIR: High-Quality Image Restoration Following Human Instructionscited by 139
  19. 2022Deep Generalized Unfolding Networks for Image Restorationcited by 119
  20. 2023PromptIR: Prompting for All-in-One Blind Image Restorationcited by 113
  21. 2023Controlling Vision-Language Models for Multi-Task Image Restorationcited by 108
  22. 2022Cross Aggregation Transformer for Image Restorationcited by 92
  23. 2024OneRestore: A Universal Restoration Framework for Composite Degradationcited by 91
  24. 2020FD-GAN: Generative Adversarial Networks with Fusion-discriminator for Single Image Dehazingcited by 79
  25. 2023ProRes: Exploring Degradation-aware Visual Prompt for Universal Image Restorationcited by 77
  26. 2019A General Decoupled Learning Framework for Parameterized Image Operatorscited by 72
  27. 2024Complexity Experts are Task-Discriminative Learners for Any Image Restorationcited by 72
  28. 2023GridFormer: Residual Dense Transformer with Grid Structure for Image Restoration in Adverse Weather Conditionscited by 62
  29. 2022TAPE: Task-Agnostic Prior Embedding for Image Restorationcited by 54
  30. 2023Prompt-In-Prompt Learning for Universal Image Restorationcited by 43
  31. 2025Vision-Language Gradient Descent-driven All-in-One Deep Unfolding Networkscited by 37
  32. 2024UniProcessor: A Text-induced Unified Low-level Image Processorcited by 22
  33. 2024LoRA-IR: Taming Low-Rank Experts for Efficient All-in-One Image Restorationcited by 18
  34. 2024HAIR: Hypernetworks-based All-in-One Image Restorationcited by 12
  35. 2025Efficient Degradation-agnostic Image Restoration via Channel-Wise Functional Decomposition and Manifold Regularizationcited by 8
  36. 2023Parameter Efficient Adaptation for Image Restoration with Heterogeneous Mixture-of-Expertscited by 7
  37. 2024Adaptive Blind All-in-One Image Restorationcited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026SLER-IR: Spherical Layer-wise Expert Routing for All-in-One Image Restoration
  2. 2025ClusIR: Towards Cluster-Guided All-in-One Image Restoration
  3. 2026UnSCAR: Universal, Scalable, Controllable, and Adaptable Image Restoration
  4. 2025M2Restore: Mixture-of-Experts-based Mamba-CNN Fusion Framework for All-in-One Image Restoration
  5. 2024Training-Free Large Model Priors for Multiple-in-One Image Restoration
  6. 2026Leveraging Multimodal Large Language Models for All-in-One Image Restoration via a Mixture of Frequency Experts
  7. 2026TTTIR: Unlocking Instance-Specific State Evolution via Test-Time Training for Image Restoration
  8. 2025RestoRect: Degraded Image Restoration via Latent Rectified Flow & Feature Distillation