arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2605.08577cs.CVcs.LG

通过自蒸馏改进生成对抗网络

Improving Generative Adversarial Networks with Self-Distillation

Poznan University of Technology(波兹南技术大学)

Antoni Nowinowski, Krzysztof Krawiec

AI总结:

本文提出自蒸馏GAN(SD-GAN),利用指数移动平均生成器作为教师模型,通过感知损失指导训练生成器,提升图像质量并稳定优化轨迹。

摘要:

在现代GAN中,维护生成器权重的指数移动平均(EMA)是标准实践,因为平均模型在训练过程中表现优于活跃训练的生成器。然而,EMA生成器仅用于最终部署,不影响训练过程。为利用这一机会,我们引入自蒸馏GAN(SD-GAN),利用EMA生成器作为教师模型,通过感知损失指导活跃生成器(学生)。我们证明了SD-GAN在Dirac-GAN设置下的局部渐近稳定性,并展示了其能抑制传统GAN中的寄生循环行为。在多个架构和数据集上的实验证明,SD-GAN在多个指标(如FID和随机FID)上提升了最终图像质量,稳定了优化轨迹,并提供了与传统对抗损失不完全相关的额外学习指导。此外,它在微调预训练GAN模型方面也证明了有效性。

英文摘要:

In modern GANs, maintaining an Exponential Moving Average (EMA) of the generator's weights is a standard practice, as such an averaged model consistently outperforms the actively trained generator. However, the EMA generator is used for final deployment only and does not influence the training process. To address this missed opportunity, we introduce Self-Distilled GAN (SD-GAN) that employs the EMA generator as a teacher to guide the active generator (student) via perceptual loss. We prove the local asymptotic stability of SD-GAN in the Dirac-GAN setting and show that it dampens the parasitic cycling behavior that plagues the conventional GANs. Empirical evaluations across established architectures and datasets demonstrate that SD-GAN improves the final image quality on several metrics (FID and random-FID in particular), stabilizes the optimization trajectory and provides additional learning guidance that is not trivially correlated with the conventional adversarial loss. It also proves effective for fine-tuning pretrained GAN models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