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

DRNet: All-in-One Image Restoration via Prior-Guided Dynamic Reparameterization

About

All-in-one image restoration aims to handle diverse degradations within a single model. However, existing methods often suffer from three key limitations: 1) per-input computational overhead from dynamic degradation estimation; 2) optimization challenges due to task heterogeneity; and 3) inefficient, frequency-agnostic encoder designs. To overcome these, we introduce the Dynamic Reparameterization Network (DRNet), a novel framework operating on an initialization-stage reconfiguration paradigm that fundamentally eliminates per-input overhead. At its core, a Dynamic Reparameterization MLP (DRMLP) guided by a Task-Specific Modulator (TSM), which effectively mitigates task heterogeneity by orchestrating both specific restoration goals and a versatile general-purpose mode within a unified architecture. Furthermore, we incorporate a Continuous Wavelet Transform Encoder (CWTE) that explicitly leverages frequency characteristics via wavelet decomposition for a lightweight yet powerful design. Extensive experiments demonstrate that DRNet achieves state-of-the-art performance across five restoration tasks with superior parameter efficiency. Crucially, it showcases unique flexibility, excelling as both a highly competitive foundation model for blind restoration and a top-performing user-guided specialist.

Ao Li, Xiaoning Liu, Sheng Li, Yapeng Du, Zhen Long, Lei Luo, Le Zhang, Ce Zhu• 2026

Related benchmarks

TaskDatasetResultRank
Image DeblurringGoPro
PSNR29.01
414
DerainingRain100L
PSNR38.28
280
DehazingSOTS
PSNR31.28
238
DenoisingBSD68 sigma=25
PSNR31.54
118
Image DenoisingSIDD
PSNR28.23
114
Image DenoisingUrban100 (test)--
72
Low-light Image EnhancementLOL
PSNR22.3
53
Image DerainingRealRain1K L
PSNR28.1
40
All-in-one Image RestorationCBSD68, SOTS, and Rain100L
PSNR32.69
36
Image DenoisingBSD68
PSNR (sigma=15)34.2
33
Showing 10 of 14 rows

Other info

Follow for update