arXiv · Machine Learning· Yanjin He, Cuifeng Shen, Wanzhou Lei, Maohua Li, Tao Lan, Kan Liu, Hanlin Tang, Qing Qu·· 9 小时前AI 评分43
SWIFT:通过目标外推实现自引导弱到强扩散模型
SWIFT: Self-guided Weak-to-strong dIFfusion models via Target extrapolation
AI 导读
SWIFT提出通过目标外推进行自引导训练的方法,用冻结的弱去噪器提升更强的扩散模型,且不增加推理开销。在CIFAR-10和ImageNet-256上的实验显示,该方法在不同模型规模和表示上均有提升;一个ImageNet-256配置以约一半总训练计算量超越基线最佳FID。RAEv2上,SWIFT无推理时引导的FID为1.15,使用引导时为0.99。
来源:arXiv · Machine Learning · arxiv.org