Abstract:In this paper, we address the unified image restoration challenge by reframing it as a contrastive learning-based classification problem. Despite the significant strides made by deep learning methods in enhancing image restoration quality, their limited capacity to generalize across diverse degradation types and intensities necessitates the training of separate models for each specific degradation scenario. We proposes an all-encompassing approach that can restore images from various unknown corruption types and levels. We devise a method that learns representations of the latent sharp image’s degradation and accompanying textual features (such as dataset categories and image content descriptions), converting these into prompts which are then embedded within a reconstruction network model to enhance cross-database restoration performance. This culminates in a unified image reconstruction framework. The study involves two stages: In the first stage, we design a MultiContentNet that learns multi-modal features (MMFs) of the latent sharp image. This network encodes the visual degradation expressions and contextual text features into latent variables, thereby exerting a guided classification effect. Specifically, MultiContentNet is trained as an auxiliary controller capable of taking the degraded input image and, through contrastive learning, extracts MMFs of the latent target image. This effectively generates natural classifiers tailored for different degradation types. The second phase integrates the learned MMFs into an image restoration network via cross-attention mechanisms. This guides the restoration model to learn high-fidelity image recovery. Experiments conducted on six blind image restoration tasks demonstrate that the proposed method achieves state-of-the-art performance, highlighting the potential significance of large-scale pretrained vision-language models’ MMFs in advancing high-quality unified image reconstruction.

Multi-wavelet Guided Deep Mean-Shift Prior for Image Restoration

Multi-frame Blind Deconvolution Using Sparse Priors

Deep Denoiser Prior Based Deep Analytic Network for Lensless Image Restoration.

Image Restoration using Autoencoding Priors

Deep multi-scale gabor wavelet network for image restoration

Wavelet Frame-Based Image Restoration Using Sparsity, Nonlocal, and Support Prior of Frame Coefficients

Blind multi-Poissonian image deconvolution with sparse log-step gradient prior

Multi-modal Degradation Feature Learning for Unified Image Restoration Based on Contrastive Learning

Plug-and-Play Image Restoration with Deep Denoiser Prior

Denoising Prior Driven Deep Neural Network for Image Restoration

Wide & deep learning for spatial & intensity adaptive image restoration

Image Restoration Based on Transformed Total Variation and Deep Image prior

Hybrid regularization inspired by total variation and deep denoiser prior for image restoration

Stochastic Deep Restoration Priors for Imaging Inverse Problems

Deep Plug-and-Play Prior for Hyperspectral Image Restoration

Dual Prior Learning for Blind and Blended Image Restoration

Learning Deep CNN Denoiser Prior for Image Restoration

Blind Image Restoration with Flow Based Priors

Training-Free Large Model Priors for Multiple-in-One Image Restoration

Natural Image Deblurring Based On Ringing Artifacts Removal Via Knowledge-Driven Gradient Distribution Priors

Multiscale Synergism Ensemble Progressive and Contrastive Investigation for Image Restoration