Scalable GANs with Transformers

GenAI GANs ML SSLRO Transformer
可扩展性推动了生成模型领域的近期进展,但其原则在对抗学习中的应用仍缺乏深入探索。我们通过两种在其他类型生成模型中已被证明有效的设计选择,来研究生成对抗网络(GANs)的可扩展性:一是在紧凑的变分自编码器(VAE)潜在空间中进行训练,二是采用完全基于Transformer结构的生成器和判别器。在潜在空间中训练能够实现高效的计算,同时保持良好的感知质量;而这种高效率与纯Transformer架构天然契合,因为后者性能随计算资源的增加而提升。基于这些设计,我们分析了在直接扩大GAN规模时出现的各种失败模式。具体而言,我们发现随着网络规模扩大,生成器的早期层利用不足,且优化过程变得不稳定。针对这些问题,我们提出了简单且有利于扩展的解决方案:轻量级的中间监督机制,以及考虑网络宽度的学习率调整策略。实验表明,GAT——一种完全基于Transformer且在潜在空间中运行的GAN——能够在广泛的模型容量(从S到XL)下稳定可靠地训练。此外,GAT-XL/2仅用40个训练周期,就在ImageNet-256上实现了单步、类别条件生成的最先进性能(FID为2.96),训练周期数比强基线方法减少了6倍。
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.96) on ImageNet-256 in just 40 epochs, 6x fewer epochs than strong baselines.
许愿