English

On gradient regularizers for MMD GANs

Machine Learning 2021-01-15 v5 Machine Learning

Abstract

We propose a principled method for gradient-based regularization of the critic of GAN-like models trained by adversarially optimizing the kernel of a Maximum Mean Discrepancy (MMD). We show that controlling the gradient of the critic is vital to having a sensible loss function, and devise a method to enforce exact, analytical gradient constraints at no additional cost compared to existing approximate techniques based on additive regularizers. The new loss function is provably continuous, and experiments show that it stabilizes and accelerates training, giving image generation models that outperform state-of-the art methods on 160×160160 \times 160 CelebA and 64×6464 \times 64 unconditional ImageNet.

Keywords

Cite

@article{arxiv.1805.11565,
  title  = {On gradient regularizers for MMD GANs},
  author = {Michael Arbel and Danica J. Sutherland and Mikołaj Bińkowski and Arthur Gretton},
  journal= {arXiv preprint arXiv:1805.11565},
  year   = {2021}
}

Comments

Code at https://github.com/MichaelArbel/Scaled-MMD-GAN

R2 v1 2026-06-23T02:12:14.972Z