Title

EfficientAutoGAN: Predicting the Rewards in Reinforcement-Based Neural Architecture Search for Generative Adversarial Networks

Publication Name

IEEE Transactions on Cognitive and Developmental Systems

Abstract

This article 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 (GANs), 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 subnetworks 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 subnetworks 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.

Open Access Status

This publication is not available as open access

Volume

14

Issue

1

First Page

234

Last Page

245

Share

COinS
 

Link to publisher version (DOI)

http://dx.doi.org/10.1109/TCDS.2020.3040796