arXiv 人工智能· Donghoon Ahn, Runqian Wang, Alexei A. Efros, Taesung Park, Yash Belhe·· 12 小时前AI 评分39
扩散模型的校正训练方法
Correction Training for Diffusion Models
AI 导读
研究者提出面向扩散模型的校正训练方法,通过校正器在采样中迭代精炼降噪预测,有效缩小训练与推理差距并减少误差累积。该方法在 DiT、SiT 和 JiT 上显著改善了无引导与无条件生成的 FID 分数,表现媲美 CFG,二者结合还能提升多样性并减轻过度饱和。此外,精炼预测可蒸馏回基础扩散模型,实现零额外推理开销保留生成质量。
来源:arXiv 人工智能 · arxiv.org