← Forthcoming posts

Alpha preview · Work in progress

An early version of a forthcoming blog post. The text, proofs, and examples may still evolve.

An Axiomatic Definition of the Wasserstein Distance

A transport plan is a convex decomposition of two distributions into pairs of points.

Suppose a cost \(c(x,y)\) tells us how expensive it is to move one point to another. How should we extend it to compare two probability distributions? Optimal transport answers by introducing a coupling and minimizing an average cost. But there is another way to arrive at the same construction: ask for the largest extension compatible with convexity and stability under weak limits.

This viewpoint makes the transport plan almost inevitable. A coupling records a way to express a pair of distributions as an average of point-pairs, and convexity puts an upper bound on the cost of every such decomposition. The best of those bounds is precisely the Kantorovich value. Savaré and Sodini develop this relaxation approach in a much more general setting [1]; the compact probability version is Theorem 3.35 of my book [2].

The two requirements

Let \(X,Y\) be compact metric spaces and let \(c:X\times Y\to[0,\infty)\) be continuous. Write \(\mathcal P(X)\) for probability measures on \(X\), and \(\delta_x\) for a unit point mass. We seek a functional \(F(\alpha,\beta)\) which does not exceed \(c(x,y)\) on point masses:

\[ F(\delta_x,\delta_y)\le c(x,y). \]

The first requirement is joint convexity. Randomizing between two pairs of distributions should not cost more than randomizing their costs:

\[ \begin{aligned} F((1-t)\alpha_0+t\alpha_1,\, (1-t)\beta_0+t\beta_1) \le (1-t)F(\alpha_0,\beta_0)+tF(\alpha_1,\beta_1). \end{aligned} \]

The second is joint weak lower semicontinuity: if both marginals converge weakly, the limiting cost cannot exceed their limiting lower cost,

\[ F(\alpha,\beta)\le\liminf_{j\to\infty}F(\alpha_j,\beta_j). \]

Weak convergence means convergence of integrals against every continuous test function. It is the natural topology for approximating a distribution by increasingly fine point clouds on a compact space.

These convex combinations mix measures; they do not move particles along Wasserstein geodesics. The theorem concerns convexity in the ordinary linear structure of measures, not displacement convexity.

“Jointly” is essential too. Independent averaging, \(F(\alpha,\beta)=\iint c\,d\alpha\,d\beta\), is affine in each marginal separately. But for \(c(x,y)=|x-y|\) on \(\{0,1\}\), it is zero at both \((\delta_0,\delta_0)\) and \((\delta_1,\delta_1)\), while it is \(1/2\) at their common midpoint. Thus it fails joint convexity. The post on independence and optimal transport examines this coupling from another angle.

The maximal extension theorem

Let \(\Pi(\alpha,\beta)\) denote the joint probability measures on \(X\times Y\) with the given marginals. Define

\[ T_c(\alpha,\beta) =\min_{\pi\in\Pi(\alpha,\beta)}\int c(x,y)\,d\pi(x,y). \]

Compactness ensures that the minimum exists. The product law is always feasible, so \(T_c\) is finite.

Theorem 1 (the largest convex, weakly lower-semicontinuous extension). The functional \(T_c\) is jointly convex, jointly weakly lower semicontinuous, and satisfies \(T_c(\delta_x,\delta_y)=c(x,y)\). If a proper functional \(F:\mathcal P(X)\times\mathcal P(Y)\to\mathbb R\cup\{+\infty\}\) has the same convexity and lower-semicontinuity properties and satisfies the Dirac upper bound, then

\[ F(\alpha,\beta)\le T_c(\alpha,\beta) \quad\text{for every }\alpha,\beta. \]
Proof: mix plans, then approximate any plan by finitely many point-pairs

Step 1: the transport value belongs to the class. If \(\pi_0,\pi_1\) are optimal plans, then \((1-t)\pi_0+t\pi_1\) couples the corresponding mixtures of marginals. Its cost is the mixture of the two optimal costs. Taking the minimum proves joint convexity.

For lower semicontinuity, let \((\alpha_j,\beta_j)\) converge weakly. Choose a subsequence along which the optimal values realize their lower limit, and optimal plans \(\pi_j\). Compactness of \(\mathcal P(X\times Y)\) gives a further convergent subsequence. The limit \(\pi\) has marginals \(\alpha,\beta\), and continuity of \(c\) gives

