Timezone: »

Improving GAN Training with Probability Ratio Clipping and Sample Reweighting
Yue Wu · Pan Zhou · Andrew Wilson · Eric Xing · Zhiting Hu

Wed Dec 09 09:00 AM -- 11:00 AM (PST) @ Poster Session 3 #945

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.

Author Information

Yue Wu (Carnegie Mellon University)
Pan Zhou (National University of Singapore)
Andrew Wilson (New York University)

I am a professor of machine learning at New York University.

Eric Xing (Petuum Inc. / Carnegie Mellon University)
Zhiting Hu (UC San Diego)

More from the Same Authors