跳到正文
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