仅需采样一次:通过自协同扩散生成对抗网络驯化单步文本到图像合成
💡
原文中文,约400字,阅读约需1分钟。
📝
内容提要
本文介绍了YONOS-SR,一种新的稳定的基于扩散的图像超分辨率方法。通过尺度蒸馏方法训练SR模型,蒸馏模型明显优于直接训练高放大尺度的模型。使用只需一步的扩散模型可以冻结U-Net并微调解码器。经过空间蒸馏的U-Net和经过微调的解码器的组合比需要200步的最先进方法更优越。
🎯
关键要点
- YONOS-SR是一种新的稳定的基于扩散的图像超分辨率方法。
- 该方法仅使用一次DDIM步骤即可获得最先进的结果。
- 提出了一种新颖的尺度蒸馏方法来训练SR模型。
- SR问题被简化为教师模型在较小放大尺度上的训练。
- 训练过程中使用教师的预测作为目标来训练较高放大尺度的学生模型。
- 蒸馏模型明显优于直接训练高放大尺度的模型,推断过程中只需少量步骤。
- 强大的扩散模型使得可以冻结U-Net并微调解码器。
- 经过空间蒸馏的U-Net和微调的解码器组合比需要200步的最先进方法更为优越。
➡️