arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.14320cs.LGcs.ITmath.ITstat.ML

$(α_D,α_G)$-GANs:通过双目标解决GAN训练不稳定性问题

$(α_D,α_G)$-GANs: Addressing GAN Training Instabilities via Dual Objectives

Arizona State University(亚利桑那州立大学);IIIT Hyderabad(印度国际信息技术学院海得拉巴分校)

Monica Welfert, Kyle Otstot, Gowtham R. Kurri, Lalitha Sankar

AI总结:

本研究提出双目标$(α_D,α_G)$-GANs框架,通过为生成器和判别器设置基于$α$-损失的不同目标,在理论上证明其可简化为$f$-散度最小化并推导估计误差上界,实验验证其能有效缓解GAN训练不稳定性。

摘要:

为解决生成对抗网络(GAN)的训练不稳定性问题,我们提出一类双目标GAN,其生成器(G)和判别器(D)采用不同的价值函数(目标)。具体而言,我们使用可调节分类损失$α$-损失对每个目标建模,得到以$(α_D,α_G)\in (0,\infty]^2$为参数的$(α_D,α_G)$-GANs。在样本量及G和D的容量足够大的情况下,我们证明,在$(α_D,α_G)$满足适当条件时,所得非零和博弈可简化为最小化一种$f$-散度。在有限样本和容量设置下,我们定义估计误差来量化生成器性能相对于无限样本最优设置的差距,并推导该误差的上界,证明其在特定条件下达到阶最优。最后,我们在合成二维高斯混合环数据集和Stacked MNIST数据集上验证了调节$(α_D,α_G)$在缓解训练不稳定性方面的价值。

英文摘要:

In an effort to address the training instabilities of GANs, we introduce a class of dual-objective GANs with different value functions (objectives) for the generator (G) and discriminator (D). In particular, we model each objective using $α$-loss, a tunable classification loss, to obtain $(α_D,α_G)$-GANs, parameterized by $(α_D,α_G)\in (0,\infty]^2$. For sufficiently large number of samples and capacities for G and D, we show that the resulting non-zero sum game simplifies to minimizing an $f$-divergence under appropriate conditions on $(α_D,α_G)$. In the finite sample and capacity setting, we define estimation error to quantify the gap in the generator's performance relative to the optimal setting with infinite samples and obtain upper bounds on this error, showing it to be order optimal under certain conditions. Finally, we highlight the value of tuning $(α_D,α_G)$ in alleviating training instabilities for the synthetic 2D Gaussian mixture ring and the Stacked MNIST datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