arXiv:2503.01288cs.CV2025-03CVPR被引 9

用单一扩散模型同时实现高保真与美观的零样本图像修复

Reconciling Stochastic and Deterministic Strategies for Zero-shot Image Restoration using Diffusion Model in Dual

  • 用同一扩散模型交替做确定性去噪和随机采样,互补提升修复效果
  • 在FFHQ和ImageNet上均优于现有方法,兼顾保真度与视觉质量
  • 仅需调节一个超参数即可控制失真与感知质量的平衡,实用性强

插件式(PnP)方法通过使用预训练的判别性去噪器作为隐式先验,实现了零样本图像修复的迭代求解。近期,基于生成扩散模型的采样变体因其出色的感知质量而受到青睐,但其在数据保真度上存在不足。本文提出一种新方案——双模态扩散修复(RDMD),仅依赖一个预训练扩散模型构建两个互补正则化器。该方法在迭代中交替进行确定性去噪与随机采样,旨在实现高保真且视觉效果佳的修复结果。此外,用户可通过单一超参数灵活调控失真与感知质量之间的权衡,增强实际应用中的适应性。在多种图像修复任务上的实验表明,该方法在FFHQ与ImageNet数据集上均显著优于现有方法。

原文摘要 · Abstract (English)

Plug-and-play (PnP) methods offer an iterative strategy for solving image restoration (IR) problems in a zero-shot manner, using a learned \textit{discriminative denoiser} as the implicit prior. More recently, a sampling-based variant of this approach, which utilizes a pre-trained \textit{generative diffusion model}, has gained great popularity for solving IR problems through stochastic sampling. The IR results using PnP with a pre-trained diffusion model demonstrate distinct advantages compared to those using discriminative denoisers, \ie improved perceptual quality while sacrificing the data fidelity. The unsatisfactory results are due to the lack of integration of these strategies in the IR tasks. In this work, we propose a novel zero-shot IR scheme, dubbed Reconciling Diffusion Model in Dual (RDMD), which leverages only a \textbf{single} pre-trained diffusion model to construct \textbf{two} complementary regularizers. Specifically, the diffusion model in RDMD will iteratively perform deterministic denoising and stochastic sampling, aiming to achieve high-fidelity image restoration with appealing perceptual quality. RDMD also allows users to customize the distortion-perception tradeoff with a single hyperparameter, enhancing the adaptability of the restoration process in different practical scenarios. Extensive experiments on several IR tasks demonstrate that our proposed method could achieve superior results compared to existing approaches on both the FFHQ and ImageNet datasets.

图像修复扩散模型零样本双模态

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