arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.06723cs.LGcs.AI

On Duality Gap as a Measure for Monitoring GAN Training

Indian Institute of Technology Ropar(印度理工学院罗巴尔分校)

Sahil Sidheekh, Aroof Aimen, Vineet Madan, Narayanan C. Krishnan

英文摘要:

Generative adversarial network (GAN) is among the most popular deep learning models for learning complex data distributions. However, training a GAN is known to be a challenging task. This is often attributed to the lack of correlation between the training progress and the trajectory of the generator and discriminator losses and the need for the GAN's subjective evaluation. A recently proposed measure inspired by game theory - the duality gap, aims to bridge this gap. However, as we demonstrate, the duality gap's capability remains constrained due to limitations posed by its estimation process. This paper presents a theoretical understanding of this limitation and proposes a more dependable estimation process for the duality gap. At the crux of our approach is the idea that local perturbations can help agents in a zero-sum game escape non-Nash saddle points efficiently. Through exhaustive experimentation across GAN models and datasets, we establish the efficacy of our approach in capturing the GAN training progress with minimal increase to the computational complexity. Further, we show that our estimate, with its ability to identify model convergence/divergence, is a potential performance measure that can be used to tune the hyperparameters of a GAN.

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

↑