On the regularization of Wasserstein GANs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Petzka, Henning, Fischer, Asja, Lukovnikov, Denis
Format: Preprint
Published: 2017
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910270890180608
author Petzka, Henning
Fischer, Asja
Lukovnikov, Denis
author_facet Petzka, Henning
Fischer, Asja
Lukovnikov, Denis
contents Since their invention, generative adversarial networks (GANs) have become a popular approach for learning to model a distribution of real (unlabeled) data. Convergence problems during training are overcome by Wasserstein GANs which minimize the distance between the model and the empirical distribution in terms of a different metric, but thereby introduce a Lipschitz constraint into the optimization problem. A simple way to enforce the Lipschitz constraint on the class of functions, which can be modeled by the neural network, is weight clipping. It was proposed that training can be improved by instead augmenting the loss by a regularization term that penalizes the deviation of the gradient of the critic (as a function of the network's input) from one. We present theoretical arguments why using a weaker regularization term enforcing the Lipschitz constraint is preferable. These arguments are supported by experimental results on toy data sets.
format Preprint
id arxiv_https___arxiv_org_abs_1709_08894
institution arXiv
publishDate 2017
record_format arxiv
spellingShingle On the regularization of Wasserstein GANs
Petzka, Henning
Fischer, Asja
Lukovnikov, Denis
Machine Learning
Since their invention, generative adversarial networks (GANs) have become a popular approach for learning to model a distribution of real (unlabeled) data. Convergence problems during training are overcome by Wasserstein GANs which minimize the distance between the model and the empirical distribution in terms of a different metric, but thereby introduce a Lipschitz constraint into the optimization problem. A simple way to enforce the Lipschitz constraint on the class of functions, which can be modeled by the neural network, is weight clipping. It was proposed that training can be improved by instead augmenting the loss by a regularization term that penalizes the deviation of the gradient of the critic (as a function of the network's input) from one. We present theoretical arguments why using a weaker regularization term enforcing the Lipschitz constraint is preferable. These arguments are supported by experimental results on toy data sets.
title On the regularization of Wasserstein GANs
topic Machine Learning
url https://arxiv.org/abs/1709.08894