arXiv 机器学习· Yuxiang Fu, Qi Yan, Zike Wu, Yongxing Zhang, Purang Abolmaesumi, Lele Wang, Renjie Liao·· 7 小时前AI 评分35
Consistent Distribution Matching:面向扩散模型的无数据蒸馏方法
Consistent Distribution Matching for Data-Free Diffusion Distillation
AI 导读
研究人员提出无数据且无需模拟的扩散模型蒸馏方法 Consistent Distribution Matching,通过单个学生网络统一样本生成与得分估计以加速推理。在 ImageNet 256×256 基准上,该方法训练 40 个 epoch 即可在 1-NFE 下实现 2.04 的 FID,在 4-NFE 下实现 1.37 的 FID。相关代码与模型已公开发布。
来源:arXiv 机器学习 · arxiv.org