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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Deblurring | GoPro | PSNR29.01 | 414 | |
| Deraining | Rain100L | PSNR38.28 | 280 | |
| Dehazing | SOTS | PSNR31.28 | 238 | |
| Denoising | BSD68 sigma=25 | PSNR31.54 | 118 | |
| Image Denoising | SIDD | PSNR28.23 | 114 | |
| Image Denoising | Urban100 (test) | -- | 72 | |
| Low-light Image Enhancement | LOL | PSNR22.3 | 53 | |
| Image Deraining | RealRain1K L | PSNR28.1 | 40 | |
| All-in-one Image Restoration | CBSD68, SOTS, and Rain100L | PSNR32.69 | 36 | |
| Image Denoising | BSD68 | PSNR (sigma=15)34.2 | 33 |