生成的敵対的ネットワーク(GAN)は、画像生成タスクで大成功を収めていることが証明されていますが、GANトレーニングには不安定性の問題があります。多くの作業により、GANアーキテクチャを手動で変更することにより、GANトレーニングの安定性が向上しました。これには、人間の専門知識と広範な試行錯誤が必要です。このように、モデル設計の自動化を目的としたニューラルアーキテクチャ検索(NAS)は、無条件の画像生成のタスクでGANを検索するために適用されています。初期のNAS-GANは、難易度を下げるために検索ジェネレーターのみを機能させます。最近のいくつかの研究では、GANのパフォーマンスを向上させるために、ジェネレーター(G)とディスクリミネーター(D)の両方を検索しようとしましたが、検索中のGANトレーニングが不安定であるという問題があります。不安定性の問題を軽減するために、EAGANと呼ばれるGANを検出するための効率的な2段階進化的アルゴリズム(EA)ベースのNASフレームワークを提案します。具体的には、GとDの検索を2つの段階に分離し、GANトレーニングの安定性を向上させるための重みリセット戦略を提案します。さらに、進化操作を実行して、複数の目的に基づいてパレートフロントアーキテクチャを作成し、GとDの優れた組み合わせを実現します。重み共有戦略と忠実度の低い評価を活用することで、EAGANは検索時間を大幅に短縮できます。 EAGANは、CIFAR-10(IS = 8.81±0.10、FID = 9.91)で非常に競争力のある結果を達成し、STL-10データセット(IS = 10.44±0.087、FID = 22.18)で以前のNAS検索GANを上回っています。
Generative Adversarial Networks (GANs) have been proven hugely successful in image generation tasks, but GAN training has the problem of instability. Many works have improved the stability of GAN training by manually modifying the GAN architecture, which requires human expertise and extensive trial-and-error. Thus, neural architecture search (NAS), which aims to automate the model design, has been applied to search GANs on the task of unconditional image generation. The early NAS-GAN works only search generators for reducing the difficulty. Some recent works have attempted to search both generator (G) and discriminator (D) to improve GAN performance, but they still suffer from the instability of GAN training during the search. To alleviate the instability issue, we propose an efficient two-stage evolutionary algorithm (EA) based NAS framework to discover GANs, dubbed EAGAN. Specifically, we decouple the search of G and D into two stages and propose the weight-resetting strategy to improve the stability of GAN training. Besides, we perform evolution operations to produce the Pareto-front architectures based on multiple objectives, resulting in a superior combination of G and D. By leveraging the weight-sharing strategy and low-fidelity evaluation, EAGAN can significantly shorten the search time. EAGAN achieves highly competitive results on the CIFAR-10 (IS=8.81±0.10, FID=9.91) and surpasses previous NAS-searched GANs on the STL-10 dataset (IS=10.44±0.087, FID=22.18).