Abstract:Although learning-based image restoration methods have made significant progress, they still struggle with limited generalization to real-world scenarios due to the substantial domain gap caused by training on synthetic data. Existing methods address this issue by improving data synthesis pipelines, estimating degradation kernels, employing deep internal learning, and performing domain adaptation and regularization. Previous domain adaptation methods have sought to bridge the domain gap by learning domain-invariant knowledge in either feature or pixel space. However, these techniques often struggle to extend to low-level vision tasks within a stable and compact framework. In this paper, we show that it is possible to perform domain adaptation via the noise space using diffusion models. In particular, by leveraging the unique property of how auxiliary conditional inputs influence the multi-step denoising process, we derive a meaningful diffusion loss that guides the restoration model in progressively aligning both restored synthetic and real-world outputs with a target clean distribution. We refer to this method as denoising as adaptation. To prevent shortcuts during joint training, we present crucial strategies such as channel-shuffling layer and residual-swapping contrastive learning in the diffusion model. They implicitly blur the boundaries between conditioned synthetic and real data and prevent the reliance of the model on easily distinguishable features. Experimental results on three classical image restoration tasks, namely denoising, deblurring, and deraining, demonstrate the effectiveness of the proposed method.

Boosting denoisers with reinforcement learning for image restoration

Image Quality Assessment–driven Reinforcement Learning for Mixed Distorted Image Restoration

REPNP: Plug-and-Play with Deep Reinforcement Learning Prior for Robust Image Restoration

Plug-and-Play Image Restoration with Deep Denoiser Prior

Distort-and-Recover: Color Enhancement using Deep Reinforcement Learning

Denoising Prior Driven Deep Neural Network for Image Restoration

Mixed distortion image enhancement method based on joint of deep residuals learning and reinforcement learning

R3L: Connecting Deep Reinforcement Learning to Recurrent Neural Networks for Image Denoising via Residual Recovery

Unsupervised Image Restoration Using Partially Linear Denoisers

JE2NET: Joint Exploitation and Exploration in Reinforcement Learning Based Image Restoration

A Double Feature Fusion Network with Progressive Learning for Sharper Inpainting

Synthesizing Realistic Image Restoration Training Pairs: A Diffusion Approach

Denoising as Adaptation: Noise-Space Domain Adaptation for Image Restoration

Image Restoration Based on End-to-End Unrolled Network

Denoising Diffusion Models for Plug-and-Play Image Restoration

Stochastic Deep Restoration Priors for Imaging Inverse Problems

Dual Prior Learning for Blind and Blended Image Restoration

Multiscale Synergism Ensemble Progressive and Contrastive Investigation for Image Restoration

A Progressive Image Restoration Network for High-order Degradation Imaging in Remote Sensing

LoRA-IR: Taming Low-Rank Experts for Efficient All-in-One Image Restoration

Rank-One Network: An Effective Framework for Image Restoration