Earth Mover's Distance

In statistics, the earth mover's distance (EMD) is a measure of the distance between two probability distributions over a region ''D''. In mathematics, this is known as the Wasserstein metric. Informally, if the distributions are interpreted as two different ways of piling up a certain amount of earth (dirt) over the region ''D'', the EMD is the minimum cost of turning one pile into the other; where the cost is assumed to be the amount of dirt moved times the distance by which it is moved.

The above definition is valid only if the two distributions have the same integral (informally, if the two piles have the same amount of dirt), as in normalized histograms or probability density functions. In that case, the EMD is equivalent to the 1st Mallows distance or 1st Wasserstein distance between the two distributions.

Theory

Assume that we have a set of points in $\mathbb^d$ (dimension $d$). Instead of assigning one distribution to the set of points, we can cluster them and represent the point set in terms of the clusters. Thus, each cluster is a single point in $\mathbb^d$ and the weight of the cluster is decided by the fraction of the distribution present in that cluster. This representation of a distribution by a set of clusters is called the ''signature''. Two signatures can have different sizes, for example, a bimodal distribution has shorter signature (2 clusters) than complex ones. One cluster representation (mean or mode in $\mathbb^d$) can be thought of as a single feature in a signature. The distance between each of the features is called as ''ground distance''.

The Earth Mover's Distance can be formulated and solved as a transportation problem. Suppose that several suppliers, each with a given amount of goods, are required to supply several consumers, each with a given limited capacity. For each supplier-consumer pair, the cost of transporting a single unit of goods is given. The transportation problem is then to find a least-expensive flow of goods from the suppliers to the consumers that satisfies the consumers' demand. Similarly, here the problem is transforming one signature($P$) to another($Q$) with minimum work done.

Assume that signature $P$ has $m$ clusters with $P=\$, where $p_i$ is the cluster representative and $w_ > 0$ is the weight of the cluster. Similarly another signature $Q=\$ has $n$ clusters. Let D= _ be the ground distance between clusters $p_i$ and $q_j$.

We want to find a flow F= _, with $f_$ the flow between $p_i$ and $q_j$, that minimizes the overall cost.

:$\min\limits_F$

subject to the constraints:

:$f_\ge0, 1\le i \le m, 1\le j \le n$

:$\sum_^n \le w_, 1 \le i \le m$

:$\sum_^m \le w_, 1 \le j \le n$

:$\sum_^m\sum_^n f_ = \sum_^m w_ = \sum_^n w_$

The optimal flow $F$ is found by solving this linear optimization problem. The earth mover's distance is defined as the work normalized by the total flow:
:$\text(P,Q) = \frac$

Equivalent Form

Equivalently, we can write:

:\text(P,Q) = \inf\limits_ \mathbb_ x - y\, ,

where $\Pi(P, Q)$ is the set of all joint distributions whose marginals are $P$ and $Q$.

Kantorovich-Rubinstein duality

By Kantorovich-Rubinstein duality, the following is an equivalent expression:

:\text(P,Q) = \sup\limits_ \mathbb_ , f(x), - \mathbb_ , f(x),

that is, to supremum over all 1-Lipschitz continuous functions, i.e. $\, \nabla f(x)\, \leq 1 \quad \forall x$.

Extensions

Some applications may require the comparison of distributions with different total masses. One approach is to allow for a '' partial match'', where dirt from the most massive distribution is rearranged to make the least massive, and any leftover "dirt" is discarded at no cost. Under this approach, the EMD is no longer a true distance between distributions.

Another approach is to allow for mass to be created or destroyed, on a global and/or local level, as an alternative to transportation, but with a cost penalty. In that case one must specify a real parameter σ, the ratio between the cost of creating or destroying one unit of "dirt", and the cost of transporting it by a unit distance. This is equivalent to minimizing the sum of the earth moving cost plus σ times the ''L''¹ distance between the rearranged pile and the second distribution.

