A central error measure in Gaussian DDPMs is the path-space KL divergence between the exact reverse chain and the learned Gaussian reverse process. This quantity is especially relevant for procedures such as classifier guidance, which perturb the entire reverse trajectory rather than only the terminal sample. Prior analyses show that standard isotropic reverse covariances suffer an unavoidable $Ω(1/T)$ path-KL error as the number of denoising steps $T$ grows. We show that matching the full posterior covariance breaks this barrier, yielding an order-wise improvement that reduces the path KL to $O(1/T^2)$. To make full covariance matching practical, we introduce the Lanczos Gaussian sampler (LGS), a training-free, matrix-free method for sampling from the optimal reverse covariance using only covariance-vector products, which are available through Jacobian-vector products of the posterior mean. LGS avoids dense covariance storage and auxiliary covariance models. We prove that LGS approximation error decays exponentially in the number of Lanczos steps, where each Lanczos step requires a single Jacobian-vector product. Empirically, using only just three such steps improves sample quality over strong diagonal-covariance baselines, including OCM-DDPM, across standard image benchmarks. This identifies full covariance matching as both theoretically valuable and practically accessible for fast DDPM sampling.


翻译:在高斯型去噪扩散概率模型(DDPM)中,一个核心误差度量是精确反向链与学习得到的高斯反向过程之间的路径空间KL散度。该量对于分类器引导等过程尤为重要——这些过程扰动的是整个反向轨迹,而非仅终端样本。先前分析表明,随着去噪步数T增加,标准各向同性反向协方差会导致不可避免的$\Omega(1/T)$路径KL误差。本研究证明,匹配完整后验协方差可突破这一壁垒,实现阶次改进,将路径KL降至$O(1/T^2)$。为使完整协方差匹配具有实用性,我们提出Lanczos高斯采样器(LGS)——一种无需训练、免矩阵构建的方法,仅通过协方差-向量乘积(可通过后验均值的雅可比-向量乘积获得)从最优反向协方差中采样。LGS避免了稠密协方差存储和辅助协方差模型。我们证明,LGS近似误差随Lanczos步数呈指数衰减,而每步仅需一次雅可比-向量乘积。实验表明,在标准图像基准测试中,仅使用三步Lanczos即可在样本质量上超越包括OCM-DDPM在内的强对角线协方差基线方法。这证实了完整协方差匹配在理论上具有重要价值,且在实践中可实现快速DDPM采样。

0
下载
关闭预览

相关内容

《线性协方差方法在非线性导弹制导中的应用​》219页
专知会员服务
17+阅读 · 2025年5月26日
【ICML2025】用于概率时间序列预测的非平稳扩散方法
专知会员服务
10+阅读 · 2025年5月10日
【CVPR2025】在去噪扩散模型中优化最短路径
专知会员服务
16+阅读 · 2025年3月10日
生成式人工智能的扩散模型概述
专知会员服务
67+阅读 · 2024年12月8日
【AAAI2021】对比聚类,Contrastive Clustering
专知
26+阅读 · 2021年1月30日
半监督深度学习小结:类协同训练和一致性正则化
推荐算法:Match与Rank模型的交织配合
从0到1
15+阅读 · 2017年12月18日
各种相似性度量及Python实现
机器学习算法与Python学习
11+阅读 · 2017年7月6日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2014年12月31日
国家自然科学基金
1+阅读 · 2014年12月31日
VIP会员
最新内容
驱动军事决策变革的顶尖人工智能指挥系统
专知会员服务
9+阅读 · 8月11日
非对称防御中的自组织临界性:俄乌战争
专知会员服务
10+阅读 · 8月10日
《战争中的大语言模型监管》
专知会员服务
14+阅读 · 8月10日
《边缘计算关键技术分析及美军作战实践应用》
边缘计算的军事应用
专知会员服务
13+阅读 · 8月9日
相关基金
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2014年12月31日
国家自然科学基金
1+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员