arXiv reaDer
CoopInit: 協調学習による敵対的生成ネットワークの初期化
CoopInit: Initializing Generative Adversarial Networks via Cooperative Learning
正則化やアーキテクチャ設計など、Generative Adversarial Networks (GAN) のトレーニングを安定させるために、数多くの研究努力がなされてきました。しかし、不安定性は、敵対的学習の初期段階での壊れやすいバランスからも発生する可能性があることを確認しています。このホワイト ペーパーでは、トレーニング中の計算オーバーヘッドが非常に小さく、GAN の適切な開始点をすばやく学習できる、シンプルかつ効果的な協調学習ベースの初期化戦略である CoopInit を提案します。提案されたアルゴリズムは、2 つの学習段階で構成されています。学習勾配を近似する合成データを提供します。 EBM は、MCMC ティーチングを介してジェネレーターの MLE 学習もガイドします。 (ii) 敵対的ファイナライゼーション段階: 初期化を数回繰り返した後、アルゴリズムは、収束するまで通常のミニマックス敵対的トレーニングにシームレスに移行します。動機は、MLE ベースの初期化段階がモデルをモード カバレッジに向けて駆動することです。これは、敵対的学習段階でのモード ドロップの問題を軽減するのに役立ちます。広範な実験を通じて、提案されたアプローチが画像生成と片側の対になっていない画像から画像への変換タスクに有効であることを実証します。
Numerous research efforts have been made to stabilize the training of the Generative Adversarial Networks (GANs), such as through regularization and architecture design. However, we identify the instability can also arise from the fragile balance at the early stage of adversarial learning. This paper proposes the CoopInit, a simple yet effective cooperative learning-based initialization strategy that can quickly learn a good starting point for GANs, with a very small computation overhead during training. The proposed algorithm consists of two learning stages: (i) Cooperative initialization stage: The discriminator of GAN is treated as an energy-based model (EBM) and is optimized via maximum likelihood estimation (MLE), with the help of the GAN's generator to provide synthetic data to approximate the learning gradients. The EBM also guides the MLE learning of the generator via MCMC teaching; (ii) Adversarial finalization stage: After a few iterations of initialization, the algorithm seamlessly transits to the regular mini-max adversarial training until convergence. The motivation is that the MLE-based initialization stage drives the model towards mode coverage, which is helpful in alleviating the issue of mode dropping during the adversarial learning stage. We demonstrate the effectiveness of the proposed approach on image generation and one-sided unpaired image-to-image translation tasks through extensive experiments.
updated: Tue Mar 21 2023 07:49:32 GMT+0000 (UTC)
published: Tue Mar 21 2023 07:49:32 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト