arXiv cs.LG· Woobin Park, Jaeyeon Kim, Yilun Du, Sitan Chen, Bohyung Han·· 7 小时前AI 评分29
Priming Models:一步生成是否被过度复杂化?
Are We Overthinking One-Step Generation?
AI 导读
研究团队提出 Priming Models,一个两阶段一步生成器训练框架:先以固定噪声水平的像素空间去噪等重建目标预训练,再进行特征空间分布后训练。在 ImageNet 256×256 上,该方法以约十分之一于同类 FID 方法的训练算力,实现一步生成器中 SOTA 的 FID 1.19,并已开源代码。
来源:arXiv cs.LG · arxiv.org