arXiv:2505.24222cs.CVcs.LG2025-05ICCV被引 5

用二阶优化提升扩散模型采样质量,计算开销却几乎不变。

Unleashing High-Quality Image Generation in Diffusion Sampling Using Second-Order Levenberg-Marquardt-Langevin

  • 基于低秩近似和阻尼机制,无训练地逼近扩散模型的海森矩阵几何。
  • 在多个预训练模型上显著提升图像生成质量,且计算开销可忽略。
  • 适合关注高质量图像生成与高效采样的研究人员。

扩散模型(DMs)通过学习数据分布的噪声得分函数,展现出强大的图像生成能力。当前的采样技术通常在每个噪声级别使用一阶朗之万动力学,研究重点集中在跨层级去噪策略的优化。尽管在马尔可夫链蒙特卡洛(MCMC)中利用额外的二阶海森几何以提升朗之万采样质量是常见做法,但直接在高维扩散模型中应用海森几何会导致二次复杂度计算开销,难以扩展。本文提出一种新颖的无训练莱文贝格-马夸特-朗之万(LML)方法,通过借鉴著名的莱文贝格-马夸特优化算法,近似扩散模型的海森几何。本方法包含两项关键创新:(1) 利用扩散模型的内在结构进行海森矩阵的低秩近似,避免显式的二次复杂度计算;(2) 引入阻尼机制稳定近似海森矩阵。该近似海森几何使扩散采样能够执行更精确的步骤,从而提升图像生成质量。我们进一步进行了理论分析,证明了低秩近似的误差界以及阻尼机制的收敛性。大量实验验证,该方法在多个预训练扩散模型上显著提升了图像生成质量,且计算开销极小。

原文摘要 · Abstract (English)

The diffusion models (DMs) have demonstrated the remarkable capability of generating images via learning the noised score function of data distribution. Current DM sampling techniques typically rely on first-order Langevin dynamics at each noise level, with efforts concentrated on refining inter-level denoising strategies. While leveraging additional second-order Hessian geometry to enhance the sampling quality of Langevin is a common practice in Markov chain Monte Carlo (MCMC), the naive attempts to utilize Hessian geometry in high-dimensional DMs lead to quadratic-complexity computational costs, rendering them non-scalable. In this work, we introduce a novel Levenberg-Marquardt-Langevin (LML) method that approximates the diffusion Hessian geometry in a training-free manner, drawing inspiration from the celebrated Levenberg-Marquardt optimization algorithm. Our approach introduces two key innovations: (1) A low-rank approximation of the diffusion Hessian, leveraging the DMs' inherent structure and circumventing explicit quadratic-complexity computations; (2) A damping mechanism to stabilize the approximated Hessian. This LML approximated Hessian geometry enables the diffusion sampling to execute more accurate steps and improve the image generation quality. We further conduct a theoretical analysis to substantiate the approximation error bound of low-rank approximation and the convergence property of the damping mechanism. Extensive experiments across multiple pretrained DMs validate that the LML method significantly improves image generation quality, with negligible computational overhead.

扩散模型采样优化图像生成

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