← Forthcoming posts

Alpha preview · Work in progress

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

Weak transport and mean-preserving spreads

Move the conditional means, then let the mass spread for free.

Imagine splitting a particle at the origin into two equal pieces, one sent to \(-1\), the other to \(+1\). Ordinary quadratic optimal transport charges one unit of cost. Yet the average destination has not moved at all.

Barycentric weak transport measures precisely this average displacement. It distinguishes moving a population’s conditional means from spreading mass around those means. This leads to a beautiful connection with convex order and martingales: first transport to a suitable intermediate distribution, then apply a mean-preserving spread. The construction belongs to the theory of weak transport [1]; its projection structure is developed by Gozlan and Juillet [2] and Backhoff-Veraguas, Beiglböck, and Pammer [3].

One coupling, two costs

Let \(\alpha,\beta\in\mathcal P_2(\mathbb R^d)\): probability measures with finite second moments. A coupling \(\pi\in\Pi(\alpha,\beta)\) is a joint distribution of \((X,Y)\) with these marginals. Disintegrate it as \(\pi(dx,dy)=\alpha(dx)\pi_x(dy)\). Its barycentric projection is

\[ T_\pi(x)=\int y\,d\pi_x(y)=\mathbb E[Y\mid X=x]. \]

The two optimization problems are

\[ \begin{aligned} W_2(\alpha,\beta)^2 &=\min_{\pi\in\Pi(\alpha,\beta)}\mathbb E|X-Y|^2,\\ V(\alpha,\beta) &=\min_{\pi\in\Pi(\alpha,\beta)} \mathbb E|X-\mathbb E[Y\mid X]|^2. \end{aligned} \]

The second is the quadratic barycentric weak transport cost. We write \(V\), rather than a distance symbol, because it is generally asymmetric and can vanish for different measures.

Conditional expectation is an orthogonal projection in \(L^2\). Expanding around \(T_\pi(X)\) gives

\[ \mathbb E|X-Y|^2 =\mathbb E|X-T_\pi(X)|^2 +\mathbb E|Y-T_\pi(X)|^2. \]

The cross term vanishes after conditioning on \(X\). Weak transport pays for the first term and ignores the conditional variance in the second. In particular \(0\le V(\alpha,\beta)\le W_2(\alpha,\beta)^2\).

Convex order: a precise notion of being more spread out

Write \(\eta\preceq_{\mathrm{cx}}\beta\) if

\[ \int\varphi\,d\eta\le\int\varphi\,d\beta \quad\text{for every finite convex }\varphi:\mathbb R^d\to\mathbb R. \]

The integrals may equal \(+\infty\); their negative parts are integrable because a finite convex function has an affine lower bound and the laws have finite first moments. Applying the definition to affine functions and their negatives forces the means to agree. Every convex penalty regards \(\beta\) as at least as spread out as \(\eta\).

Strassen’s theorem [5] gives the probabilistic equivalent:

\[ \eta\preceq_{\mathrm{cx}}\beta \quad\Longleftrightarrow\quad \exists(Z,Y):\ Z\sim\eta,\ Y\sim\beta,\ \mathbb E[Y\mid Z]=Z. \]

The pair \((Z,Y)\) is a one-step martingale. The forward-looking interpretation is literal: once \(Z\) is known, the later position \(Y\) has mean \(Z\). Necessity is just conditional Jensen. Sufficiency is the substantive part of Strassen’s theorem.

For finite clouds, sufficiency is a short linear-programming argument

Let \(\eta=\sum_i a_i\delta_{z_i}\) and \(\beta=\sum_j b_j\delta_{y_j}\), with positive weights. A martingale plan is a nonnegative matrix \(P\) satisfying row sums \(a\), column sums \(b\), and \(\sum_jP_{ij}(y_j-z_i)=0\) for every \(i\).

If these linear constraints were infeasible, the finite-dimensional separation theorem (Farkas’ lemma) would provide numbers \(u_i,v_j\) and vectors \(h_i\) with

\[ \begin{aligned} u_i+v_j+h_i\cdot(y_j-z_i)&\ge0 &&\text{for every }i,j,\\ \sum_i a_i u_i+\sum_j b_jv_j&\lt0. \end{aligned} \]

But \(\varphi(y)=\max_i\{-u_i-h_i\cdot(y-z_i)\}\) is convex, \(v_j\ge\varphi(y_j)\), and \(\varphi(z_i)\ge-u_i\). Therefore

\[ \sum_i a_i u_i+\sum_j b_jv_j \ge-\sum_i a_i\varphi(z_i)+\sum_j b_j\varphi(y_j) \ge0, \]

contradicting the strict inequality. Thus convex order ensures a feasible martingale plan. \(\square\)

