跳到正文
arXiv · cs.LG· Yanjin He, Cuifeng Shen, Wanzhou Lei, Maohua Li, Tao Lan, Kan Liu, Hanlin Tang, Qing Qu·· 3 小时前

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

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

AI 导读

SWIFT 是一种自引导训练方法,用冻结的弱去噪器外推干净回归目标,提升更强扩散模型且不增加推理开销。在 ImageNet-256 上,它以约一半总训练算力超过基线最佳 FID;在 RAEv2 上无引导推理 FID 为 1.15,有引导为 0.99。

来源:arXiv · cs.LG · arxiv.org