Gradient Penalty is a regularization technique used primarily in GAN training (WGAN-GP) — penalizing the norm of the discriminator's gradient with respect to its input, enforcing the Lipschitz constraint required by the Wasserstein distance formulation.
How Does Gradient Penalty Work?
- WGAN-GP: $mathcal{L}_{GP} = lambda cdot mathbb{E}_{hat{x}}[(||\nabla_{hat{x}} D(hat{x})||_2 - 1)^2]$
- Interpolation: $hat{x} = alpha x_{real} + (1-alpha) x_{fake}$ with $alpha sim U(0,1)$.
- Target: The gradient norm should be 1 everywhere along interpolation paths.
- Paper: Gulrajani et al., "Improved Training of Wasserstein GANs" (2017).
Why It Matters
- GAN Stability: Replaced weight clipping in WGAN, dramatically improving training stability and sample quality.
- Lipschitz Constraint: Provides a soft, differentiable enforcement of the 1-Lipschitz constraint.
- Widely Adopted: Standard in most modern GAN architectures (StyleGAN, BigGAN, etc.).
Gradient Penalty is the smoothness enforcer for GANs — ensuring the discriminator function changes gradually, preventing the adversarial training from becoming unstable.
gradient penaltygenerative models
Related Topics
Explore 500+ Semiconductor & AI Topics
From EUV lithography to CUDA optimization — search the full knowledge base or chat with our AI assistant.