Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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 \textbf{Continuous Expert Assembly} (CEA), a token-wise dynamic parameterization framework for all-in-one image restoration. CEA employs a lightweight \textbf{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.

Haisen He, Xiangyu Zou, SongLin Dong, Heng Li, Yihong Gong, Zhiheng Ma• 2026

Related benchmarks

TaskDatasetResultRank
Image DeblurringGoPro
PSNR29.9
414
DerainingRain100L
PSNR38.95
280
Image DerainingRain100L
PSNR38.92
249
DehazingSOTS
PSNR31.96
238
Image DehazingSOTS
PSNR32.33
171
DenoisingBSD68 sigma=25
PSNR31.53
118
Gaussian DenoisingBSD68
PSNR (sigma=50)28.32
39
Image RestorationCDD11
Average PSNR29.79
19
All-in-one Image RestorationAIO-3 Protocol SOTS, Rain100L, BSD68
Average PSNR33.07
18
Low-light enhancementLOL v1
PSNR23.58
18
Showing 10 of 13 rows

Other info

Follow for update