AIExplainer

What is Wasserstein loss?

A measure of the difference between two probability distributions

Wasserstein loss is a way to calculate how similar or different two sets of data are, by looking at the distance between them. It's often used in machine learning to train models that can generate new data that's similar to existing data.

Imagine you're trying to move a pile of sand from one spot to another. The Wasserstein loss is like measuring the amount of effort it would take to move all the sand from the original pile to the new pile, taking into account the distance and the amount of sand that needs to be moved.

For example, in image generation, Wasserstein loss can be used to train a model to generate new images of faces that are similar to existing images of faces. The model would try to minimize the Wasserstein loss between the generated images and the real images, resulting in more realistic generated images.

Wasserstein loss is used in generative models, such as GANs (Generative Adversarial Networks), to train the model to generate new data that's similar to the training data. It's also used in other applications, such as image and speech processing.

One common misconception is that Wasserstein loss is only used in GANs, when in fact it can be used in other types of models as well. Another misconception is that it's only used for image generation, when in fact it can be used for other types of data, such as speech or text.

The Wasserstein loss is named after the Russian mathematician Leonid Vaseršteĭn, who first introduced the concept in the 1960s. It has since been widely adopted in the field of machine learning.

Earth Mover's Distance Kantorovich-Rubinstein metric

Three products for different needs — explore what’s relevant to you.