English

Bridging Degradation Discrimination and Generation for Universal Image Restoration

Computer Vision and Pattern Recognition 2026-02-03 v1

Abstract

Universal image restoration is a critical task in low-level vision, requiring the model to remove various degradations from low-quality images to produce clean images with rich detail. The challenges lie in sampling the distribution of high-quality images and adjusting the outputs on the basis of the degradation. This paper presents a novel approach, Bridging Degradation discrimination and Generation (BDG), which aims to address these challenges concurrently. First, we propose the Multi-Angle and multi-Scale Gray Level Co-occurrence Matrix (MAS-GLCM) and demonstrate its effectiveness in performing fine-grained discrimination of degradation types and levels. Subsequently, we divide the diffusion training process into three distinct stages: generation, bridging, and restoration. The objective is to preserve the diffusion model's capability of restoring rich textures while simultaneously integrating the discriminative information from the MAS-GLCM into the restoration process. This enhances its proficiency in addressing multi-task and multi-degraded scenarios. Without changing the architecture, BDG achieves significant performance gains in all-in-one restoration and real-world super-resolution tasks, primarily evidenced by substantial improvements in fidelity without compromising perceptual quality. The code and pretrained models are provided in https://github.com/MILab-PKU/BDG.

Keywords

Cite

@article{arxiv.2602.00579,
  title  = {Bridging Degradation Discrimination and Generation for Universal Image Restoration},
  author = {JiaKui Hu and Zhengjian Yao and Lujia Jin and Yanye Lu},
  journal= {arXiv preprint arXiv:2602.00579},
  year   = {2026}
}

Comments

Accepted by ICLR 2026

R2 v1 2026-07-01T09:29:11.038Z