arXiv reaDer
生成的敵対的ネットワークを1つの段階でトレーニングする
Training Generative Adversarial Networks in One Stage
生成的敵対的ネットワーク(GAN)は、さまざまな画像生成タスクで前例のない成功を収めています。ただし、有望な結果は、ジェネレータとディスクリミネータが2段階で交互に更新される面倒なトレーニングプロセスを犠牲にしてもたらされます。この論文では、GANを1つの段階で効率的にトレーニングできる一般的なトレーニングスキームを調査します。ジェネレーターとディスクリミネーターの敵対的損失に基づいて、GANを対称GANと非対称GANの2つのクラスに分類し、2つを統合する新しい勾配分解法を導入して、両方のクラスを1つのステージでトレーニングできるようにし、トレーニングを軽減します。努力。いくつかのデータセットとさまざまなネットワークアーキテクチャでの計算分析と実験結果は、提案された1段階のトレーニングスキームが、ジェネレータとディスクリミネータのネットワークアーキテクチャに関係なく、従来のトレーニングスキームよりも確実な1.5倍の加速をもたらすことを示しています。さらに、提案された方法は、データフリーの知識蒸留など、他の敵対的訓練シナリオに容易に適用できることを示します。ソースコードはまもなく公開されます。
Generative Adversarial Networks (GANs) have demonstrated unprecedented success in various image generation tasks. The encouraging results, however, come at the price of a cumbersome training process, during which the generator and discriminator are alternately updated in two stages. In this paper, we investigate a general training scheme that enables training GANs efficiently in only one stage. Based on the adversarial losses of the generator and discriminator, we categorize GANs into two classes, Symmetric GANs and Asymmetric GANs, and introduce a novel gradient decomposition method to unify the two, allowing us to train both classes in one stage and hence alleviate the training effort. Computational analysis and experimental results on several datasets and various network architectures demonstrate that, the proposed one-stage training scheme yields a solid 1.5× acceleration over conventional training schemes, regardless of the network architectures of the generator and discriminator. Furthermore, we show that the proposed method is readily applicable to other adversarial-training scenarios, such as data-free knowledge distillation. Our source code will be published soon.
updated: Sun Feb 28 2021 09:03:39 GMT+0000 (UTC)
published: Sun Feb 28 2021 09:03:39 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アソシエイト