Skip to article frontmatterSkip to article content
Site not loading correctly?

This may be due to an incorrect BASE_URL configuration. See the MyST Documentation for reference.

Statistical Optimal Transport

Optimal transport is rarely evaluated on population measures directly. In machine learning and statistics, the inputs are usually empirical laws, histograms, simulated particles or minibatches, and the central question is therefore no longer only how to compute OT, but how OT behaves as a random estimator. This chapter studies this statistical layer: qualitative consistency of empirical measures, non-asymptotic sample-complexity rates, and asymptotic bias--variance decompositions for exact and regularized transport costs.

This statistical convergence is conceptually different from the algorithmic convergence studied in the previous chapter. There the marginals and the temperature were fixed and one asked how Sinkhorn iterates approach a regularized optimizer. Here the number of samples grows, the empirical measures themselves move, and the regularization parameter may either remain fixed or be sent to zero. The resulting picture explains why exact OT is statistically expensive in high intrinsic dimension, why fixed-temperature Sinkhorn has smoother parametric fluctuations, and why approximating exact OT with entropy always involves a bias--variance tradeoff.

Law of Large Numbers and Central Limit Theorem

Before discussing sample complexity, it is useful to separate consistency from rates. If X1,,XnX_1,\ldots,X_n are i.i.d. samples with common law α\alpha, the associated empirical measure is the random probability measure

α^n:=1ni=1nδXi.\hat\alpha_n \eqdef \frac1n\sum_{i=1}^n\delta_{X_i}.

The ordinary law of large numbers says that empirical averages converge to expectations. In measure language this means that α^n\hat\alpha_n converges weakly toward α\alpha, because testing α^n\hat\alpha_n against a bounded continuous function φ\varphi gives the sample average n1iφ(Xi)n^{-1}\sum_i\varphi(X_i). Wasserstein distances strengthen this statement by also recording moment convergence. Thus, if α\alpha has a finite pp-th moment, the empirical law converges to α\alpha in Wp\Wass_p, almost surely and in pp-th mean in the sense that EWp(α^n,α)p0\mathbb E\Wass_p(\hat\alpha_n,\alpha)^p\to0. This is the qualitative consistency statement behind empirical OT plug-in estimators: empirical transport distances converge to their population counterparts once the sampled laws themselves converge in Wasserstein distance. It says nothing yet about the speed, which is the topic of Sample Complexity.

Proof

Fix a reference point x0Xx_0\in\Xx, and write r(x)=d(x,x0)r(x)=d(x,x_0). Since X\Xx is Polish, the weak topology on P(X)\Pp(\Xx) admits a countable convergence-determining class (φk)k1Cb(X)(\varphi_k)_{k\geq1}\subset C_b(\Xx). For each fixed kk, the strong law of large numbers gives

φkdα^n=1ni=1nφk(Xi)Eφk(X1)=φkdα\int \varphi_k\,\d\hat\alpha_n = \frac1n\sum_{i=1}^n\varphi_k(X_i) \longrightarrow \mathbb E\varphi_k(X_1) = \int\varphi_k\,\d\alpha

almost surely. Intersecting these probability-one events over the countable set of indices gives convergence against every φk\varphi_k, hence weak convergence α^nα\hat\alpha_n\rightharpoonup\alpha.

The moment condition αPp(X)\alpha\in\Pp_p(\Xx) means rpdα<+\int r^p\,\d\alpha<+\infty. Applying the strong law again to r(X1)pr(X_1)^p gives

rpdα^n=1ni=1nr(Xi)prpdα\int r^p\,\d\hat\alpha_n = \frac1n\sum_{i=1}^n r(X_i)^p \longrightarrow \int r^p\,\d\alpha

almost surely. Weak convergence plus convergence of pp-th moments is equivalent to Wp\Wass_p convergence on Pp(X)\Pp_p(\Xx), so Wp(α^n,α)0\Wass_p(\hat\alpha_n,\alpha)\to0 almost surely.

For convergence in expectation, set An=rpdα^nA_n=\int r^p\,\d\hat\alpha_n and M=rpdαM=\int r^p\,\d\alpha. The triangle inequality through the Dirac mass δx0\delta_{x_0}, followed by (a+b)p2p1(ap+bp)(a+b)^p\leq2^{p-1}(a^p+b^p), gives

Wp(α^n,α)p2p1(An+M).\Wass_p(\hat\alpha_n,\alpha)^p \leq 2^{p-1}(A_n+M).

The family (An)n(A_n)_n is uniformly integrable. By the de la Vallee--Poussin criterion, choose a convex superlinear function Ψ\Psi such that EΨ(r(X1)p)<+\mathbb E\Psi(r(X_1)^p)<+\infty; Jensen’s inequality gives

EΨ(An)1ni=1nEΨ(r(Xi)p)=EΨ(r(X1)p).\mathbb E\Psi(A_n) \leq \frac1n\sum_{i=1}^n\mathbb E\Psi(r(X_i)^p) = \mathbb E\Psi(r(X_1)^p).

Thus (An+M)n(A_n+M)_n, and hence (Wp(α^n,α)p)n(\Wass_p(\hat\alpha_n,\alpha)^p)_n, is uniformly integrable. Together with almost-sure convergence to zero, this implies EWp(α^n,α)p0\mathbb E\Wass_p(\hat\alpha_n,\alpha)^p\to0.

Central-Limit Fluctuations

The previous proposition is a law-of-large-numbers statement: a random empirical measure converges to the law that generated it. The central limit theorem describes a different, fluctuation-scale limit. As recalled in Remark Remark: Central limit theorem, if (Xi)i1(X_i)_{i\geq1} are centered i.i.d. random vectors with identity covariance, the law of n1/2iXin^{-1/2}\sum_i X_i converges weakly toward a Gaussian. Equivalently, if α\alpha is the common law of the XiX_i, this law is the rescaled convolution (D1/n)αn(D_{1/\sqrt n})_\sharp\alpha^{*n}. Wasserstein distances make this qualitative convergence quantitative. The next result is a W1\Wass_1 form of the Berry--Esseen theorem: it controls the error uniformly over all 1-Lipschitz test functions. The qualitative Bernoulli example was visualized earlier in Figure Div.

Proof

By Kantorovich--Rubinstein duality,

W1(αn,γ)=supLip(h)1Eh(Sn)Eh(G),Sn=n1/2iXi,Gγ.\Wass_1(\alpha_n,\gamma) = \sup_{\Lip(h)\leq1} \left|\mathbb{E}h(S_n)-\mathbb{E}h(G)\right|, \qquad S_n=n^{-1/2}\sum_iX_i, \quad G\sim\gamma.

For each such hh, solve Stein’s equation fh(x)xfh(x)=h(x)Eh(G)f_h'(x)-xf_h(x)=h(x)-\mathbb{E}h(G). Its solution satisfies fh+fhC\norm{f_h'}_\infty+\norm{f_h''}_\infty\leq C. Writing Sn(i)=SnXi/nS_n^{(i)}=S_n-X_i/\sqrt n, independence and EXi=0\mathbb E X_i=0 give

E[Snfh(Sn)]=1ni=1nE[Xi(fh(Sn(i)+Xi/n)fh(Sn(i)))].\mathbb E[S_nf_h(S_n)] = \frac1{\sqrt n}\sum_{i=1}^n \mathbb E\left[X_i\left(f_h(S_n^{(i)}+X_i/\sqrt n)-f_h(S_n^{(i)})\right)\right].

Taylor’s formula with integral remainder and EXi2=1\mathbb E X_i^2=1 show that this differs from n1iEfh(Sn(i))n^{-1}\sum_i\mathbb E f_h'(S_n^{(i)}) by at most CEX13/nC\mathbb E|X_1|^3/\sqrt n. The Lipschitz bound on fhf_h' replaces this average by Efh(Sn)\mathbb E f_h'(S_n) at the same order. Stein’s identity and the duality formula then prove the claim Berry, 1941Esseen, 1942Chen et al., 2011Bobkov, 2018Rio, 2011.

The universal n1/2n^{-1/2} bound is sharp over broad classes of input laws, but it need not describe the asymptotic behavior of a fixed law. For symmetric inputs, the smooth skewness correction vanishes. What replaces it depends on whether the law is lattice-valued or has a density.

Proof

Write FnF_n for the CDF of αn\alpha_n, and Φ\Phi and φ\varphi for the standard Gaussian CDF and density. With ψ(u)=12{u}\psi(u)=\frac12-\{u\}, the integrated lattice Edgeworth expansion gives