Notationally, if $\pi:P\to Q$ is a partial function which is a bijection on subsets $P'\subset P$ and $Q'\subset Q$, then one is interested in the distance function
:$, P-Q, _\pi = , P\setminus P', + , Q\setminus Q',$
where $P\setminus P'$ denotes set minus. Here, $P'$ would be the portion of the earth that was moved; thus $P\setminus P'$ would be the portion not moved, and $, P\setminus P',$ the size of the pile not moved. By symmetry, one contemplates $Q'\subset Q$ as the pile at the destination that 'got there' from ''P'', as compared to the total ''Q'' that we ''want to have there''. Formally, this distance indicates how much an injective correspondence differs from an isomorphism.

The EMD can be extended naturally to the case where more than two distributions are compared. In this case, the "distance" between the many distributions is defined as the optimal value of a linear program. This generalized EMD may be computed exactly using a greedy algorithm, and the resulting functional has been shown to be Minkowski additive and convex monotone.

Computing the EMD

The EMD can be computed by solving an instance of transportation problem, using any algorithm for minimum-cost flow problem, e.g. the network simplex algorithm.

The Hungarian algorithm can be used to get the solution if the domain ''D'' is the set . If the domain is integral, it can be translated for the same algorithm by representing integral bins as multiple binary bins.

As a special case, if ''D'' is a one-dimensional array of "bins" of size ''n'', the EMD can be efficiently computed by scanning the array and keeping track of how much dirt needs to be transported between consecutive bins. Here the bins are zero-indexed:

: $\begin \text_0 &= 0 \\ \text_ &= P_i + \text_i - Q_i \\ \text &= \sum_^, \text_i, \end$

PyTorch Implementation

import torch

def EMD(p, q):
x = p - q
y = torch.cumsum(x, dim=0)
return y.abs().sum()

EMD-based similarity analysis

EMD-based similarity analysis (EMDSA) is an important and effective tool in many multimedia information retrieval and pattern recognition applications. However, the computational cost of EMD is super-cubic to the number of the "bins" given an arbitrary "D". Efficient and scalable EMD computation techniques for large scale data have been investigated using MapReduce, as well as bulk synchronous parallel and resilient distributed dataset.

Applications

An early application of the EMD in computer science was to compare two grayscale images that may differ due to dithering, blurring, or local deformations. In this case, the region is the image's domain, and the total amount of light (or ink) is the "dirt" to be rearranged.

The EMD is widely used in content-based image retrieval to compute distances between the color histograms of two digital images. In this case, the region is the RGB color cube, and each image pixel is a parcel of "dirt". The same technique can be used for any other quantitative pixel attribute, such as luminance, gradient, apparent motion in a video frame, etc..

More generally, the EMD is used in pattern recognition to compare generic summaries or surrogates of data records called signatures. A typical signature consists of list of pairs ((''x''₁,''m''₁), ... (''x''_''n'',''m''_''n'')), where each ''x''_''i'' is a certain "feature" (e.g., color in an image, letter in a text, etc.), and ''m''_''i'' is "mass" (how many times that feature occurs in the record). Alternatively, ''x''_''i'' may be the centroid of a data cluster, and ''m''_''i'' the number of entities in that cluster. To compare two such signatures with the EMD, one must define a distance between features, which is interpreted as the cost of turning a unit mass of one feature into a unit mass of the other. The EMD between two signatures is then the minimum cost of turning one of them into the other.

EMD analysis has been used for quantitating multivariate changes in biomarkers measured by flow cytometry, with potential applications to other technologies that report distributions of measurements.

History

The concept was first introduced by Gaspard Monge in 1781, in the context of transportation theory. The use of the EMD as a distance measure for monochromatic images was described in 1989 by S. Peleg, M. Werman and H. Rom. The name "earth movers' distance" was proposed by J. Stolfi in 1994, and was used in print in 1998 by Y. Rubner, C. Tomasi and L. G. Guibas.

References

{{Reflist

External links

C code for the Earth Mover's Distance
(archive
here

Python implementation with referencesPython2 wrapper for the C implementation of the Earth Mover's DistanceC++ and Matlab and Java wrappers code for the Earth Mover's Distance, especially efficient for thresholded ground distancesJava implementation of a generic generator for evaluating large-scale Earth Mover's Distance based similarity analysis

Statistical distance