基于 Taylor 近似的控制变量实现扩散模型梯度方差缩减
机器学习
2024-08-23 v1 人工智能
摘要
基于去噪评分匹配训练的评分模型在生成高维数据方面效果极佳。然而,其训练目标的高方差阻碍了优化。在本文中,我们通过对训练目标及其梯度进行 阶 Taylor 展开所导出的控制变量来尝试降低方差。我们证明了两种方法等价,并在低维问题设置下经验性地验证了我们方法的有效性;同时我们也研究了其在更大规模问题上的效果。
引用
@article{arxiv.2408.12270,
title = {Variance reduction of diffusion model's gradients with Taylor approximation-based control variate},
author = {Paul Jeha and Will Grathwohl and Michael Riis Andersen and Carl Henrik Ek and Jes Frellsen},
journal= {arXiv preprint arXiv:2408.12270},
year = {2024}
}
备注
14 pages, ICML Structured Probabilistic Inference & Generative Modeling 2024