Continuous Expert Assembly: Instance-Conditioned Low-Rank Residuals for All-in-One Image Restoration
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
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2021SwinIR: Image Restoration Using Swin Transformercited by 1,869
- 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,618
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2020Rethinking Attention with Performerscited by 1,584
- 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,352
- 2016HyperNetworkscited by 1,003
- 2020Pre-Trained Image Processing Transformercited by 891
37 more
- 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 864
- 2021Multi-Stage Progressive Image Restorationcited by 813
- 2021Scaling Vision with Sparse Mixture of Expertscited by 740
- 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 740
- 2022Simple Baselines for Image Restorationcited by 701
- 2018Efficient Attention: Attention with Linear Complexitiescited by 366
- 2019Dynamic Convolution: Attention over Convolution Kernelscited by 355
- 2024MambaIR: A Simple Baseline for Image Restoration with State-Space Modelcited by 349
- 2016Dynamic Filter Networkscited by 318
- 2022MAXIM: Multi-Axis MLP for Image Processingcited by 301
- 2019CondConv: Conditionally Parameterized Convolutions for Efficient Inferencecited by 273
- 2023From Sparse to Soft Mixtures of Expertscited by 245
- 2021TransWeather: Transformer-based Restoration of Images Degraded by Adverse Weather Conditionscited by 243
- 2021HINet: Half Instance Normalization Network for Image Restorationcited by 236
- 2022Restoring Vision in Adverse Weather Conditions with Patch-Based Denoising Diffusion Modelscited by 215
- 2020Focal Frequency Loss for Image Reconstruction and Synthesiscited by 204
- 2017Burst Denoising with Kernel Prediction Networkscited by 156
- 2024InstructIR: High-Quality Image Restoration Following Human Instructionscited by 139
- 2022Deep Generalized Unfolding Networks for Image Restorationcited by 119
- 2023PromptIR: Prompting for All-in-One Blind Image Restorationcited by 113
- 2023Controlling Vision-Language Models for Multi-Task Image Restorationcited by 108
- 2022Cross Aggregation Transformer for Image Restorationcited by 92
- 2024OneRestore: A Universal Restoration Framework for Composite Degradationcited by 91
- 2020FD-GAN: Generative Adversarial Networks with Fusion-discriminator for Single Image Dehazingcited by 79
- 2023ProRes: Exploring Degradation-aware Visual Prompt for Universal Image Restorationcited by 77
- 2019A General Decoupled Learning Framework for Parameterized Image Operatorscited by 72
- 2024Complexity Experts are Task-Discriminative Learners for Any Image Restorationcited by 72
- 2023GridFormer: Residual Dense Transformer with Grid Structure for Image Restoration in Adverse Weather Conditionscited by 62
- 2022TAPE: Task-Agnostic Prior Embedding for Image Restorationcited by 54
- 2023Prompt-In-Prompt Learning for Universal Image Restorationcited by 43
- 2025Vision-Language Gradient Descent-driven All-in-One Deep Unfolding Networkscited by 37
- 2024UniProcessor: A Text-induced Unified Low-level Image Processorcited by 22
- 2024LoRA-IR: Taming Low-Rank Experts for Efficient All-in-One Image Restorationcited by 18
- 2024HAIR: Hypernetworks-based All-in-One Image Restorationcited by 12
- 2025Efficient Degradation-agnostic Image Restoration via Channel-Wise Functional Decomposition and Manifold Regularizationcited by 8
- 2023Parameter Efficient Adaptation for Image Restoration with Heterogeneous Mixture-of-Expertscited by 7
- 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
- 2026SLER-IR: Spherical Layer-wise Expert Routing for All-in-One Image Restoration
- 2025ClusIR: Towards Cluster-Guided All-in-One Image Restoration
- 2026UnSCAR: Universal, Scalable, Controllable, and Adaptable Image Restoration
- 2025M2Restore: Mixture-of-Experts-based Mamba-CNN Fusion Framework for All-in-One Image Restoration
- 2024Training-Free Large Model Priors for Multiple-in-One Image Restoration
- 2026Leveraging Multimodal Large Language Models for All-in-One Image Restoration via a Mixture of Frequency Experts
- 2026TTTIR: Unlocking Instance-Specific State Evolution via Test-Time Training for Image Restoration
- 2025RestoRect: Degraded Image Restoration via Latent Rectified Flow & Feature Distillation