Equal mean and a larger covariance are necessary for convex order, but are not sufficient for general distributions. For Gaussian laws with the same mean they are sufficient: adding an independent centered Gaussian with the covariance difference constructs the required martingale coupling. Counting target particles, or merely looking at the size of a cloud, is not a test for convex order.

A Wasserstein projection onto a convex-order constraint

Theorem 1 (the weak transport projection). For all \(\alpha,\beta\in\mathcal P_2(\mathbb R^d)\),

\[ V(\alpha,\beta) =\min_{\eta\preceq_{\mathrm{cx}}\beta}W_2(\alpha,\eta)^2. \]

Both minima are attained. The weak optimizer has a unique barycentric map \(T_*\), up to \(\alpha\)-almost-everywhere equality, although its coupling need not be unique. The minimizing intermediate law is unique and equals \(\eta_*=(T_*)_\#\alpha\); here \(T_\#\alpha\) denotes the law of \(T(X)\) when \(X\sim\alpha\). Moreover, \(T_*\) is an ordinary quadratic optimal transport map from \(\alpha\) to \(\eta_*\), even when \(\alpha\) has atoms. In particular,

\[ V(\alpha,\beta)=0 \quad\Longleftrightarrow\quad \alpha\preceq_{\mathrm{cx}}\beta. \]

The geometric picture is: move \(\alpha\) to the closest law \(\eta_*\) that can spread into \(\beta\) without changing conditional means.

Proof: conditional Jensen in one direction, a martingale in the other

From a plan to an intermediate law. For a feasible \((X,Y)\), set \(Z=\mathbb E[Y\mid X]\) and \(\eta=\operatorname{Law}(Z)\). Jensen gives \(\eta\preceq_{\mathrm{cx}}\beta\). Since \((X,Z)\) is a coupling,

\[ \inf_{\eta\preceq_{\mathrm{cx}}\beta}W_2(\alpha,\eta)^2 \le W_2(\alpha,\operatorname{Law}(Z))^2 \le\mathbb E|X-Z|^2. \]

From an intermediate law to a plan. Given \(\eta\preceq_{\mathrm{cx}}\beta\), couple \(X\sim\alpha\) and \(Z\sim\eta\) optimally. By Strassen, there is a kernel from \(Z\) to \(Y\sim\beta\) with conditional mean \(Z\). Generate \(Y\) from this kernel independently of \(X\) conditional on \(Z\). Then \(\mathbb E[Y\mid X]=\mathbb E[Z\mid X]\), so

\[ V(\alpha,\beta) \le\mathbb E|X-\mathbb E[Z\mid X]|^2 \le\mathbb E|X-Z|^2 =W_2(\alpha,\eta)^2. \]

Taking the infimum proves equality of the values.

Attainment: why a minimizing sequence has a limit

The set \(\Pi(\alpha,\beta)\) is weakly compact. The barycentric objective has the representation

\[ \int|x-T_\pi(x)|^2\,d\alpha(x) = \sup_{h\in C_b(\mathbb R^d;\mathbb R^d)} \int\{2h(x)\cdot(x-y)-|h(x)|^2\}\,d\pi. \]

Indeed, this is the usual identity \(\|v\|_2^2=\sup_h(2\langle h,v\rangle-\|h\|_2^2)\) in \(L^2(\alpha)\); bounded continuous vector fields are dense in that space. For fixed \(h\), the integral on the right is continuous on the coupling set: truncate \(x,y\), then use the fixed marginals to control their first-moment tails uniformly. The objective is therefore a supremum of continuous functions, hence lower semicontinuous.

A weak optimizer \(\pi_*\) exists. For \(\eta_*=(T_{\pi_*})_\#\alpha\), the first chain of inequalities becomes

\[ V(\alpha,\beta) \le W_2(\alpha,\eta_*)^2 \le\int|x-T_{\pi_*}(x)|^2\,d\alpha(x) =V(\alpha,\beta). \]

All terms are equal. Thus \(\eta_*\) attains the projection minimum, and its graph coupling with \(\alpha\) is an ordinary OT optimizer.

Uniqueness of the map. If \(\pi_0,\pi_1\) are weak optimizers with barycentric maps \(T_0,T_1\), their average is feasible and has map \((T_0+T_1)/2\). The squared-norm identity gives its cost as

\[ V(\alpha,\beta)-\frac14\int|T_0-T_1|^2\,d\alpha. \]

Minimality forces \(T_0=T_1\) almost everywhere.

Uniqueness of the intermediate law. If \(\eta\) is any minimizing intermediate law, the glued construction above has equality in conditional Jensen. Consequently \(Z=\mathbb E[Z\mid X]=T_*(X)\) almost surely, so \(\eta=(T_*)_\#\alpha\).

Finally, a zero-cost optimizer satisfies \(\mathbb E[Y\mid X]=X\), exactly the martingale condition. Conversely a martingale plan has zero weak cost. Strassen gives the stated equivalence. \(\square\)

The full Brenier–Strassen theorem goes further: \(T_*\) can be represented, \(\alpha\)-almost everywhere, by the gradient of a convex \(C^1\) function whose gradient is \(1\)-Lipschitz [2], [3]. That regularity is a deeper structural result; the projection identity and uniqueness above already follow from conditional expectations and convexity.

Projecting an OT plan is a different operation

One can also start with an ordinary quadratic OT optimizer \(\pi_{\mathrm{OT}}\) and take its barycentric map \(T_{\mathrm{OT}}\). This map solves OT from \(\alpha\) to its own output law \((T_{\mathrm{OT}})_\#\alpha\), but it does not usually solve the weak problem.

Proposition 1 (optimality to the barycentric output). If \(\pi\) minimizes ordinary quadratic transport from \(\alpha\) to \(\beta\), then

\[ W_2(\alpha,(T_\pi)_\#\alpha)^2 =\int|x-T_\pi(x)|^2\,d\alpha(x). \]
Proof: a cheaper first leg would make the original plan cheaper

For \((X,Y)\sim\pi\), put \(Z=T_\pi(X)\). The tower property gives \(\mathbb E[Y\mid Z]=Z\). If a coupling \((X',Z')\) between \(\alpha\) and \(\operatorname{Law}(Z)\) had a smaller quadratic cost, glue it to the conditional law of \(Y\) given \(Z\), independently of \(X'\) conditional on \(Z'\). The resulting \(Y'\) still has law \(\beta\), and

\[ \mathbb E|X'-Y'|^2 =\mathbb E|X'-Z'|^2+\mathbb E|Y-Z|^2. \]

The original cost is \(\mathbb E|X-Z|^2+\mathbb E|Y-Z|^2\). A cheaper first leg would contradict optimality of \(\pi\). \(\square\)

The qualification “ordinary OT optimizer” matters. A general coupling need not have an optimal barycentric map. Also, the output law is usually different from \(\beta\); for a source supported on \(n\) points it has at most \(n\) atoms. See the post on barycentric projection for this distinction.

Here is a concrete separation between the two optimizations. Take

\[ \alpha=\tfrac12(\delta_{-1/2}+\delta_{1/2}), \qquad \beta=\tfrac12(\delta_{-1}+\delta_1). \]

Ordinary OT pairs each point with the endpoint of the same sign. Its barycentric map is \(T_{\mathrm{OT}}(x)=2x\), and its barycentric cost is \(1/4\). But the feasible plan, with rows ordered \(-1/2,1/2\) and columns ordered \(-1,1\),

\[ P=\begin{pmatrix}3/8&1/8\\1/8&3/8\end{pmatrix} \]

has barycenters \(-1/2,1/2\), so \(T_*(x)=x\) and \(V(\alpha,\beta)=0\). It accepts additional random motion to keep the conditional means fixed. Reversing the measures costs \(V(\beta,\alpha)=1/4\), since the conditional mean must lie in \([-1/2,1/2]\). This also exhibits the asymmetry.

A convex quadratic program for finite clouds

Let \(\alpha=\sum_{i=1}^n a_i\delta_{x_i}\) and \(\beta=\sum_{j=1}^m b_j\delta_{y_j}\), with positive weights summing to one. Use distinct support points, merging coincident atoms first, so that conditioning on \(X=x_i\) means selecting row \(i\). A plan is a nonnegative matrix with row sums \(a\) and column sums \(b\), denoted \(P\in\Pi(a,b)\). Its barycenters and weak objective are

\[ z_i(P)=\frac1{a_i}\sum_j P_{ij}y_j, \qquad \mathcal F(P)=\sum_i a_i|x_i-z_i(P)|^2. \]

The map \(P\mapsto z(P)\) is linear, so \(\mathcal F\) is convex quadratic. It need not be strictly convex in \(P\), but it is strictly convex in the vector of barycenters. This explains how several plans can give the same unique minimizing map.

Frank–Wolfe solves a succession of ordinary linear transport problems [6]. At a feasible \(P\), its linearized cost is

\[ C_{ij}=\frac{\partial\mathcal F}{\partial P_{ij}} =2(z_i(P)-x_i)\cdot y_j. \]

Find \(S\in\arg\min_{Q\in\Pi(a,b)}\langle C,Q\rangle\), then move towards \(S\). With \(d_i=z_i(S)-z_i(P)\), the exact quadratic line search is

\[ \gamma= \operatorname{clip}_{[0,1]} \frac{-\sum_i a_i(z_i(P)-x_i)\cdot d_i} {\sum_i a_i|d_i|^2}, \qquad P^+=P+\gamma(S-P). \]

Take \(\gamma=0\) when the denominator is zero. Every iterate remains feasible.

Proposition 2 (a certificate for both cost and map). The Frank–Wolfe gap \(g=\langle C,P-S\rangle\) bounds

\[ \sum_i a_i|z_i(P)-z_i^*|^2 \le\mathcal F(P)-\mathcal F_* \le g. \]
Proof: first-order convexity and the geometry of projection

Convexity gives \(\mathcal F(P)-\mathcal F(P_*)\le\langle C,P-P_*\rangle\). Since \(S\) minimizes the linearized objective, this is at most \(\langle C,P-S\rangle=g\).

For the other inequality, feasible barycenter vectors form a convex set. At its minimizer \(z^*\), the directional derivative towards any feasible \(z\) is nonnegative:

\[ \sum_i a_i(z_i^*-x_i)\cdot(z_i-z_i^*)\ge0. \]

Expand \(\sum_i a_i|z_i-x_i|^2\) around \(z^*\). The cross term is nonnegative, leaving at least \(\sum_i a_i|z_i-z_i^*|^2\) above the minimum. \(\square\)

A small gap thus certifies not just the objective but the root-mean-square error of the displayed endpoints, bounded by \(\sqrt g\). The matrix itself can remain nonunique.

Numerical experimentation

Drag red source points or blue target points, and change the two cloud sizes. The left panel shows an ordinary quadratic OT plan; the center shows its conditional means; the right recomputes the weak optimal barycentric map. Diamonds are mean destinations, not selected target atoms. The displayed weak gap bounds objective error and squared RMS endpoint error; use “Refine weak solution” to reduce it. Uniform masses are recomputed when cardinalities change.

The ordinary problem uses a minimum-cost-flow solver. The weak problem uses feasible Frank–Wolfe steps, including away steps to remove obsolete components from its current mixture of plans. The ordinary Frank–Wolfe gap remains the certificate. A finite iteration budget, especially while dragging, can leave a visible residual gap; the plot does not silently equate that iterate with the exact optimizer.

A useful check is the lower bound

\[ V(\alpha,\beta)\ge|m_\alpha-m_\beta|^2, \qquad m_\alpha=\int x\,d\alpha(x). \]

It follows from Jensen because \(\int T_\pi\,d\alpha=m_\beta\). Dragging changes the centroids without automatic recentering. Even when the centroids agree, weak cost need not vanish: finite clouds with the same mean need not be ordered in convex order.

Why the intermediate law is useful

Weak transport separates a deterministic correction from residual uncertainty. It asks which conditional means can be assigned to source points while retaining exactly the prescribed target distribution. That is a different modeling question from making every individual displacement small.

The same convex-order constraint appears in martingale optimal transport and model-independent finance: distributions at two times must be compatible with a martingale, while one optimizes over the remaining joint dependence [7]. Weak transport turns failure of that compatibility into an optimization problem: move the initial law as little as possible until a mean-preserving spread becomes feasible.

The important distinction is between the law of the conditional means and the final law. The first is selected by an optimal map; the second may require splitting every particle. Keeping both visible makes the term “weak” precise.

References

  1. Nathael Gozlan, Cyril Roberto, Paul-Marie Samson, and Prasad Tetali. Kantorovich duality for general transport costs and applications. Journal of Functional Analysis 273(11), 3327–3405, 2017.
  2. Nathael Gozlan and Nicolas Juillet. On a mixture of Brenier and Strassen theorems. Proceedings of the London Mathematical Society 120(3), 434–463, 2020.
  3. Julio Backhoff-Veraguas, Mathias Beiglböck, and Gudmund Pammer. Existence, duality, and cyclical monotonicity for weak transport costs. Calculus of Variations and Partial Differential Equations 58, article 203, 2019.
  4. Gabriel Peyré. Optimal Transport for Machine Learners, 2025–2026. §12.6, Propositions 12.33 and 12.44.
  5. Volker Strassen. The Existence of Probability Measures with Given Marginals. Annals of Mathematical Statistics 36(2), 423–439, 1965.
  6. Martin Jaggi. Revisiting Frank-Wolfe: Projection-Free Sparse Convex Optimization. ICML, PMLR 28(1), 427–435, 2013.
  7. Mathias Beiglböck, Pierre Henry-Labordère, and Friedrich Penkner. Model-independent bounds for option prices: a mass transport approach. Finance and Stochastics 17, 477–501, 2013.