arXiv:2602.00579cs.CV2026-02中稿 · ICLR

用新方法同时识别图像退化类型和修复,提升通用图像恢复效果。

Bridging Degradation Discrimination and Generation for Universal Image Restoration

  • 设计多角度多尺度灰度共生矩阵,精细区分退化类型与程度。
  • 分三阶段训练扩散模型,修复纹理更丰富,保真度显著提升。
  • 无需改架构,适合处理多种退化场景,适合图像修复研究者。

通用图像恢复是低层视觉中的关键任务,需从低质量图像中去除各类退化以生成细节丰富的清晰图像。挑战在于高质图像分布采样及基于退化类型的输出调整。本文提出一种新方法——退化判别与生成桥梁(BDG),旨在协同解决上述问题。首先,提出多角度多尺度灰度共生矩阵(MAS-GLCM),有效实现退化类型与程度的细粒度判别。随后,将扩散训练过程划分为生成、桥梁与恢复三个阶段,旨在保留扩散模型还原丰富纹理的能力,同时将MAS-GLCM的判别信息融入恢复过程,增强其在多任务、多退化场景下的表现。无需改变网络结构,BDG在全任务图像恢复与真实世界超分辨率任务中均取得显著性能提升,主要体现在保真度大幅提升且未牺牲感知质量。代码与预训练模型已开源于 https://github.com/MILab-PKU/BDG。

原文摘要 · Abstract (English)

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.

图像恢复扩散模型退化识别多任务学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。