Fn(x)Φ(x)=hnψ ⁣(nxnah)φ(x)+rn(x),rnL1(R)=o(n1/2);F_n(x)-\Phi(x) = \frac h{\sqrt n}\, \psi\!\left(\frac{\sqrt n\,x-na}{h}\right)\varphi(x)+r_n(x), \qquad \|r_n\|_{L^1(\mathbb R)}=o(n^{-1/2});

see Petrov, 1975Bhattacharya & Ranga Rao, 2010Kolassa & McCullagh, 1990. Vallender’s identity Vallender, 1974 states that W1(αn,γ)=RFnΦ\Wass_1(\alpha_n,\gamma)=\int_{\mathbb R}|F_n-\Phi|. The inequality u+vuv\big||u+v|-|u|\big|\leq|v| reduces the result to periodic averaging:

Rψ ⁣(nxnah)φ(x)dx01ψ(u)du=14.\int_{\mathbb R} \left|\psi\!\left(\frac{\sqrt n\,x-na}{h}\right)\right|\varphi(x)\,dx \longrightarrow \int_0^1|\psi(u)|\,du =\frac14.

For completeness, prove this first for compactly supported step functions, where it is a Riemann sum over periods, and conclude by L1L^1 approximation of φ\varphi.

The absence of a lattice correction reveals the next smooth Edgeworth term. A density automatically satisfies Cramér’s non-lattice condition because its characteristic function vanishes at infinity.

Proof

The non-lattice Edgeworth expansion through order n1n^{-1} has an n1/2n^{-1/2} term proportional to the third cumulant, and n1n^{-1} terms proportional to the fourth cumulant and to the square of the third one. Symmetry makes the third cumulant vanish and gives

Fn(x)Φ(x)=κ424nH3(x)φ(x)+rn(x),rnL1(R)=o(n1);F_n(x)-\Phi(x) =-\frac{\kappa_4}{24n}H_3(x)\varphi(x)+r_n(x), \qquad \|r_n\|_{L^1(\mathbb R)}=o(n^{-1});

see Petrov, 1975Bhattacharya & Ranga Rao, 2010. Insert this expansion into Vallender’s identity and use u+vuv\big||u+v|-|u|\big|\leq|v|.

These two results explain the different behaviors in Figure Div. For the symmetric Bernoulli law, the maximal span is h=2h=2, hence W1(αn,γ)=1/(2n)+o(n1/2)\Wass_1(\alpha_n,\gamma)=1/(2\sqrt n)+o(n^{-1/2}). The density proposition cannot be used: this law has no density and its characteristic function is E[eitX]=cost\mathbb E[e^{itX}]=\cos t, whose modulus returns to one at arbitrarily large frequencies. Thus symmetry removes the smooth skewness term but not the lattice sawtooth. Conversely, XUnif[3,3]X\sim\operatorname{Unif}[-\sqrt3,\sqrt3] has a density, characteristic function sin(3t)/(3t)0\sin(\sqrt3t)/(\sqrt3t)\to0, and κ4=6/5\kappa_4=-6/5. Since

RH3(x)φ(x)dx=2φ(0)+8φ(3)=2+8e3/22π,\int_{\mathbb R}|H_3(x)|\varphi(x)\,dx =2\varphi(0)+8\varphi(\sqrt3) =\frac{2+8e^{-3/2}}{\sqrt{2\pi}},

one obtains

W1(αn,γ){12n,X12(δ1+δ1),1+4e3/2102π1n,XUnif[3,3].\Wass_1(\alpha_n,\gamma) \sim \begin{cases} \dfrac1{2\sqrt n}, &X\sim\frac12(\delta_{-1}+\delta_1),\\[2mm] \dfrac{1+4e^{-3/2}}{10\sqrt{2\pi}}\dfrac1n, &X\sim\operatorname{Unif}[-\sqrt3,\sqrt3]. \end{cases}

Figure Div confronts these equivalents with exact one-dimensional computations. For a centered, unit-variance input law α0\alpha_0, define

αn=(D1/n)α0n,n1,\alpha_n=(D_{1/\sqrt n})_\sharp\alpha_0^{*n}, \qquad n\geq1,

so that α1=α0\alpha_1=\alpha_0. For Bernoulli input, the distance is evaluated by the exact quantile formula and atom masses are divided by the current lattice spacing in the density display. For continuous-uniform input, the normalized convolution is an affine image of the Irwin--Hall distribution and Vallender’s CDF formula is integrated numerically. Neither computation uses Monte Carlo sampling.

<IPython.core.display.Image object>

Sharp lattice and density central-limit asymptotics in W1\Wass_1. The two left panels show α0\alpha_0, α2\alpha_2, and α6\alpha_6 for symmetric Bernoulli and continuous-uniform inputs; the gray curve is the standard Gaussian density. The right panel compares the exact numerical distances (solid) with the sharp equivalents in (17) (dashed). The Bernoulli curve follows its lattice rate 1/(2n)1/(2\sqrt n), while the continuous-uniform curve approaches (1+4e3/2)/(102πn)(1+4e^{-3/2})/(10\sqrt{2\pi}\,n).

Empirical-Process Fluctuations

The fluctuation object used by statistical OT is the empirical law itself, not only a normalized sum in Euclidean space. There is no canonical Gaussian law on an unrestricted infinite-dimensional space of signed measures, so the precise finite-dimensional statement is made through test functions. Uniform convergence over an infinite class requires a Donsker condition Vaart & Wellner, 1996.

Proof

Apply the multivariate central limit theorem to the centered i.i.d. vectors (φr(Xi)φrdα)r=1q(\varphi_r(X_i)-\int\varphi_r\,d\alpha)_{r=1}^q.

This linear CLT is the input to the delta-method arguments in Bias and Variance of OT. Exact OT may have only directional derivatives, whereas fixed-temperature entropic OT is smooth in the relative interiors of finite probability simplices.

Sample Complexity

This section compares four statistical regimes. Exact OT resolves geometry at all scales and pays rates controlled by the intrinsic dimension. MMD replaces transport by a kernel mean embedding and has a parametric Monte-Carlo rate for bounded kernels. Fixed-temperature Sinkhorn divergences smooth the dual potentials and also recover parametric fluctuations, at the price of regularization bias. Sliced Wasserstein inherits one-dimensional empirical rates through its projected laws.

Unregularized OT

The previous section proves qualitative convergence of empirical laws. The sample-complexity question is how fast this convergence happens. The sample complexity of unregularized OT suffers from the curse of dimensionality, but the relevant dimension is geometric. If the distributions are supported on a regular lower-dimensional set, for instance a dd'-dimensional submanifold of Rd\RR^d, the empirical rate is governed by dd' rather than by the ambient dimension dd. In the high-dimensional regime for Wp\Wass_p, namely d>2pd'>2p, this gives the characteristic rate n1/dn^{-1/d'}. Exact OT is therefore dimension-adaptive: it sees the intrinsic dimension of the data support through its covering numbers Dudley, 1969Weed & Bach, 2019. Related two-sample-testing viewpoints are developed in Ramdas et al., 2017.

Figure Div gives a numerical overview: exact OT exhibits dimension-dependent empirical fluctuations, whereas MMD and fixed-temperature Sinkhorn divergences lie much closer to the parametric n1/2n^{-1/2} scale.

<IPython.core.display.Image object>

Empirical fluctuations in dimensions three and six. For each sample size nn, two independent empirical measures are drawn from the same standard Gaussian law. Exact OT follows a slower dimension-dependent scale, while MMD and the fixed-ϵ\epsilon Sinkhorn divergence behave closer to the parametric n1/2n^{-1/2} guide. This is a statistical illustration, not a solver benchmark.

Interactive panel. Vary dimension, sample budget and temperature to compare the curse-of-dimensionality OT guide with the parametric fluctuation and bias floor of entropic OT.

Proof Sketch

Match the common mass inside the finest cells, then recursively match the remaining common mass inside each parent cell. Mass first matched at scale jj moves by at most d,2j\sqrt d,2^{-j}, and its total amount is bounded by the sum of the child-cell imbalances. The residual at the finest unresolved scale has mass at most one and moves by at most d,2J\sqrt d,2^{-J}.

Proof

The triangle inequality reduces the result to EW1(α^N,α)\mathbb E\Wass_1(\hat\alpha_N,\alpha). For each dyadic cell QQ,

Eα^N(Q)α(Q)α(Q)/N.\mathbb E|\hat\alpha_N(Q)-\alpha(Q)| \leq \sqrt{\alpha(Q)/N}.

Summing by Cauchy--Schwarz and applying Proposition: Dyadic Partition Bound for W1\Wass_1 gives

