跳到正文
原文
AGI Hunt· CompVis·· 3 小时前AI 评分44

CompVis 改进分布扩散模型,ImageNet 上 4 步采样达 4.48 FID

AI 导读

CompVis 发布分布扩散模型改进版 iDDM,将粒子扩展推迟到 transformer 后期层,改用随时间变化的 scoring rule 调度。DiT-XL/2 在 ImageNet-256² 上从零单阶段训练达 4 步 4.48 FID、50 步 2.38 FID,FID 不随 4 到 50 NFE 的采样预算增长而退化。代码与模型已开源。

来源:AGI Hunt · agihunt.info