跳到正文
原文
HF Daily Papers·· 2 天前精选AI 评分40

SWIFT:通过目标外推实现自引导弱到强扩散模型训练

SWIFT: Self-guided Weak-to-strong dIFfusion models via Target extrapolation

AI 导读

SWIFT 提出一种自引导训练方法,利用冻结的弱去噪器将回归目标沿远离其预测的方向外推,以改进更强的扩散模型且不增加推理开销。弱引导器可采用训练早期检查点,主模型训练无需反向传播,之后可将其丢弃。

推荐理由

论文给出目标外推机制、理论证明及 CIFAR-10 和 ImageNet-256 实验,可用于比较弱到强扩散模型的训练效率。

来源:HF Daily Papers · huggingface.co