Distribution Fitting for Combating Mode Collapse in GANs

Yanxiang Gong,Zhiwei Xie,Guozhen Duan,Zheng Ma,Mei Xie
DOI: https://doi.org/10.48550/arXiv.2212.01521
IF: 5.414
2022-12-03
Machine Learning
Abstract:Mode collapse is still a major unsolved problem in generative adversarial networks. In this work, we analyze the causes of mode collapse from a new perspective. Due to the nonuniform sampling in the training process, some sub-distributions can be missed while sampling data. Therefore, the GAN objective can reach the minimum when the generated distribution is not the same as the real one. To alleviate the problem, we propose a global distribution fitting (GDF) method by a penalty term to constrain generated data distribution. On the basis of not changing the global minimum of the GAN objective, GDF will make it harder to reach the minimum value when the generated distribution is not the same as the real one. Furthermore, we also propose a local distribution fitting (LDF) method to cope with the situation that the real distribution is unknown. Experiments on several benchmarks demonstrate the effectiveness and competitive performance of GDF and LDF.
What problem does this paper attempt to address?