EW1(α^N,α)d2J+N1/2j=0J12j(d/21).\mathbb E\Wass_1(\hat\alpha_N,\alpha) \lesssim_d 2^{-J}+N^{-1/2}\sum_{j=0}^{J-1}2^{j(d/2-1)}.

Optimizing JJ yields the three displayed regimes. Replacing dyadic cubes by covers of cardinality O(2jd)O(2^{jd'}) gives the intrinsic-dimensional extension Dudley, 1969Fournier & Guillin, 2015Weed & Bach, 2019.

Lower Bounds and Minimax Optimality

The high-dimensional rate is not merely a defect of the empirical plug-in estimator. A hypercube of localized density perturbations gives the following distribution-estimation obstruction Singh & Póczos, 2018Weed & Bach, 2019Chewi et al., 2025.

Proof Sketch

Partition the cube into M=mdM=m^d cells of width h=1/mh=1/m, and put an independent signed, zero-mean bump of fixed small amplitude in each cell. Kantorovich--Rubinstein duality gives a separation of order hd+1h^{d+1} per differing sign. Neighboring experiments have one-sample KL divergence O(hd)O(h^d), so choosing MNM\asymp N keeps their NN-sample total variation bounded away from one. Assouad’s lemma then gives Mhd+1=hN1/dMh^{d+1}=h\asymp N^{-1/d}.

Leveraging Smoothness

Smooth densities permit a better estimator: first remove fine empirical oscillations with a wavelet or kernel smoother, then compute OT between the smoothed laws Niles-Weed & Berthet, 2022.

Proof Sketch

At wavelet scale JJ, the B1,11B^{-1}_{1,1} approximation bias is O(2J(s+1))O(2^{-J(s+1)}) and the stochastic term is O(N1/22J(d/21))O(N^{-1/2}2^{J(d/2-1)}). Since W1\Wass_1 is controlled by this negative Besov norm for densities bounded away from zero, balancing the two terms with 2JN1/(d+2s)2^J\asymp N^{1/(d+2s)} gives the rate. Projection onto nonnegative unit-mass densities preserves its order.

This gain is statistical rather than automatically computational: a grid at bandwidth hh has O(hd)O(h^{-d}) degrees of freedom, so direct smoothing remains expensive in high dimension. Sum-of-squares relaxations offer theoretical dimension-free exponents but are still costly in practice Vacher et al., 2021.

MMD

Proof

Let Φ(x)=k(x,)\Phi(x)=k(x,\cdot) be the feature map and mα=EΦ(X)m_\alpha=\mathbb E\Phi(X). The reverse triangle inequality gives

MMDk(α^n,β^m)MMDk(α,β)MMDk(α^n,α)+MMDk(β^m,β).\abs{ \operatorname{MMD}_k(\hat\alpha_n,\hat\beta_m) - \operatorname{MMD}_k(\alpha,\beta) } \leq \operatorname{MMD}_k(\hat\alpha_n,\alpha) + \operatorname{MMD}_k(\hat\beta_m,\beta).

Independence cancels cross terms after taking squared norms and expectation:

EMMDk(α^n,α)2=1nEΦ(X)mαHk2=1n(Ek(X,X)Ek(X,X)).\mathbb E\operatorname{MMD}_k(\hat\alpha_n,\alpha)^2 = \frac1n \mathbb E\norm{\Phi(X)-m_\alpha}_{\mathcal H_k}^2 = \frac1n \left( \mathbb E k(X,X)-\mathbb E k(X,X') \right).

Jensen’s inequality and k(x,x)κ2k(x,x)\leq\kappa^2 give the displayed bound.

Entropic OT

Entropic regularization keeps the transport geometry while smoothing the dual potentials. The sharp fixed-temperature statement below has polynomial, rather than exponential, dependence on the inverse temperature Genevay et al., 2019Mena & Niles-Weed, 2019.

Proof Sketch

Dual optimality bounds a one-marginal perturbation by the empirical process indexed by normalized entropic potentials. Mena--Niles-Weed prove polynomial local Hölder bounds for these potentials under a common subgaussian proxy. Their covering numbers have a finite Dudley integral, hence the empirical process is O(n1/2)O(n^{-1/2}) with constant Cd(1+σqd)C_d(1+\sigma^{q_d}) at ϵ=1\epsilon=1. The rescaling xx/ϵx\mapsto x/\sqrt\epsilon gives the displayed Λd,σ(ϵ)\Lambda_{d,\sigma}(\epsilon). Perturbing both marginals and applying the same estimate to the cross and two self terms yields the debiased result.

Sample Complexity of Estimating OT Maps

The preceding estimates concern transport values. For map estimation, one solves the empirical dual problem for potentials (fn,mϵ,gn,mϵ)Rn×Rm(\mathbf f_{n,m}^\epsilon,\mathbf g_{n,m}^\epsilon)\in\RR^n\times\RR^m, with ϵ=0\epsilon=0 denoting the unregularized Kantorovich problem, and then uses the soft cc-transform as an out-of-sample extrapolator.

For the quadratic cost c2(x,y)=xy2/2c_2(x,y)=\norm{x-y}^2/2, let Pn,mϵ=(Pijϵ)\P_{n,m}^\epsilon=(\P_{ij}^\epsilon) be the empirical entropic coupling and let gn,mϵ\mathbf g_{n,m}^\epsilon be the target-side dual potential. Define

un,mϵ(x)=ϵlogjbjexp(gn,m,jϵc2(x,Yj)ϵ),Tn,mϵ(x)=xun,mϵ(x).u_{n,m}^\epsilon(x) = -\epsilon\log \sum_j b_j \exp\left( \frac{\mathbf g_{n,m,j}^\epsilon-c_2(x,Y_j)}{\epsilon} \right), \qquad T_{n,m}^\epsilon(x)=x-\nabla u_{n,m}^\epsilon(x).

Equivalently, subtracting the quadratic marginal terms turns c2c_2 into the bilinear cost cip(x,y)=x,yc_{\rm ip}(x,y)=-\langle x,y\rangle. With the sign convention used here, the associated soft convex potential is

φn,mϵ(x)=12x2un,mϵ(x)=ϵlogjbjexp(x,Yj+gn,m,jϵ12Yj2ϵ),Tn,mϵ=φn,mϵ.\varphi_{n,m}^\epsilon(x) = \frac12\norm{x}^2-u_{n,m}^\epsilon(x) = \epsilon\log\sum_j b_j \exp\left( \frac{\langle x,Y_j\rangle+\mathbf g_{n,m,j}^\epsilon-\frac12\norm{Y_j}^2}{\epsilon} \right), \qquad T_{n,m}^\epsilon=\nabla\varphi_{n,m}^\epsilon .

Differentiating this log-sum-exp gives the explicit soft barycentric form

Tn,mϵ(x)=jωjϵ(x)Yj,ωjϵ(x)=bje(gn,m,jϵc2(x,Yj))/ϵkbke(gn,m,kϵc2(x,Yk))/ϵ.T_{n,m}^\epsilon(x)=\sum_j\omega_j^\epsilon(x)Y_j, \qquad \omega_j^\epsilon(x)= \frac{b_j e^{(\mathbf g_{n,m,j}^\epsilon-c_2(x,Y_j))/\epsilon}} {\sum_k b_k e^{(\mathbf g_{n,m,k}^\epsilon-c_2(x,Y_k))/\epsilon}} .

At a sampled point XiX_i, Sinkhorn normalization gives

Tn,mϵ(Xi)=1aijPijϵYj,T_{n,m}^\epsilon(X_i)=\frac1{a_i}\sum_jP_{ij}^\epsilon Y_j,

which is the barycentric projection of the entropic plan. At ϵ=0\epsilon=0, the hard cc-transform is generally nondifferentiable: away from Laguerre boundaries its gradient selects one target site, whereas the barycentric projection of an optimal unregularized plan may average several target sites when a row splits. The consistency statement is therefore formulated directly for barycentric projections. It uses uniform empirical weights; deterministic nonuniform weights require the corresponding entropy bound.

Proof Sketch

Compare the entropic minimizer with an unregularized empirical optimal plan. On a uniform n×mn\times m grid, this relative entropy is the mutual information of two uniform indices and is at most min{logn,logm}\min\{\log n,\log m\}. Thus the entropic coupling has asymptotically optimal quadratic cost when ϵlog(min{n,m})0\epsilon\log(\min\{n,m\})\to0. Any weak limit is therefore an optimal population plan, hence the Brenier graph (Id,T)#α(\operatorname{Id},T)_\#\alpha. Jensen’s inequality applied to each conditional law reduces the claim to the convergence of yT(x)2dPn,mϵ(x,y)\iint\norm{y-T(x)}^2\,dP_{n,m}^{\epsilon}(x,y). This last convergence is not a continuity tautology, because the Brenier map is only defined α\alpha-a.e. At every differentiability point of ϕ\phi, the subdifferential is the singleton {ϕ(x)}\{\nabla\phi(x)\}; compactness of the target bounds the selected subgradients, and the closed-graph property of the subdifferential then gives continuity of any selected subgradient there. Since convex functions are differentiable Lebesgue-a.e. and α\alpha is absolutely continuous, the portmanteau theorem for bounded test functions continuous almost everywhere under the limiting measure gives the desired limit.

Quantitative map rates require stronger stability assumptions on Brenier potentials. For unregularized map estimation, Hütter--Rigollet, Deb--Ghosal--Sen, and Manole--Balakrishnan--Niles-Weed--Wasserman obtain, under smoothness and curvature assumptions, squared L2(α)L^2(\alpha) risk of order

ET^nT0L2(α)2(n2s2(s1)+dn1)\mathbb E\norm{\widehat T_n-T_0}_{L^2(\alpha)}^2 \lesssim \left(n^{-\frac{2s}{2(s-1)+d}}\vee n^{-1}\right)

up to logarithms, and this rate is minimax optimal Hütter & Rigollet, 2021Deb et al., 2021Manole et al., 2024. For the Sinkhorn-computable entropic barycentric estimator, Pooladian--Niles-Weed prove

ETϵ,(n,n)T0L2(α)2ϵd/2log(n)n1/2+ϵ(sˉ+1)/2+ϵ2I0(α,β),\mathbb E\norm{T_{\epsilon,(n,n)}-T_0}_{L^2(\alpha)}^2 \lesssim \epsilon^{-d/2}\log(n)n^{-1/2} + \epsilon^{(\bar s+1)/2} + \epsilon^2\mathcal I_0(\alpha,\beta),

with sˉ=s3\bar s=s\wedge3 Pooladian & Niles-Weed, 2021. The first term is the statistical fluctuation of the entropic map, while the last two terms are regularization-bias terms. Thus fixed ϵ\epsilon gives a parametric empirical fluctuation for the regularized map, whereas ϵn1/(d+sˉ+1)\epsilon\simeq n^{-1/(d+\bar s+1)} gives a convergent estimator of the Brenier map. The unregularized case can achieve sharper minimax rates when combined with enough smooth extension machinery; the regularized estimator is instead directly Sinkhorn-computable.

The interactive demo below is only a scaling guide: change the dimension to see the exact-OT exponent flatten, and change ϵ\epsilon to move the Sinkhorn bias floor against its parametric fluctuation term.

Interactive panel. This exploratory panel is a scaling guide. Use dimension, sample size, and epsilon to compare statistical fluctuation with regularization bias.

Sliced Wasserstein

For Pθ(x)=θ,xP_\theta(x)=\langle\theta,x\rangle and normalized surface measure σ\sigma on Sd1\mathbb S^{d-1}, define

SWp(α,β)p=Sd1Wp((Pθ)#α,(Pθ)#β)p,dσ(θ).\SW_p(\alpha,\beta)^p = \int_{\mathbb S^{d-1}} \Wass_p((P_\theta)_\#\alpha,(P_\theta)_\#\beta)^p,d\sigma(\theta).

The construction is studied in Sliced Wasserstein Distances. Statistically, it inherits one-dimensional empirical rates rather than ambient-dimensional matching rates Nadjahi et al., 2019Nadjahi et al., 2020Manole et al., 2022.

Proof

The triangle inequality reduces the claim to two one-sample sliced distances. Jensen’s inequality moves expectation inside the spherical integral, after which the assumed projected bounds apply. For p=1p=1, use the Dvoretzky--Kiefer--Wolfowitz inequality and W1=Fα^nFα\Wass_1=\int|F_{\hat\alpha_n}-F_\alpha|. For p=2p=2, use the one-dimensional J2J_2 quantile-process estimate Bobkov & Ledoux, 2019.

Bias and Variance of OT

The previous section answered a coarse but essential question: how many samples are needed before an empirical OT quantity is accurate, typically up to universal constants. We now ask for a finer statistical description. Rather than only estimating the size of the error, we seek the first expansion of the plug-in estimator. Given empirical laws α^n\hat\alpha_n and β^m\hat\beta_m, and a scalar OT value V\mathcal V, this expansion separates the leading statistical bias

Bn,m(V):=EV(α^n,β^m)V(α,β)B_{n,m}(\mathcal V) \eqdef \mathbb E\mathcal V(\hat\alpha_n,\hat\beta_m)-\mathcal V(\alpha,\beta)

and the centered fluctuation

Zn,m(V):=V(α^n,β^m)EV(α^n,β^m),Z_{n,m}(\mathcal V) \eqdef \mathcal V(\hat\alpha_n,\hat\beta_m) -\mathbb E\mathcal V(\hat\alpha_n,\hat\beta_m),

for values such as V(α,β)=Wpp(α,β)\mathcal V(\alpha,\beta)=\Wass_p^p(\alpha,\beta), V(α,β)=Lc(α,β)\mathcal V(\alpha,\beta)=\MK_c(\alpha,\beta), or their entropic analogues. This goes beyond Proposition Proposition: Finite-Dimensional Empirical-Process CLT: the empirical fluctuation is now pushed through a nonlinear, often nonsmooth, transport value. The resulting law is governed by local differentiability and by the geometry of the optimal dual face.

Literature Map

The shape of the limit depends sharply on the analytic regularity of the transport value. For exact OT on a finite or countable space, the map from empirical weights to optimal cost is convex and piecewise affine. Sommerfeld and Munk Sommerfeld & Munk, 2018, and the countable-space extensions of Tameling--Sommerfeld--Munk Tameling et al., 2019, show that the natural limit is a directional delta-method limit: often a support function of a Gaussian process over an optimal dual face, rather than a Gaussian random variable. For Euclidean costs, del Barrio--Loubes and collaborators obtain central-limit theorems for empirical transportation costs under regularity and uniqueness assumptions on the Kantorovich potentials Barrio & Loubes, 2019Barrio et al., 2021. Entropic regularization makes the problem smoother. For fixed ϵ>0\epsilon>0, the value is differentiable with respect to the marginals, and the limiting variance is read directly from the entropic dual potentials. This is the point of view developed by Bigot--Cazelles--Papadakis, Klatt--Tameling--Munk, Hundrieser--Klatt--Munk, and Mena--Niles-Weed Bigot et al., 2019Klatt et al., 2020Hundrieser et al., 2021Mena & Niles-Weed, 2019. More recent results clarify weak limits beyond smooth costs and second-order null limits for Sinkhorn divergences González-Sanz & Hundrieser, 2023Goldfeld et al., 2024. These asymptotic statements do not replace the dimension-dependent bounds of Sample Complexity; they explain what the leading random object is, what constants appear, and where the bias comes from.

Bias Versus Centered Fluctuation

The decomposition

V(α^n,β^m)V(α,β)=EV(α^n,β^m)V(α,β)bias+V(α^n,β^m)EV(α^n,β^m)centered fluctuation\mathcal V(\hat\alpha_n,\hat\beta_m)-\mathcal V(\alpha,\beta) = \underbrace{\mathbb E\mathcal V(\hat\alpha_n,\hat\beta_m)-\mathcal V(\alpha,\beta)}_{\text{bias}} + \underbrace{\mathcal V(\hat\alpha_n,\hat\beta_m)-\mathbb E\mathcal V(\hat\alpha_n,\hat\beta_m)}_{\text{centered fluctuation}}

is elementary, but in OT it is genuinely diagnostic because the two terms may live on different scales. For the self-distance V(α^n,α)=Wpp(α^n,α)\mathcal V(\hat\alpha_n,\alpha)=\Wass_p^p(\hat\alpha_n,\alpha), the population value is zero and the deterministic bias EWpp(α^n,α)\mathbb E\Wass_p^p(\hat\alpha_n,\alpha) is the leading term. On a regular dd'-dimensional support, the high-dimensional matching scale is np/dn^{-p/d'} when d>2pd'>2p, with a critical logarithmic correction when d=2pd'=2p. Below this threshold the rate is controlled by empirical-process fluctuations and can be faster; in one dimension, smooth positive densities give the familiar quantile-process asymptotics for Wpp\Wass_p^p, of order np/2n^{-p/2} for p>1p>1, and in particular n1n^{-1} for W22\Wass_2^2. Thus p/dp/d' is the high-dimensional matching exponent, not a universal rate formula Dudley, 1969Fournier & Guillin, 2015Weed & Bach, 2019. By contrast, when V(α,β)\mathcal V(\alpha,\beta) is nonzero and the OT value is differentiable at (α,β)(\alpha,\beta), the centered fluctuation is often governed by an ordinary n1/2n^{-1/2} central limit theorem. A ``slow OT statistic’’ can therefore have three distinct causes: a large empirical bias, a nonsmooth directional limit, or a regularization bias Lcϵ(α,β)Lc(α,β)\MK_c^\epsilon(\alpha,\beta)-\MK_c(\alpha,\beta) that still has to be removed.

Proof

The multinomial central limit theorem gives

N(a^Na)Ga,Ga{ξ:1,ξ=0}.\sqrt N(\hat a_N-a)\Longrightarrow G_a, \qquad G_a\in\{\xi:\dotp{\ones}{\xi}=0\}.

The dual formulation writes LC(,b)\MKD_\C(\cdot,b) as the supremum, over the feasible polyhedron fi+gjCijf_i+g_j\leq\C_{ij}, of the affine functions af,a+g,ba\mapsto\dotp f a+\dotp g b. Its directional derivative at aa, in any tangent direction hh with 1,h=0\dotp{\ones}{h}=0, is therefore

DaLC(a,b)[h]=sup(f,g)D(a,b)f,h,D_a\MKD_\C(a,b)[h] = \sup_{(f,g)\in\mathcal D^\star(a,b)}\dotp f h,

which is the finite-dimensional form of Danskin’s theorem already used in Proposition: First variations of unregularized OT. The directional delta method gives the displayed distributional limit. After fixing a gauge, the relevant dual face is bounded, so the support functions above have at most linear growth in the multinomial fluctuation. Uniform integrability then yields convergence of the first two moments, hence the bias and variance statements. If ff^\star is unique modulo constants, the support function reduces to the linear form f,Ga\dotp{f^\star}{G_a}. Its expectation is zero, and the covariance of GaG_a gives exactly the displayed variance.

The same argument gives the independent two-sample limit. If N/(N+M)λ(0,1)N/(N+M)\to\lambda\in(0,1), then, with rN,M=NM/(N+M)r_{N,M}=\sqrt{NM/(N+M)},

rN,M(LC(a^N,b^M)LC(a,b))sup(f,g)D(a,b)(1λf,Ga+λg,Gb),r_{N,M}\big(\MKD_\C(\hat a_N,\hat b_M)-\MKD_\C(a,b)\big) \Longrightarrow \sup_{(f,g)\in\mathcal D^\star(a,b)} \left( \sqrt{1-\lambda}\,\dotp{f}{G_a} + \sqrt{\lambda}\,\dotp{g}{G_b} \right),

where GaG_a and GbG_b are independent multinomial Gaussian limits. When both dual potentials are unique, the first-order bias is negligible and the asymptotic variance of the unscaled estimator is

1Niai(fifˉ)2+1Mjbj(gjgˉ)2,fˉ=iaifi,gˉ=jbjgj.\frac{1}{N}\sum_i a_i(f^\star_i-\bar f^\star)^2 + \frac{1}{M}\sum_j b_j(g^\star_j-\bar g^\star)^2, \qquad \bar f^\star=\sum_i a_i f_i^\star, \quad \bar g^\star=\sum_j b_j g_j^\star.

If the optimal face contains several dual potentials, the limit is the supremum of a Gaussian process over that face and need not itself be Gaussian. The leading bias is then the expectation of this supremum divided by the square-root sample size. This nonsmoothness is the basic reason why inference for exact OT can remain delicate even on a finite space.

Proof

For positive histograms and fixed ϵ>0\epsilon>0, the entropic problem has a unique optimizer, and the KL-normalized value is smooth on a neighborhood of (a,b)(a,b) inside the relative interiors of the simplices. Proposition: First variations of entropic OT gives

DaVC,ϵ(a,b)[h]=fϵ,h.D_a\mathcal V_{\C,\epsilon}(a,b)[h]=\dotp{f_\epsilon}{h}.

Applying the ordinary delta method to the multinomial CLT N(a^Na)Ga\sqrt N(\hat a_N-a)\Rightarrow G_a gives the one-sample limit fϵ,Ga\dotp{f_\epsilon}{G_a}, whose variance is the displayed covariance formula. Fix a relative-interior neighborhood UU of aa on which the Hessian is bounded. Multinomial concentration gives P(a^NU)CecN\mathbb P(\hat a_N\notin U)\leq Ce^{-cN}. On UU, Taylor’s formula has a centered linear term and a quadratic remainder bounded by Ca^Na2C\norm{\hat a_N-a}^2, whose expectation is O(N1)O(N^{-1}). On the exponentially unlikely complement, the value remains bounded; on a boundary face, its natural definition simply discards zero-weight atoms. This proves the bias bound. The same expansion and bounded multinomial moments give convergence of the rescaled variance. The two-sample formula follows independently for the two marginals.

For the entropy-only convention LCϵ\MKD_\C^\epsilon used in Entropic Regularization for Discrete Measures,

LCϵ(a,b)=VC,ϵ(a,b)ϵH(a)ϵH(b).\MKD_\C^\epsilon(a,b) =\mathcal V_{\C,\epsilon}(a,b)-\epsilon\HD(a)-\epsilon\HD(b).

The marginal entropy derivatives described after Proposition: First variations of entropic OT must therefore be added to the potentials. For the debiased Sinkhorn divergence Lˉcϵ\bar\MK_c^\epsilon, the derivative is instead the difference between cross and self entropic potentials. At the null α=β\alpha=\beta, this first derivative can vanish for the debiased statistic, and second-order limits then become relevant Goldfeld et al., 2024. This is a useful warning: the asymptotic law is determined not only by a rate, but also by the local geometry of the functional.

Three Error Terms When Entropy Estimates Exact OT

If the target is Lc(α,β)\MK_c(\alpha,\beta) but the statistic uses Lcϵ\MK_c^\epsilon, three errors should be kept separate:

Lcϵ(α^n,β^m)Lc(α,β)=Lcϵ(α,β)Lc(α,β)regularization bias+Bn,m(Lcϵ)statistical bias+Zn,m(Lcϵ)centered fluctuation.\MK_c^\epsilon(\hat\alpha_n,\hat\beta_m)-\MK_c(\alpha,\beta) = \underbrace{\MK_c^\epsilon(\alpha,\beta)-\MK_c(\alpha,\beta)}_{\text{regularization bias}} + \underbrace{B_{n,m}(\MK_c^\epsilon)}_{\text{statistical bias}} + \underbrace{Z_{n,m}(\MK_c^\epsilon)}_{\text{centered fluctuation}}.

Although algebraic, this identity is important statistically. It is usually read after choosing ϵ=ϵn\epsilon=\epsilon_n. For fixed ϵ\epsilon, the statistical bias is typically O(n1+m1)O(n^{-1}+m^{-1}) in finite dimension, and the centered fluctuation is OP(n1/2+m1/2)O_{\mathbb P}(n^{-1/2}+m^{-1/2}); boundary events where an empirical finite histogram has a zero entry have exponentially small probability when the population weights are positive. The regularization bias disappears only when ϵ0\epsilon\to0, whereas the variance constants usually deteriorate as ϵ0\epsilon\downarrow0. This is the asymptotic form of the bias--variance tradeoff illustrated in Div.

What Changes in Continuous Spaces

The finite-dimensional formulas above are not merely toy models; they are the cleanest template for the general mechanism. Whenever an OT value is Hadamard differentiable and admits a unique sufficiently regular dual potential ff^\star, the empirical-process CLT and the delta method give a first-order Gaussian limit with one-sample variance Varα(f(X))\operatorname{Var}_\alpha(f^\star(X)). When differentiability fails, as can happen for exact OT because the dual optimizer is not unique or because one studies the degenerate self-distance Lc(α,α)=0\MK_c(\alpha,\alpha)=0, the limit may be non-Gaussian or may occur at a slower non-parametric scale. Entropic regularization smooths the dual potentials and typically restores a conventional first-order CLT for fixed ϵ\epsilon, even under weak assumptions on the cost. If the target is unregularized OT, however, this statistical expansion must still be balanced against the bias Lcϵ(α,β)Lc(α,β)\MK_c^\epsilon(\alpha,\beta)-\MK_c(\alpha,\beta). Thus Sample Complexity gives dimension-dependent magnitudes, while the present section identifies the local asymptotic constants and covariance formulas needed for inference.

Sketching Sinkhorn in Linear Time

The statistical results above explain why entropic OT is attractive at fixed temperature, but they also expose a computational tension. Sampling the marginals replaces the input laws by empirical measures and reduces the problem to finitely many support points; subsampling or coarsening these supports is governed by the sample-complexity estimates of this chapter. Even after the supports are fixed, however, a Sinkhorn sweep still costs O(nm)O(nm), because it repeatedly applies the Gibbs kernel to vectors. Kernel sketching attacks this second bottleneck: it keeps the sampled marginals fixed and samples, or otherwise compresses, the feature representation of the kernel used by Sinkhorn. With a rank RR factorization, NitN_{\rm it} scaling sweeps cost O(Nit(n+m)R)O(N_{\rm it}(n+m)R), which is linear in the support sizes only when both the required rank and the iteration count are controlled. The rank generally depends on dimension, temperature, and target accuracy.

PSD Kernels and Ordinary Feature Sketches

The RKHS/MMD section, Dual RKHS Norms and Maximum Mean Discrepancies, already used positive semidefinite kernels to define Hilbertian discrepancies between measures. Recall that a symmetric kernel k:X×XRk:\Xx\times\Xx\to\RR is positive semidefinite if, for every finite family (xi)i=1n(x_i)_{i=1}^n, the Gram matrix (k(xi,xj))ij(k(x_i,x_j))_{ij} is positive semidefinite. Many scalable kernel methods start from an integral feature representation

k(x,y)=Zϕ(x,z),ϕ(y,z)Rpdξ(z),ϕ:X×ZRp,k(x,y)=\int_Z \dotp{\phi(x,z)}{\phi(y,z)}_{\RR^p}\,d\xi(z), \qquad \phi:\Xx\times Z\to\RR^p,

where ξ\xi is a latent distribution. Drawing z1,,zrξz_1,\ldots,z_r\sim\xi gives

k~r(x,y)=1r=1rϕ(x,z),ϕ(y,z)Rp.\widetilde k_r(x,y) = \frac1r\sum_{\ell=1}^r \dotp{\phi(x,z_\ell)}{\phi(y,z_\ell)}_{\RR^p}.

This is the classical random-feature principle. For data points (xi)i=1n(x_i)_{i=1}^n, the dense Gram matrix K=(k(xi,xj))ijK=(k(x_i,x_j))_{ij} is replaced by K~=ΦΦ\widetilde K=\Phi\Phi^\top, where ΦRn×rp\Phi\in\RR^{n\times rp} is the sampled feature matrix with entries Φi,(,q)=r1/2ϕq(xi,z)\Phi_{i,(\ell,q)}=r^{-1/2}\phi_q(x_i,z_\ell), 1r1\leq \ell\leq r, 1qp1\leq q\leq p. The number rr of sampled latent variables is thus a second sample size, distinct from the number of data points. At this stage no positivity of ϕ\phi is assumed; signed or oscillatory features are exactly what make Fourier sketches useful.

For translation-invariant kernels on Rd\RR^d, Bochner’s theorem gives the standard Fourier sketch. If k(x,y)=κ(xy)k(x,y)=\kappa(x-y) and κ(t)=eiω,tdΛ(ω)\kappa(t)=\int e^{i\dotp{\omega}{t}}\,d\Lambda(\omega), with Λ\Lambda a positive spectral measure, one may use real features ϕ(x,ω)=(cosω,x,sinω,x)\phi(x,\omega)=(\cos\dotp{\omega}{x},\sin\dotp{\omega}{x}). For the Gaussian kernel kσ(x,y)=exp(xy2/(2σ2))k_\sigma(x,y)=\exp(-\norm{x-y}^2/(2\sigma^2)), this corresponds to ωN(0,σ2Id)\omega\sim\mathcal N(0,\sigma^{-2}\Id). This random Fourier feature construction was introduced by Rahimi and Recht to accelerate large-scale kernel machines Rahimi & Recht, 2007; its statistical role for kernel ridge regression and related supervised methods is analyzed, for instance, in Rudi & Rosasco, 2017Avron et al., 2017.

Application to Sinkhorn Kernels

Let

α=i=1naiδxi,β=j=1mbjδyj,Kij=kϵ(xi,yj):=ec(xi,yj)/ϵ.\alpha=\sum_{i=1}^n a_i\delta_{x_i}, \qquad \beta=\sum_{j=1}^m b_j\delta_{y_j}, \qquad K_{ij}=k_\epsilon(x_i,y_j)\eqdef e^{-c(x_i,y_j)/\epsilon}.

Assume here that the weights are positive and that Kij>0K_{ij}>0. The Sinkhorn scaling equations are

u=aKv,v=bKu,u=\frac{a}{Kv}, \qquad v=\frac{b}{K^\top u},

where divisions are componentwise. The factorization used by Sinkhorn is rectangular, because the source points xix_i and target points yjy_j need not coincide. Starting from the same sketch k(x,y)r1=1rϕ(x,z),ϕ(y,z)Rpk(x,y)\simeq r^{-1}\sum_{\ell=1}^r \dotp{\phi(x,z_\ell)}{\phi(y,z_\ell)}_{\RR^p}, set R=rpR=rp and flatten the pair (,q)(\ell,q) into a single column index. Define

(ΦX)i,(,q)=r1/2ϕq(xi,z),(ΦY)j,(,q)=r1/2ϕq(yj,z).(\Phi_X)_{i,(\ell,q)}=r^{-1/2}\phi_q(x_i,z_\ell), \qquad (\Phi_Y)_{j,(\ell,q)}=r^{-1/2}\phi_q(y_j,z_\ell).

Then ΦXRn×R\Phi_X\in\RR^{n\times R}, ΦYRm×R\Phi_Y\in\RR^{m\times R}, and

K~ij=(ΦXΦY)ij=1r=1rϕ(xi,z),ϕ(yj,z)RpKij.\widetilde K_{ij} = (\Phi_X\Phi_Y^\top)_{ij} = \frac1r\sum_{\ell=1}^r \dotp{\phi(x_i,z_\ell)}{\phi(y_j,z_\ell)}_{\RR^p} \simeq K_{ij}.

When the two supports are the same, one has ΦX=ΦY=Φ\Phi_X=\Phi_Y=\Phi and this reduces to the symmetric Gram approximation K~=ΦΦ\widetilde K=\Phi\Phi^\top. In the rectangular case, if KK is replaced by the rank-RR factorization K~=ΦXΦY\widetilde K=\Phi_X\Phi_Y^\top, the two matrix-vector products are evaluated as

K~v=ΦX(ΦYv),K~u=ΦY(ΦXu),\widetilde K v=\Phi_X(\Phi_Y^\top v), \qquad \widetilde K^\top u=\Phi_Y(\Phi_X^\top u),

so each scaling sweep costs O((n+m)R)O((n+m)R) operations and stores only O((n+m)R)O((n+m)R) numbers. The difficulty is that Sinkhorn is not a generic kernel method: it divides by K~v\widetilde K v and, in dual variables, applies logarithms. Thus one needs K~ij>0\widetilde K_{ij}>0, preferably with a relative error on the entries, because ϵlogKij-\epsilon\log K_{ij} is the effective cost. This is a more stringent requirement than the usual spectral or Frobenius approximation of a PSD Gram matrix.

The next proposition is not a substitute for the sharp low-rank Sinkhorn theory. It is an elementary concentration lemma whose role is to isolate the two facts that matter for Sinkhorn: positivity of the approximate kernel and accuracy after taking logarithms. The full algorithmic statements rely on more refined Nyström or Gaussian-kernel approximations together with stability of Sinkhorn scaling Altschuler et al., 2018Altschuler et al., 2019.

Proof

For a fixed pair (i,j)(i,j), Hoeffding’s inequality applied to bounded variables in [M,M][-M,M] gives

P(K~ijKij>η)2exp(rη22M2).\mathbb P\left(|\widetilde K_{ij}-K_{ij}|>\eta\right) \leq 2\exp\left(-\frac{r\eta^2}{2M^2}\right).

A union bound over the nmnm entries gives the first claim. If maxijK~ijKijη<κ\max_{ij}|\widetilde K_{ij}-K_{ij}|\leq\eta<\kappa, then K~ijκη>0\widetilde K_{ij}\geq\kappa-\eta>0. Writing K~ij=Kij(1+sij)\widetilde K_{ij}=K_{ij}(1+s_{ij}), one has sijη/κ1/2|s_{ij}|\leq\eta/\kappa\leq1/2, hence log(1+sij)2sij|\log(1+s_{ij})|\leq2|s_{ij}|, which proves the absolute logarithmic bound. Applying the same Hoeffding argument to ζij(Z)\zeta_{ij}(Z), whose expectation is 1, gives the relative estimate. On the corresponding event, K~ij=Kij(1+sij)\widetilde K_{ij}=K_{ij}(1+s_{ij}) with sijδ1/2|s_{ij}|\leq\delta\leq1/2, so positivity and ϵlog(1+sij)2ϵδ\epsilon|\log(1+s_{ij})|\leq2\epsilon\delta follow.

This proposition pinpoints the obstruction. When ϵ\epsilon is small or the data diameter is large, κ=minKij\kappa=\min K_{ij} can be exponentially small, so an ordinary signed sketch may need a very large rr just to keep all entries positive. What Sinkhorn really needs is relative control of the kernel, equivalently additive control of the sketched costs. Positive features do not solve the whole approximation problem, but they remove the most basic failure mode, negative entries in K~\widetilde K. The log-normal positive features below are unbounded, so this Hoeffding proposition does not apply to them without truncation or a different concentration argument.

Available worst-case bounds retain explicit dimension dependence. For the quadratic cost on points contained in a radius-RR ball of Rd\RR^d, the Nyström analysis of Altschuler, Bach, Rudi and Niles-Weed Altschuler et al., 2019 studies Kij=exp(ηxixj2)K_{ij}=\exp(-\eta\norm{x_i-x_j}^2) and bounds the effective rank by a quantity of the form

r(X,η,ϵ0)(ηR2+log(n/ϵ0))dr_\ast(X,\eta,\epsilon_0) \lesssim \left(\eta R^2+\log(n/\epsilon_0)\right)^d

up to universal constants and harmless logarithms. Thus the resulting Sinkhorn computation is near-linear in nn once this rank is fixed, but the required rank is polynomial in the inverse kernel bandwidth, with exponent dd and only logarithmic dependence on nn in this bound. When the same machinery is used to approximate unregularized quadratic OT to additive accuracy τ\tau, one takes η\eta of order τ1logn\tau^{-1}\log n, so the complexity contains a factor τO(d)\tau^{-O(d)}, often summarized informally as an ϵd\epsilon^{-d}-type dependence on the target precision. The same work also explains why this can be much better on data with lower intrinsic dimension.

Positive Features and Total Positivity

A positive feature sketch asks for more than positive semidefiniteness. In finite dimension, let

DNNn={ARn×n:A0, Aij0},CPn={BB:BR+n×q for some q1}.\mathrm{DNN}_n=\{A\in\RR^{n\times n}: A\succeq0,\ A_{ij}\geq0\}, \qquad \mathrm{CP}_n=\{BB^\top: B\in\RR_+^{n\times q}\hbox{ for some }q\geq1\}.

Matrices in DNNn\mathrm{DNN}_n are doubly nonnegative, while matrices in CPn\mathrm{CP}_n are completely positive. One always has CPnDNNn\mathrm{CP}_n\subset\mathrm{DNN}_n, with equality for n4n\leq4 and strict inclusion for n5n\geq5 Berman & Shaked-Monderer, 2003Burer et al., 2009. Checking membership in the completely positive cone is computationally hard: strong membership is NP-hard, and weak membership for the completely positive cone and its copositive dual is NP-hard Dickinson & Gijben, 2014.

For kernels, it is useful to distinguish the two corresponding notions. A kernel kk is doubly positive if every finite Gram matrix belongs to DNNn\mathrm{DNN}_n, i.e. if kk is PSD and pointwise nonnegative. It is totally positive in the positive-feature sense if every finite Gram matrix belongs to CPn\mathrm{CP}_n. This terminology should not be confused with Karlin’s classical total positivity, which concerns nonnegativity of all minors of an ordered kernel Karlin, 1968; it is a complete-positive-type condition, studied for instance in DeCorte et al., 2022.

The positive-feature construction of Scetbon and Cuturi Scetbon & Cuturi, 2020 makes the reason for this definition direct. If

k(x,y)=Zϕ(x,z),ϕ(y,z)Rpdξ(z),ϕ(x,z)R+p,k(x,y)=\int_Z\dotp{\phi(x,z)}{\phi(y,z)}_{\RR^p}\,d\xi(z), \qquad \phi(x,z)\in\RR_+^p,

then every finite Gram matrix is a conic combination of rank-one matrices vvvv^\top with v0v\geq0, and hence is completely positive. Conversely, on a fixed finite set, KCPnK\in\mathrm{CP}_n is exactly the existence of a nonnegative factorization K=BBK=BB^\top. Thus total positivity is the relevant finite-sample kernel condition behind positive random features. For rectangular source--target matrices, one applies the same statement to the Gram matrix on the union of the sampled source and target points and then extracts its n×mn\times m cross block.

Proof

For sums, concatenate the nonnegative features and rescale them by the square roots of the nonnegative coefficients. For products, tensor the nonnegative features:

ϕ1(x,z1),ϕ1(y,z1)ϕ2(x,z2),ϕ2(y,z2)=ϕ1(x,z1)ϕ2(x,z2),ϕ1(y,z1)ϕ2(y,z2).\dotp{\phi_1(x,z_1)}{\phi_1(y,z_1)} \dotp{\phi_2(x,z_2)}{\phi_2(y,z_2)} = \dotp{\phi_1(x,z_1)\otimes\phi_2(x,z_2)} {\phi_1(y,z_1)\otimes\phi_2(y,z_2)}.

The tensor feature remains entrywise nonnegative. A measurable mixture is a limit of conic combinations in the closed completely positive cone on every finite point set. The autocorrelation formula is already a positive feature representation with latent variable uu and scalar feature g(ux)g(u-x).

The Gaussian Gibbs kernels used by Sinkhorn admit an especially simple positive feature representation, which is the construction exploited by Scetbon and Cuturi Scetbon & Cuturi, 2020. The same construction extends to the generalized Gaussian kernels exyp/ϵe^{-\norm{x-y}^p/\epsilon} for 0<p20<p\leq2, because these kernels are mixtures of Gaussian kernels. This is the Schoenberg--Bernstein mechanism behind the positive definiteness of radial stable kernels Schoenberg, 1938Berg et al., 1984.

Proof

The function sesas\mapsto e^{-s^a} is completely monotone for 0<a10<a\leq1. Bernstein’s theorem therefore gives the Laplace-transform representation of a positive aa-stable random variable SaS_a. The scaling Λ=ϵ1/aSa\Lambda=\epsilon^{-1/a}S_a gives

EetΛ=exp(taϵ),\mathbb E e^{-t\Lambda} = \exp\left(-\frac{t^a}{\epsilon}\right),

which is the displayed representation by νa,ϵ\nu_{a,\epsilon}. Applying this identity with t=xy2t=\norm{x-y}^2 gives the Gaussian-mixture formula.

Conditionally on Λ=λ\Lambda=\lambda, the Gaussian moment-generating function gives

EZ(ϕp(x,λ,Z)ϕp(y,λ,Z))=exp(qx(λ)2qy(λ)2)EZ(expZ,qx(λ)+qy(λ))=exp(12qx(λ)qy(λ)2)=exp(λxy2).\begin{aligned} \mathbb E_Z\left(\phi_p(x,\lambda,Z)\phi_p(y,\lambda,Z)\right) &= \exp\left(-\norm{q_x(\lambda)}^2-\norm{q_y(\lambda)}^2\right) \mathbb E_Z\left(\exp\dotp{Z}{q_x(\lambda)+q_y(\lambda)}\right) \\ &= \exp\left(-\frac12\norm{q_x(\lambda)-q_y(\lambda)}^2\right) = \exp\left(-\lambda\norm{x-y}^2\right). \end{aligned}

Taking the expectation over Λ\Lambda gives kp,ϵk_{p,\epsilon}. Since ϕp\phi_p is positive, this is a positive-feature representation; hence every finite Gram matrix is completely positive. The Monte-Carlo sketch is just the empirical average of this representation.

Choosing x0x_0 near the data can reduce the dynamic range of the features without changing the kernel. More generally, Gaussian scale mixtures give many radial totally positive kernels, including standard covariance families used in Gaussian processes Rasmussen & Williams, 2006.

The gap between doubly positive and totally positive kernels is real. On the circle, consider

kλ(x,y)=λ+cos2(xy2),λ>0.k_\lambda(x,y)=\lambda+\cos^2\left(\frac{x-y}{2}\right), \qquad \lambda>0.

This kernel is PSD and strictly positive pointwise, since cos2(t/2)=(1+cost)/2\cos^2(t/2)=(1+\cos t)/2 has nonnegative Fourier coefficients and λ\lambda adds a positive constant mode. For λ=1/20\lambda=1/20, however, its Gram matrix on five equally spaced points is not completely positive. The Horn copositive matrix HH satisfies H,B0\langle H,B\rangle\geq0 for every completely positive matrix BB, but the five-point Gram matrix KK gives

H,K=214552<0.\langle H,K\rangle = \frac{21}{4}-\frac{5\sqrt5}{2}<0.

Thus KDNN5CP5K\in\mathrm{DNN}_5\setminus\mathrm{CP}_5, and the kernel is doubly positive but not totally positive.

Figure Notebook-code displays this finite certificate and the separating witness.

<IPython.core.display.Image object>

Figure Notebook-code displays this finite certificate.

Interactive panel. Change the offset to compare an entrywise positive PSD kernel matrix with the Horn certificate obstructing complete positivity.

Positive Sketches for Sinkhorn

For the pp-power cost cp(x,y)=xypc_p(x,y)=\norm{x-y}^p, with 0<p20<p\leq2, the Sinkhorn Gibbs kernel is the generalized Gaussian kernel

kp,ϵ(x,y)=exp(xypϵ).k_{p,\epsilon}(x,y) = \exp\left(-\frac{\norm{x-y}^p}{\epsilon}\right).

The factor 1/p1/p, often used in the definition of WppW_p^p, only rescales ϵ\epsilon. For p1p\geq1, this is the usual pp-Wasserstein power cost; for 0<p<10<p<1, it should be read simply as a concave power transport cost. Proposition Proposition: Generalized Gaussian Positive Features gives a positive factorization of this kernel. With rr draws (Λ,Z)νp/2,ϵN(0,Idd)(\Lambda_\ell,Z_\ell)\sim\nu_{p/2,\epsilon}\otimes\mathcal N(0,\Id_d), define the normalized features

φ(x)=r1/2ϕp(x,Λ,Z),1r.\varphi_\ell(x)=r^{-1/2}\phi_p(x,\Lambda_\ell,Z_\ell), \qquad 1\leq \ell\leq r .

They satisfy

k~p,ϵ(x,y)==1rφ(x)φ(y)0,E[k~p,ϵ(x,y)]=kp,ϵ(x,y).\widetilde k_{p,\epsilon}(x,y) = \sum_{\ell=1}^r\varphi_\ell(x)\varphi_\ell(y) \geq0, \qquad \mathbb E[\widetilde k_{p,\epsilon}(x,y)]=k_{p,\epsilon}(x,y).

The corresponding matrices (ΦX)i=φ(xi)(\Phi_X)_{i\ell}=\varphi_\ell(x_i) and (ΦY)j=φ(yj)(\Phi_Y)_{j\ell}=\varphi_\ell(y_j) give a positive factorization K~=ΦXΦY\widetilde K=\Phi_X\Phi_Y^\top. Sinkhorn can therefore be run safely on the sketched kernel, and the resulting plan is represented as

P~=diag(u)ΦXΦYdiag(v)=LR,L=diag(u)ΦX,R=diag(v)ΦY.\widetilde P = \diag(u)\Phi_X\Phi_Y^\top\diag(v) = LR^\top, \qquad L=\diag(u)\Phi_X, \quad R=\diag(v)\Phi_Y .

The output is the exact entropic coupling for the sketched effective cost c~p,ϵ=ϵlogk~p,ϵ\widetilde c_{p,\epsilon}=-\epsilon\log\widetilde k_{p,\epsilon}, and it approximates the original WpW_p-type entropic problem only insofar as this logarithmic kernel approximation is accurate. This route is complementary to worst-case near-linear Sinkhorn analyses Altschuler et al., 2017, low-rank Gaussian-kernel approximations for quadratic transport Altschuler et al., 2018, and factored-coupling models in which the coupling itself is constrained to have low rank Scetbon et al., 2021.

Figure Notebook-code compares the exact Sinkhorn plan with positive-feature approximations and displays the associated approximation of the ground cost.

<IPython.core.display.Image object>

Figure Notebook-code shows the object that is changed by this approximation. The top row shows the entropic couplings, while the bottom row shows the matching effective costs c~ϵ(x,y)=ϵlogk~ϵ(x,y)\widetilde c_\epsilon(x,y)=-\epsilon\log\widetilde k_\epsilon(x,y) with black level sets. The dense Gaussian Gibbs kernel gives the usual entropic plan and the quadratic cost xy2|x-y|^2. A sufficiently rich rank-40 positive-feature sketch remains visually close to both, whereas rank-10 and rank-3 sketches still enforce the prescribed marginals after Sinkhorn scaling but progressively lose the geometric concentration of the true coupling because the logarithmic kernel approximation has become too coarse.

Interactive panel. Change the regularization and random seed to compare the exact Gibbs kernel with positive low-rank sketches and their effective logarithmic costs.

Connection with Linear Attention

The same algebra appears in transformer attention, studied later from a continuous-depth transport viewpoint in Evolution in Depth of Transformers. A softmax attention matrix has entries proportional to exp(qi,kj)\exp(\dotp{q_i}{k_j}); linear-attention methods replace this positive kernel by features Φ(qi)Φ(kj)\Phi(q_i)^\top\Phi(k_j), so that attention can be applied as Φ(Q)(Φ(K)V)\Phi(Q)(\Phi(K)^\top V) rather than by forming the full n×nn\times n matrix Katharopoulos et al., 2020Choromanski et al., 2021Wang et al., 2020Xiong et al., 2021. Sinkhorn sketching is the transport analogue of this idea: replace the Gibbs matrix by a positive feature factorization, keep only the scaling vectors and feature factors, and control the approximation in the logarithmic scale relevant to entropic potentials. The transport case is more constrained, however, since the approximate kernel is reused inside nonlinear normalizations until the prescribed marginals are reached for the sketched problem.

References
  1. Berry, A. C. (1941). The Accuracy of the Gaussian Approximation to the Sum of Independent Variates. Transactions of the American Mathematical Society, 49(1), 122–136.
  2. Esseen, C.-G. (1942). On the Liapunoff Limit of Error in the Theory of Probability. Arkiv for Matematik, Astronomi Och Fysik, 28A(9), 1–19.
  3. Chen, L. H. Y., Goldstein, L., & Shao, Q.-M. (2011). Normal Approximation by Stein’s Method. Springer.
  4. Bobkov, S. G. (2018). Berry–Esseen bounds and Edgeworth expansions in the central limit theorem for transport distances. Probability Theory and Related Fields, 170, 229–262.
  5. Rio, E. (2011). Asymptotic constants for minimal distance in the central limit theorem. Electronic Communications in Probability, 16, 96–103.
  6. Petrov, V. V. (1975). Sums of Independent Random Variables (Vol. 82). Springer-Verlag. 10.1007/978-3-642-65809-9
  7. Bhattacharya, R. N., & Ranga Rao, R. (2010). Normal Approximation and Asymptotic Expansions (Vol. 64). Society for Industrial. 10.1137/1.9780898719895
  8. Kolassa, J. E., & McCullagh, P. (1990). Edgeworth Series for Lattice Distributions. The Annals of Statistics, 18(2), 981–985. 10.1214/aos/1176347637
  9. Vallender, S. S. (1974). Calculation of the Wasserstein Distance Between Probability Distributions on the Line. Theory of Probability and Its Applications, 18(4), 784–786. 10.1137/1118101
  10. van der Vaart, A. W., & Wellner, J. A. (1996). Weak Convergence and Empirical Processes: With Applications to Statistics. Springer. 10.1007/978-1-4757-2545-2
  11. Dudley, R. M. (1969). The speed of mean Glivenko-Cantelli convergence. Annals of Mathematical Statistics, 40(1), 40–50.
  12. Weed, J., & Bach, F. (2019). Sharp asymptotic and finite-sample rates of convergence of empirical measures in Wasserstein distance. Bernoulli, 25(4A), 2620–2648. 10.3150/18-BEJ1065
  13. Ramdas, A., Trillos, N. G., & Cuturi, M. (2017). On Wasserstein two-sample testing and related families of nonparametric tests. Entropy, 19(2), 47.
  14. Fournier, N., & Guillin, A. (2015). On the rate of convergence in Wasserstein distance of the empirical measure. Probability Theory and Related Fields, 162(3–4), 707–738.
  15. Singh, S., & Póczos, B. (2018). Minimax Distribution Estimation in Wasserstein Distance. arXiv Preprint arXiv:1802.08855. https://arxiv.org/abs/1802.08855