Penalité de gradient refers to a technique utilisé en apprentissage automatique, particularly in the training of Generative Adversarial Networks (GANs) and other models that involve optimization. It acts as a regularization term that helps to stabilize the training process by penalizing the model for large gradients. This is crucial because large gradients can lead to instability and poor convergence during training.
The concept of Gradient Penalty is often implemented in the context of Wasserstein GANs (WGANs). In WGANs, a penalty is added to the fonction de perte based on the norm of the gradients of the critic (a type of discriminator) with respect to its input. Specifically, the gradient penalty encourages the gradients to have a norm close to one, which helps maintain the Continuité Lipschitz requise pour le cadre WGAN.
Mathématiquement, le terme de pénalité de gradient est calculé comme :
GP = λ * E[(||∇D(x)||2 - 1)²]où :
GPis the gradient penalty,λis a weighting factor that controls the strength of the penalty,D(x)is the output of the discriminator for inputx, and∇D(x)represents the gradients of the discriminator.
By adding this penalty term to the loss function, the training of the model becomes more stable, reducing the likelihood of effondrement de mode and improving the quality of generated samples. Overall, Gradient Penalty is a vital technique for enhancing the performance and reliability of various machine learning models.