arXiv:2410.18804cs.CVcs.LG2024-10NeurIPS被引 7

提出快速约束采样算法,让扩散模型生成更高效可靠。

Fast constrained sampling in pre-trained diffusion models

  • 用牛顿优化近似替代反向传播,加速推理
  • 在修复、超分等任务上效果优于现有方法
  • 适合需要快速高质量生成的科研与工程场景

大规模去噪扩散模型(如 Stable Diffusion)通过数十亿图像-文本对训练,具备丰富的图像统计先验,可用于多种推理任务。但面对新约束(如图像修复、补全)时,传统方法要么依赖反向传播,速度慢且内存高;要么仅局部约束,忽略长距离相关性。本文提出一种新算法,利用牛顿优化近似,在不进行反向传播的前提下实现快速高质量生成。该方法在线性约束(如图像修复、超分辨率)和非线性约束(如风格引导生成)下均表现优异,生成质量媲美甚至超越当前最优无训练推理方法,且耗时仅为几分之一。代码已开源:https://github.com/cvlab-stonybrook/fast-constrained-sampling。

原文摘要 · Abstract (English)

Large denoising diffusion models, such as Stable Diffusion, have been trained on billions of image-caption pairs to perform text-conditioned image generation. As a byproduct of this training, these models have acquired general knowledge about image statistics, which can be useful for other inference tasks. However, when confronted with sampling an image under new constraints, e.g. generating the missing parts of an image, using large pre-trained text-to-image diffusion models is inefficient and often unreliable. Previous approaches either utilized backpropagation through the denoiser network, making them significantly slower and more memory-demanding than simple text-to-image generation, or only enforced the constraint locally, failing to capture critical long-range correlations in the sampled image. In this work, we propose an algorithm that enables fast, high-quality generation under arbitrary constraints. We show that in denoising diffusion models, we can employ an approximation to Newton's optimization method that allows us to speed up inference and avoid the expensive backpropagation operations. Our approach produces results that rival or surpass the state-of-the-art training-free inference methods while requiring a fraction of the time. We demonstrate the effectiveness of our algorithm under both linear (inpainting, super-resolution) and non-linear (style-guided generation) constraints. An implementation is provided at https://github.com/cvlab-stonybrook/fast-constrained-sampling.

扩散模型图像修复快速生成

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