arXiv · cs.LG· Seo Taek Kong, Weina Wang, R. Srikant·· 4 小时前
扩散模型噪声调度设计:一种最优控制视角
Noise Schedule Design for Diffusion Models: An Optimal Control Perspective
AI 导读
研究者提出用最优控制框架分析和设计扩散模型的噪声调度,将 Fisher 信息作为状态变量、噪声调度作为控制输入,其目标泛函被证明是 KL 采样误差的上界。求解该问题后,他们给出了可实现 Õ(d/n) 采样误差的噪声调度充分条件(d 为数据维度,n 为离散化步数),并在额外数据分布假设下推导出一族能还原实践中常用调度函数形式的噪声调度。
来源:arXiv · cs.LG · arxiv.org