中文

基于 Taylor 近似的控制变量实现扩散模型梯度方差缩减

机器学习 2024-08-23 v1 人工智能

摘要

基于去噪评分匹配训练的评分模型在生成高维数据方面效果极佳。然而,其训练目标的高方差阻碍了优化。在本文中,我们通过对训练目标及其梯度进行 kk 阶 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