We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: On gradient regularizers for MMD GANs

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 \times 160$ CelebA and $64 \times 64$ unconditional ImageNet.
Comments: Code at this https URL
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Journal reference: Advances in Neural Information Processing Systems 31 (NeurIPS 2018), 6700-6710
Cite as: arXiv:1805.11565 [stat.ML]
  (or arXiv:1805.11565v5 [stat.ML] for this version)

Submission history

From: Danica J. Sutherland [view email]
[v1] Tue, 29 May 2018 16:23:56 GMT (5200kb,D)
[v2] Sat, 27 Oct 2018 13:25:28 GMT (8891kb,D)
[v3] Sat, 24 Nov 2018 03:35:54 GMT (8891kb,D)
[v4] Thu, 29 Nov 2018 20:56:44 GMT (8907kb,D)
[v5] Thu, 14 Jan 2021 05:27:48 GMT (8907kb,D)

Link back to: arXiv, form interface, contact.