arXiv:2504.15513cs.CV2025-04ICLR被引 2

一键增强图像,速度比传统方法快得多。

InstaRevive: One-Step Image Enhancement via Dynamic Score Matching

  • 用动态分数匹配实现单步图像增强,跳过繁琐迭代。
  • 在多个数据集上达到高保真度,提升效率并保持视觉质量。
  • 适合需要快速高质量图像处理的实时应用开发者。

图像增强在复杂环境和成像设备固有局限下具有广泛应用。近年来基于扩散的方法虽表现优异,但需长时间且计算量大的迭代采样。为此,我们提出 InstaRevive,一种简单而强大的图像增强框架,通过基于分数的扩散蒸馏,充分利用预训练扩散模型的生成能力,并显著减少采样步骤。为充分挖掘预训练模型潜力,我们设计了一种实用有效的扩散蒸馏流程,采用动态控制机制以解决分数匹配中更新方向不准确的问题。该控制策略实现动态扩散范围,促进扩散模型内去噪轨迹的精确学习,并确保训练过程中分布匹配梯度的准确性。此外,为增强生成引导,我们引入图像字幕生成的文本提示作为辅助条件,进一步激发扩散模型的生成潜力。大量实验验证了该框架在多样挑战性任务与数据集上的有效性,展现出 InstaRevive 在生成高质量、视觉吸引人结果方面的卓越性能与效率。代码已开源:https://github.com/EternalEvan/InstaRevive。

原文摘要 · Abstract (English)

Image enhancement finds wide-ranging applications in real-world scenarios due to complex environments and the inherent limitations of imaging devices. Recent diffusion-based methods yield promising outcomes but necessitate prolonged and computationally intensive iterative sampling. In response, we propose InstaRevive, a straightforward yet powerful image enhancement framework that employs score-based diffusion distillation to harness potent generative capability and minimize the sampling steps. To fully exploit the potential of the pre-trained diffusion model, we devise a practical and effective diffusion distillation pipeline using dynamic control to address inaccuracies in updating direction during score matching. Our control strategy enables a dynamic diffusing scope, facilitating precise learning of denoising trajectories within the diffusion model and ensuring accurate distribution matching gradients during training. Additionally, to enrich guidance for the generative power, we incorporate textual prompts via image captioning as auxiliary conditions, fostering further exploration of the diffusion model. Extensive experiments substantiate the efficacy of our framework across a diverse array of challenging tasks and datasets, unveiling the compelling efficacy and efficiency of InstaRevive in delivering high-quality and visually appealing results. Code is available at https://github.com/EternalEvan/InstaRevive.

图像增强扩散模型单步生成

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