arXiv reaDer
Generative Adversarial Network の共役勾配法
Conjugate Gradient Method for Generative Adversarial Networks
生成モデルのトレーニング戦略の 1 つは、モデル分布とデータ分布の間の Jensen--Shannon 発散を最小化することです。データの分布は不明であるため、敵対的生成ネットワーク (GAN) は、この問題をジェネレーターとディスクリミネーターの 2 つのモデル間のゲームとして定式化します。トレーニングは、ゲーム理論とローカル ナッシュ均衡 (LNE) のコンテキストで定式化できます。既存の方法の安定性または最適性の保証を導き出すことは実行可能ではないようです。この最適化問題は、単一の目的設定よりもはるかに困難です。ここでは、共役勾配法を使用して、GAN の LNE 問題を確実かつ効率的に解決します。提案された方法が、一定の学習率を含む 3 つの異なる学習率更新ルールを使用して LNE に収束することを示す穏やかな仮定の下で、証明と収束の分析を行います。最後に、提案された方法が、最高のフレシェ開始距離 (FID) スコアの点で確率的勾配降下法 (SGD) および運動量 SGD よりも優れており、平均して Adam よりも優れていることを示します。コードは https://github.com/Hiroki11x/ConjugateGradient_GAN で入手できます。
One of the training strategies of generative models is to minimize the Jensen--Shannon divergence between the model distribution and the data distribution. Since data distribution is unknown, generative adversarial networks (GANs) formulate this problem as a game between two models, a generator and a discriminator. The training can be formulated in the context of game theory and the local Nash equilibrium (LNE). It does not seem feasible to derive guarantees of stability or optimality for the existing methods. This optimization problem is far more challenging than the single objective setting. Here, we use the conjugate gradient method to reliably and efficiently solve the LNE problem in GANs. We give a proof and convergence analysis under mild assumptions showing that the proposed method converges to a LNE with three different learning rate update rules, including a constant learning rate. Finally, we demonstrate that the proposed method outperforms stochastic gradient descent (SGD) and momentum SGD in terms of best Frechet inception distance (FID) score and outperforms Adam on average. The code is available at https://github.com/Hiroki11x/ConjugateGradient_GAN.
updated: Tue Feb 21 2023 03:28:47 GMT+0000 (UTC)
published: Mon Mar 28 2022 04:44:45 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アソシエイト