English

AGAN: Towards Automated Design of Generative Adversarial Networks

Machine Learning 2019-06-27 v1 Artificial Intelligence Machine Learning

Abstract

Recent progress in Generative Adversarial Networks (GANs) has shown promising signs of improving GAN training via architectural change. Despite some early success, at present the design of GAN architectures requires human expertise, laborious trial-and-error testings, and often draws inspiration from its image classification counterpart. In the current paper, we present the first neural architecture search algorithm, automated neural architecture search for deep generative models, or AGAN for abbreviation, that is specifically suited for GAN training. For unsupervised image generation tasks on CIFAR-10, our algorithm finds architecture that outperforms state-of-the-art models under same regularization techniques. For supervised tasks, the automatically searched architectures also achieve highly competitive performance, outperforming best human-invented architectures at resolution 32×3232\times32. Moreover, we empirically demonstrate that the modules learned by AGAN are transferable to other image generation tasks such as STL-10.

Keywords

Cite

@article{arxiv.1906.11080,
  title  = {AGAN: Towards Automated Design of Generative Adversarial Networks},
  author = {Hanchao Wang and Jun Huan},
  journal= {arXiv preprint arXiv:1906.11080},
  year   = {2019}
}