\[ T_c(\alpha,\beta)\le\int c\,d\pi =\lim_j T_c(\alpha_j,\beta_j). \]

There is only one coupling between two Dirac masses, namely \(\delta_{(x,y)}\), proving the pointwise equality.

Step 2: every admissible extension is below it. Fix any \(\pi\in\Pi(\alpha,\beta)\). Partition the compact product into finitely many measurable cells of diameter tending to zero, and place each cell's mass at one representative point. This gives weakly converging finite approximations

\[ \pi_j=\sum_k w_{j,k}\delta_{(x_{j,k},y_{j,k})}. \]

Their marginals \(\alpha_j,\beta_j\) converge to \(\alpha,\beta\). Finite convexity and the Dirac bound give

\[ \begin{aligned} F(\alpha_j,\beta_j) &\le\sum_k w_{j,k}F(\delta_{x_{j,k}},\delta_{y_{j,k}})\\ &\le\sum_k w_{j,k}c(x_{j,k},y_{j,k}) =\int c\,d\pi_j. \end{aligned} \]

Lower semicontinuity of \(F\) and continuity of the cost integral imply \(F(\alpha,\beta)\le\int c\,d\pi\). Minimize over \(\pi\). \(\square\)

The central identity is the barycentric representation

\[ (\alpha,\beta)=\int_{X\times Y}(\delta_x,\delta_y)\,d\pi(x,y). \]

There are generally many ways to make this decomposition. Convexity bounds \(F\) by the average pointwise cost of each one, so it bounds \(F\) by the smallest of them. Lower semicontinuity is what lets finite decompositions control general distributions.

Without that last assumption, the conclusion is false. On \(X=Y=[0,1]\), set \(F=T_c\) when both marginals have finite support, and \(F=+\infty\) otherwise. This functional is convex and agrees with the cost on every Dirac pair, but it assigns infinite value to diffuse distributions. Weak lower semicontinuity excludes precisely this kind of jump from finite clouds to their limit.

A complete example on two points

Take \(X=Y=\{0,1\}\), with cost \(c(x,y)=|x-y|\). The squared cost is the same on this two-point space. Write

\[ \alpha_a=(1-a)\delta_0+a\delta_1, \qquad \beta_b=(1-b)\delta_0+b\delta_1. \]

Proposition 1 (the convex roof over a square). For \(a,b\in[0,1]\),

\[ T_c(\alpha_a,\beta_b)=|a-b|. \]
Proof: only the imbalance must cross

Any coupling satisfies \(\pi_{10}-\pi_{01}=a-b\). Its cost is \(\pi_{10}+\pi_{01}\), hence at least \(|a-b|\). If \(a\ge b\), choose

\[ \pi_{00}=1-a,\quad \pi_{11}=b,\quad \pi_{10}=a-b,\quad \pi_{01}=0. \]

This has the prescribed marginals and cost \(a-b\). The other case is symmetric. \(\square\)

The four Dirac pairs are the corners of the square of parameters \((a,b)\). The equal-point corners have cost \(0\), and the other two have cost \(1\). The V-shaped surface \(|a-b|\) is the largest convex function lying below those four heights.

This also illustrates why one should state the theorem for the transport cost. On these two points \(W_2^2=|a-b|\), whereas \(W_2=\sqrt{|a-b|}\) is not convex in the marginals: along \(b=0\), it is a strictly concave function of \(a\).

Why “largest” is essential

Joint convexity, weak lower semicontinuity, and agreement on Dirac masses do not uniquely determine a functional. On compact Euclidean supports, choose \(\varepsilon>0\) and let

\[ F_\varepsilon(\alpha,\beta) =T_c(\alpha,\beta) -\varepsilon\bigl(\operatorname{Var}(\alpha)+\operatorname{Var}(\beta)\bigr), \]

where \(\operatorname{Var}(\alpha)=\int|x|^2\,d\alpha-|\int x\,d\alpha|^2\). Variance is concave in a law, so its negative is convex. It is also weakly continuous on compact supports. Thus \(F_\varepsilon\) is another jointly convex lower-semicontinuous functional with the same values on every Dirac pair, and it is smaller wherever the subtracted variance is positive.

