← 返回资源分享
GANs Trained by a Two Time-Scale Update Rule Converge to a Local Nash Equilibrium
论文
论文
发布时间2017-06-26
发表NeurIPS 2017 12 · arXiv:1706.08500
作者:Martin Heusel,Hubert Ramsauer,Thomas Unterthiner,Bernhard Nessler,Sepp Hochreiter
详细介绍
Generative Adversarial Networks (GANs) excel at creating realistic images
with complex models for which maximum likelihood is infeasible. However, the
convergence of GAN training has still not been proved. We propose a two
time-scale update rule (TTUR) for training GANs with stochastic gradient
descent on arbitrary GAN loss functions. TTUR has an individual learning rate
for both the discriminator and the generator. Using the theory of stochastic
approximation, we prove that the TTUR converges under mild assumptions to a
stationary local Nash equilibrium. The convergence carries over to the popular
Adam optimization, for which we prove that it follows the dynamics of a heavy
ball with friction and thus prefers flat minima in the objective landscape. For
the evaluation of the performance of GANs at image generation, we introduce the
"Fr\'echet Inception Distance" (FID) which captures the similarity of generated
images to real ones better than the Inception Score. In experiments, TTUR
improves learning for DCGANs and Improved Wasserstein GANs (WGAN-GP)
outperforming conventional GAN training on CelebA, CIFAR-10, SVHN, LSUN
Bedrooms, and the One Billion Word Benchmark.
代码仓库 (69)
bioinf-jku/TTUR官方TensorFlow
mbinkowski/DeepSpeechDistances官方TensorFlow
DevashishJoshi/Transferring-GANs-FYPTensorFlow
w86763777/pytorch-image-generation-metricsPyTorch
yenchenlin/fidPyTorch
djl11/mxnet-fidMXNet
MicroprocessorX069/Comparison-of-DC-GANS-and-SA-GANSPyTorch
vict0rsch/pytorch-fid-wrapperPyTorch
milmor/latent-diffusion-transformerTensorFlow
ShwanMario/ASWDPyTorch
