Wasserstein GAN
In plain wordsA version of the forger-versus-detective image generator that measures how far its fakes are from real data in a smoother way, making training steadier and easier to monitor.
A variant of the generative adversarial networkgenerative adversarial network, n.: Two neural networks trained as rivals: one forges fake images, the other tries to spot the fakes. As both improve, the forger's output becomes realistic enough to fool people. that trains the generator to minimize an estimate of the Wasserstein distanceWasserstein distance, n.: A way of comparing two spreads of probability by imagining one as heaps of sand that must be shoveled into the shape of the other: the less shoveling, weighted by distance, the more alike they are. (earth mover's distance) between the data and model distributions instead of the Jensen–Shannon-type divergenceJensen–Shannon divergence, n.: A balanced, two-way measure of how different two sets of probabilities are, comparing each with their average. Unlike its better-known cousin, it gives the same answer in both directions. implicit in the original GAN. The discriminator becomes a 'critic' that outputs an unbounded real-valued score; by Kantorovich–Rubinstein duality, the gap between its mean scores on real and generated samples estimates the distance, provided the critic is 1-Lipschitz. Because this distance stays informative even when the two distributions barely overlap, the generator receives useful gradients throughout training, and the critic's loss tracks sample quality. The original paper enforced the Lipschitz constraint by clipping weights; WGAN-GP replaced clipping with a penalty that keeps the norm of the critic's input gradient near one, and spectral normalization later offered another route.
Origin
Proposed by Martin Arjovsky, Soumith Chintala and Léon Bottou in a paper posted to arXiv in January 2017 and published at ICML 2017. Ishaan Gulrajani and colleagues introduced the gradient-penalty version (WGAN-GP) two months later.
Related
- generative adversarial networkgenerative adversarial network, n.: Two neural networks trained as rivals: one forges fake images, the other tries to spot the fakes. As both improve, the forger's output becomes realistic enough to fool people.
- Wasserstein distanceWasserstein distance, n.: A way of comparing two spreads of probability by imagining one as heaps of sand that must be shoveled into the shape of the other: the less shoveling, weighted by distance, the more alike they are.
- mode collapsemode collapse, n.: When a generator gets stuck producing only a few kinds of output — the same face, the same joke — instead of the full variety it was meant to learn.
- Jensen–Shannon divergenceJensen–Shannon divergence, n.: A balanced, two-way measure of how different two sets of probabilities are, comparing each with their average. Unlike its better-known cousin, it gives the same answer in both directions.
- Kullback–Leibler divergenceKullback–Leibler divergence, n.: A score for the cost of believing the wrong odds: on average, how much more surprised you will be by real outcomes than someone who knew the true probabilities. It is zero only when your odds are exactly right.
Sources
- Arjovsky, M., Chintala, S. & Bottou, L. (2017). Wasserstein GAN. arXiv:1701.07875 (ICML 2017). arxiv.org/abs/1701.07875
- Gulrajani, I., Ahmed, F., Arjovsky, M., Dumoulin, V. & Courville, A. (2017). Improved Training of Wasserstein GANs. arXiv:1704.00028 (NeurIPS 2017). arxiv.org/abs/1704.00028
Mentioned in
generative adversarial networkgenerative adversarial network, n.: Two neural networks trained as rivals: one forges fake images, the other tries to spot the fakes. As both improve, the forger's output becomes realistic enough to fool people., mode collapsemode collapse, n.: When a generator gets stuck producing only a few kinds of output — the same face, the same joke — instead of the full variety it was meant to learn.