SDXL-Lightning：渐进式对抗扩散蒸馏
计算机视觉与模式识别
2024-03-05 v3 人工智能
机器学习
摘要
我们提出了一种扩散蒸馏方法，在基于 SDXL 的单步/少步 1024px 文本到图像生成方面达到了新的最先进水平 (SOTA)。我们的方法结合了渐进式和对抗式蒸馏，以实现质量与模式覆盖之间的平衡。在本文中，我们讨论了理论分析、判别器设计、模型公式化和训练技巧。我们开源了 distilled SDXL-Lightning 模型，提供 LoRA 和完整 UNet 权重两种形式。
引用
@article{arxiv.2402.13929,
title = {SDXL-Lightning: Progressive Adversarial Diffusion Distillation},
author = {Shanchuan Lin and Anran Wang and Xiao Yang},
journal= {arXiv preprint arXiv:2402.13929},
year = {2024}
}