基于权重共享的生成对抗网络进化架构搜索

Evolutionary Architecture Search for Generative Adversarial Networks Based on Weight Sharing

IEEE Transactions on Evolutionary Computation · 2023
被引 49
ABS 4

中文导读

提出EWSGAN方法,通过权重共享训练生成器超网,再用多目标进化算法搜索最优子网,在CIFAR-10和STL-10上取得领先性能,解决了GAN训练不稳定和架构设计依赖专家经验的问题。

Abstract

Generative adversarial networks (GANs) are a powerful generative technique but frequently face challenges with training stability. Network architecture plays a significant role in determining the final output of GANs, but designing a fine architecture demands extensive domain expertise. This paper aims to address this issue by searching for high-performance generator’s architectures through neural architecture search (NAS). The proposed approach, called evolutionary weight sharing generative adversarial networks (EWSGAN), is based on weight sharing and comprises two steps. First, a supernet of the generator is trained using weight sharing. Second, a multi-objective evolutionary algorithm (MOEA) is employed to identify optimal subnets from the supernet. These subnets inherit weights directly from the supernet for fitness assessment. Two strategies are used to stabilise the training of the generator supernet: a fair single-path sampling strategy and a discarding strategy. Experimental results indicate that the architecture searched by our method achieved a new state-of-the-art among NAS-GAN methods with a Fréchet inception distance (FID) of 9.09 and an inception score (IS) of 8.99 on the CIFAR-10 dataset. It also demonstrates competitive performance on the STL-10 dataset, achieving FID of 21.89 and IS of 10.51.

生成对抗网络神经架构搜索进化算法计算机视觉