Publication Type
Conference Proceeding Article
Version
acceptedVersion
Publication Date
12-2020
Abstract
Despite success on a wide range of problems related to vision, generative adversarial networks (GANs) often suffer from inferior performance due to unstable training, especially for text generation. To solve this issue, we propose a new variational GAN training framework which enjoys superior training stability. Our approach is inspired by a connection of GANs and reinforcement learning under a variational perspective. The connection leads to (1) probability ratio clipping that regularizes generator training to prevent excessively large updates, and (2) a sample re-weighting mechanism that improves discriminator training by downplaying bad-quality fake samples. Moreover, our variational GAN framework can provably overcome the training issue in many GANs that an optimal discriminator cannot provide any informative gradient to training generator. By plugging the training approach in diverse state-of-the-art GAN architectures, we obtain significantly improved performance over a range of tasks, including text generation, text style transfer, and image generation.
Discipline
Graphics and Human Computer Interfaces
Research Areas
Intelligent Systems and Optimization
Areas of Excellence
Digital transformation
Publication
Proceedings of the 34th Conference on Neural Information Processing Systems, Virtual Conference, 2020 December 6-12
First Page
1
Last Page
12
Publisher
NeurIPS
City or Country
Virtual Conference
Citation
WU, Yue; ZHOU, Pan; GORDON, Andrew Wilson; XING, Eric; and HU, Zhiting.
Improving GAN training with probability ratio clipping and sample reweighting. (2020). Proceedings of the 34th Conference on Neural Information Processing Systems, Virtual Conference, 2020 December 6-12. 1-12.
Available at: https://ink.library.smu.edu.sg/sis_research/8996
Creative Commons License
This work is licensed under a Creative Commons Attribution-NonCommercial-No Derivative Works 4.0 International License.
Additional URL
https://proceedings.neurips.cc/paper_files/paper/2020/hash/3eb46aa5d93b7a5939616af91addfa88-Abstract.html