WGAN-GP stabilizes GAN training by adding a gradient penalty to the discriminator loss. It reduces opportunities for vanishing gradients and helps the generator keep up with the discriminator, addressing mode collapse. section 9.3.3
A training step follows this sequence:
- Compute discriminator scores for real data and generated data without detaching the generated samples. section 9.3.3
Toy grid and handwritten digits
The same training function is applied first to a two-dimensional toy grid and then to 28×28 digit data, with generated samples, discriminator scores, and smoothed losses inspected afterward.
