Skip to main content
placeholder image

Efficient AutoGAN: Predicting the rewards in reinforcement-based neural architecture search for Generative Adversarial Networks

Journal Article


Abstract


  • This paper is inspired by human’s memory and recognition process to improve Neural Architecture Search (NAS), which has shown novelty and significance in the design of Generative Adversarial Networks (GAN), but the extremely enormous time consumption for searching GAN architectures based on reinforcement learning (RL) limits its applicability to a great extent. The main reason behind the challenge is that, the performance evaluation of sub-networks during the search process takes too much time. To solve this problem, we propose a new algorithm, EfficientAutoGAN, in which a Graph Convolution Network (GCN) predictor is introduced to predict the performance of sub-networks instead of formally assessing or evaluating them. Experiments show that EfficientAutoGAN saves nearly half of the search time and at the same time, demonstrates comparable overall network performance to the state-of-the-art algorithm, AutoGAN, in the field of RL-based NAS for GAN.

UOW Authors


  •   Fan, Yi (external author)
  •   Tang, Xiulian (external author)
  •   Zhou, Guoqiang (external author)
  •   Shen, Jun

Publication Date


  • 2020

Citation


  • Fan, Y., Tang, X., Zhou, G. & Shen, J. (2020). Efficient AutoGAN: Predicting the rewards in reinforcement-based neural architecture search for Generative Adversarial Networks. IEEE Transactions on Cognitive and Developmental Systems, online first 1-13.

Number Of Pages


  • 12

Start Page


  • 1

End Page


  • 13

Volume


  • online first

Place Of Publication


  • United States

Abstract


  • This paper is inspired by human’s memory and recognition process to improve Neural Architecture Search (NAS), which has shown novelty and significance in the design of Generative Adversarial Networks (GAN), but the extremely enormous time consumption for searching GAN architectures based on reinforcement learning (RL) limits its applicability to a great extent. The main reason behind the challenge is that, the performance evaluation of sub-networks during the search process takes too much time. To solve this problem, we propose a new algorithm, EfficientAutoGAN, in which a Graph Convolution Network (GCN) predictor is introduced to predict the performance of sub-networks instead of formally assessing or evaluating them. Experiments show that EfficientAutoGAN saves nearly half of the search time and at the same time, demonstrates comparable overall network performance to the state-of-the-art algorithm, AutoGAN, in the field of RL-based NAS for GAN.

UOW Authors


  •   Fan, Yi (external author)
  •   Tang, Xiulian (external author)
  •   Zhou, Guoqiang (external author)
  •   Shen, Jun

Publication Date


  • 2020

Citation


  • Fan, Y., Tang, X., Zhou, G. & Shen, J. (2020). Efficient AutoGAN: Predicting the rewards in reinforcement-based neural architecture search for Generative Adversarial Networks. IEEE Transactions on Cognitive and Developmental Systems, online first 1-13.

Number Of Pages


  • 12

Start Page


  • 1

End Page


  • 13

Volume


  • online first

Place Of Publication


  • United States