arXiv cs.AI· Donghoon Ahn, Runqian Wang, Alexei A. Efros, Taesung Park, Yash Belhe·· 19 天前AI 评分41
扩散模型的校正训练方法
Correction Training for Diffusion Models
AI 导读
研究人员提出一种用于扩散模型的校正训练方法，通过在去噪步骤中引入校正器迭代优化预测，以缩小训练与推理间的曝光偏差差距。该方法在 DiT、SiT 和 JiT 上显著改善了无引导及无条件生成的 FID 分数，效果可媲美 CFG，且结合 CFG 能提升多样性并减少过饱和。校正带来的额外推理开销还可通过蒸馏回基础模型消除，在无额外推理成本下保留质量提升。
来源：arXiv cs.AI · arxiv.org