arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.09180cs.CV
收藏

Off-Policy Reinforcement Learning for Efficient and Effective GAN Architecture Search

ETH Zürich(苏黎世联邦理工学院);UESTC(电子科技大学);Navinfo Europe(欧洲四维图新);University College London(伦敦大学学院)

Yuan Tian, Qin Wang, Zhiwu Huang, Wen Li, Dengxin Dai, Minghao Yang, Jun Wang, Olga Fink

英文摘要:

In this paper, we introduce a new reinforcement learning (RL) based neural architecture search (NAS) methodology for effective and efficient generative adversarial network (GAN) architecture search. The key idea is to formulate the GAN architecture search problem as a Markov decision process (MDP) for smoother architecture sampling, which enables a more effective RL-based search algorithm by targeting the potential global optimal architecture. To improve efficiency, we exploit an off-policy GAN architecture search algorithm that makes efficient use of the samples generated by previous policies. Evaluation on two standard benchmark datasets (i.e., CIFAR-10 and STL-10) demonstrates that the proposed method is able to discover highly competitive architectures for generally better image generation results with a considerably reduced computational burden: 7 GPU hours. Our code is available at https://github.com/Yuantian013/E2GAN.

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

↑