用纯Transformer和隐空间训练,让GAN轻松扩展到大模型
Scalable GANs with Transformers
- 在变分自编码器隐空间中训练,结合纯Transformer结构提升可扩展性
- 在ImageNet-256上仅用60轮即达FID 2.18,比基线少4倍训练轮次
- 适合想高效训练大规模生成模型的研究者或工程团队
可扩展性推动了生成建模的进展,但对抗学习中的可扩展性原则仍不明确。本文通过两种已被证明在其他生成模型中有效的设计:在紧凑的变分自编码器(VAE)隐空间中训练,以及采用纯Transformer生成器与判别器,研究了生成对抗网络(GANs)的可扩展性。隐空间训练实现高效计算并保持感知保真度,与纯Transformer结构天然契合,其性能随计算资源增加而提升。基于此,我们分析了直接扩展GAN时出现的失效模式,如生成器早期层利用率低和优化不稳定性。为此,提出轻量级中间监督和宽度感知学习率调整等简单且适应规模的解决方案。实验表明,一种纯Transformer且基于隐空间的GAN——GAT,可在从S到XL的不同容量下稳定训练。其中,GAT-XL/2在仅60轮内即达成ImageNet-256上的最优单步、类别条件生成性能(FID 2.18),训练轮次仅为强基线的1/4。
原文摘要 · Abstract (English)
Scalability has driven recent advances in generative modeling, yet its principles remain underexplored for adversarial learning. We investigate the scalability of Generative Adversarial Networks (GANs) through two design choices that have proven to be effective in other types of generative models: training in a compact Variational Autoencoder latent space and adopting purely transformer-based generators and discriminators. Training in latent space enables efficient computation while preserving perceptual fidelity, and this efficiency pairs naturally with plain transformers, whose performance scales with computational budget. Building on these choices, we analyze failure modes that emerge when naively scaling GANs. Specifically, we find issues as underutilization of early layers in the generator and optimization instability as the network scales. Accordingly, we provide simple and scale-friendly solutions as lightweight intermediate supervision and width-aware learning-rate adjustment. Our experiments show that GAT, a purely transformer-based and latent-space GANs, can be easily trained reliably across a wide range of capacities (S through XL). Moreover, GAT-XL/2 achieves state-of-the-art single-step, class-conditional generation performance (FID of 2.18) on ImageNet-256 in just 60 epochs, 4x fewer epochs than strong baselines. Project page: https://hse1032.github.io/GAT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。