About Speakers Schedule INS
2024年人工智能与计算数学会议
Artificial Intelligence and Computational Mathematics Conference

Magnifying the Three Phases of GAN Training — Fitting, Refining and Collapsing

Speaker

高卫国 Weiguo Gao , 复旦大学 Fudan University

Time

17 Mar, 15:15 - 15:45

Abstract

Generative Adversarial Networks (GANs) are efficient generative models but may suffer from mode mixture and mode collapse. We present an original global characterization of GAN training by dividing it into three successive phases — fitting, refining, and collapsing. Such a characterization underscores a strong correlation between mode mixture and the refining phase, as well as mode collapse and the collapsing phase. To analyze the causes and features of each phase, we propose a novel theoretical framework that integrates both continuous and discrete aspects of GANs, addressing a gap in existing literature that predominantly focuses on only one aspect. We develop a specialized metric to detect the phase transition from refining to collapsing and integrate it in an “early stopping” algorithm to optimize GAN training. Experiments on synthetic datasets and real-world datasets including MNIST, Fashion MNIST and CIFAR-10 substantiate our theoretical insights and highlight the efficacy of our algorithm. This is a joint work with Ming Li.