arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.01247cs.CV

用于统一图像恢复与增强的生成式扩散先验

Generative Diffusion Prior for Unified Image Restoration and Enhancement

Fudan University(复旦大学);Shanghai AI Laboratory(上海人工智能实验室);Nanyang Technological University(南洋理工大学)

Ben Fei, Zhaoyang Lyu, Liang Pan, Junzhe Zhang, Weidong Yang, Tianyue Luo, Bo Zhang, Bo Dai

AI总结:

本文提出无监督的生成式扩散先验(GDP),利用预训练DDPM解决线性、非线性和盲图像问题,通过条件引导和退化参数优化实现盲恢复,在多种任务上超越现有方法并支持任意分辨率泛化。

摘要:

现有图像恢复方法大多利用自然图像的后验分布。然而,它们通常假设退化过程已知且需要监督训练,这限制了其在复杂真实场景中的适应性。在本研究中,我们提出生成式扩散先验(GDP),以无监督采样的方式有效建模后验分布。GDP利用预训练去噪扩散生成模型(DDPM)来解决线性逆、非线性或盲问题。具体而言,GDP系统性地探索了条件引导协议,并验证其比常用引导方式更实用。此外,GDP擅长在去噪过程中优化退化模型参数,从而实现盲图像恢复。同时,我们设计了分层引导和基于分块的方法,使GDP能够生成任意分辨率的图像。实验表明,GDP在多个图像数据集上展现出广泛适用性,涵盖超分辨率、去模糊、图像修复和着色等线性问题,以及低光增强和HDR图像恢复等非线性和盲问题。在重建质量和感知质量方面,GDP在多个基准测试中均超越当前领先的零样本方法。此外,GDP对来自ImageNet训练集分布之外的自然图像或合成图像,也能保持良好的泛化能力。

英文摘要:

Existing image restoration methods mostly leverage the posterior distribution of natural images. However, they often assume known degradation and also require supervised training, which restricts their adaptation to complex real applications. In this work, we propose the Generative Diffusion Prior (GDP) to effectively model the posterior distributions in an unsupervised sampling manner. GDP utilizes a pre-train denoising diffusion generative model (DDPM) for solving linear inverse, non-linear, or blind problems. Specifically, GDP systematically explores a protocol of conditional guidance, which is verified more practical than the commonly used guidance way. Furthermore, GDP is strength at optimizing the parameters of degradation model during the denoising process, achieving blind image restoration. Besides, we devise hierarchical guidance and patch-based methods, enabling the GDP to generate images of arbitrary resolutions. Experimentally, we demonstrate GDP's versatility on several image datasets for linear problems, such as super-resolution, deblurring, inpainting, and colorization, as well as non-linear and blind issues, such as low-light enhancement and HDR image recovery. GDP outperforms the current leading unsupervised methods on the diverse benchmarks in reconstruction quality and perceptual quality. Moreover, GDP also generalizes well for natural images or synthesized images with arbitrary sizes from various tasks out of the distribution of the ImageNet training set.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