Why subtracting variance preserves convexity

Let \(m_i=\int x\,d\alpha_i(x)\). Expanding the square of the mixed mean gives

\[ \begin{aligned} \operatorname{Var}((1-t)\alpha_0+t\alpha_1) ={}&(1-t)\operatorname{Var}(\alpha_0) +t\operatorname{Var}(\alpha_1)\\ &+t(1-t)|m_0-m_1|^2. \end{aligned} \]

The extra term is nonnegative, proving concavity. On a fixed compact support, both \(x\) and \(|x|^2\) are bounded continuous test functions, so the mean and variance vary continuously under weak convergence. Subtracting the variance therefore preserves both convexity and lower semicontinuity of \(T_c\).

If one also wants nonnegative values, use \(\max(F_\varepsilon,0)\). The maximum of two convex lower-semicontinuous functions has the same properties, and the Dirac values remain unchanged because \(c\ge0\).

Couplings as certificates of convex decomposition

Assign cost \(c(x,y)\) to the pairs \((\delta_x,\delta_y)\), and \(+\infty\) to every other pair of laws. Taking the convex lower-semicontinuous envelope of this function produces \(T_c\). On finite spaces, this is the familiar operation of taking the lower convex roof of finitely many points. On general compact spaces, weak limits complete that construction.

The coupling is therefore more than an auxiliary variable introduced to relax a matching constraint. It is the certificate of a convex decomposition. Its marginal constraints say exactly that the decomposition averages to the two distributions we started with. This is one reason linear programming appears so naturally in optimal transport [3].

The dual potentials are the supporting affine functions

This picture also explains Kantorovich duality. A continuous affine function of the two marginals has the form

\[ L_{u,v}(\alpha,\beta)=\int u\,d\alpha+\int v\,d\beta, \]

with any constant absorbed into \(u\). It lies below the pointwise cost exactly when

\[ u(x)+v(y)\le c(x,y)\quad\text{for all }x,y. \]

The standard supporting-hyperplane theorem for closed convex functions then identifies the convex envelope with the supremum of these affine minorants:

\[ T_c(\alpha,\beta) =\sup_{\substack{u\in C(X),\,v\in C(Y)\\u(x)+v(y)\le c(x,y)}} \left(\int u\,d\alpha+\int v\,d\beta\right). \]
Why the envelope gives exactly this family of dual functions

Embed the probability measures in the vector spaces of finite signed measures, with the weak topologies induced by continuous test functions. Extend \(T_c\) by \(+\infty\) outside \(\mathcal P(X)\times\mathcal P(Y)\). Its epigraph is closed and convex by Theorem 1. The supporting-hyperplane representation expresses it as the supremum of its continuous affine minorants.

Continuous linear functionals for these weak topologies are integrals against \(u\in C(X)\) and \(v\in C(Y)\). On probability measures, an affine constant can be absorbed into \(u\). A minorant must obey the Dirac inequalities \(u(x)+v(y)\le c(x,y)\). Conversely, every such pair is a minorant, because integrating this inequality against any coupling gives \(L_{u,v}(\alpha,\beta)\le\int c\,d\pi\), and hence \(L_{u,v}\le T_c\).

Thus the supporting affine functions are exactly the admissible dual potentials. This argument invokes the general supporting-hyperplane theorem; the elementary maximal-extension proof above does not need it.

Savaré and Sodini use this perspective to obtain duality for more general spaces and lower-semicontinuous costs. Their formulation uses sublinearity on finite nonnegative measures; our compact probability version removes that extra machinery while preserving the central idea [1]. For the classical theory of transport duality, see also [4].

References

  1. Giuseppe Savaré and Giacomo Enrico Sodini. A Simple Relaxation Approach to Duality for Optimal Transport Problems in Completely Regular Spaces. Journal of Convex Analysis 29(1), 1–12, 2022, Theorem 4.4.
  2. Gabriel Peyré. Optimal Transport for Machine Learners, 2025–2026. Theorem 3.35, p. 59.
  3. Gabriel Peyré and Marco Cuturi. Computational Optimal Transport. Foundations and Trends in Machine Learning 11(5–6), 355–607, 2019.
  4. Cédric Villani. Topics in Optimal Transportation. Graduate Studies in Mathematics 58, American Mathematical Society, 2003.