跳到正文
格致开物MATHWIKI

瓦瑟斯坦距离

内容维护:格致开物 · 编写与核验 · 修订记录 · 纠错

瓦瑟斯坦距离比较两个概率分布时,不仅考虑概率质量相差多少,还考虑把质量从一个位置移动到另一个位置所需的代价。它常用于最优运输:想象把一种分布看成物资的供给位置,另一种看成需求位置;不同运输安排都必须满足给定的供给与需求。

耦合与运输成本

在带距离 d 的度量空间上,给定概率测度 μ,ν。一个耦合 π 是空间乘积上的概率测度,使其第一、第二边缘分布分别为 μ,ν。所有这样的耦合构成集合 Π(μ,ν)。对 p≥1,定义 Wp(μ,ν)=(infπ∈Π(μ,ν)∫d(x,y)pdπ(x,y))1/p. 当空间为波兰空间并选定参考点 x0,如果两分布的 p 阶距离矩均有限,即 ∫d(x,x0)pdμ(x)<∞、∫d(y,x0)pdν(y)<∞,则上式有限,并在这些测度的集合上构成距离。更一般的定义可能给出无穷值。[1][2]

一次能证明最优的运输

在实数轴上,令 μ=12δ0+12δ2,ν=12δ1+12δ3,其中 δa 把全部质量放在点 a。把零处的一半质量送到一,把二处的一半质量送到三,是合法耦合,总距离成本为 12|0−1|+12|2−3|=1. 故 W1≤1。另一方面,对任意耦合 (X,Y),三角不等式给出 𝔼|X−Y|≥|𝔼[X−Y]|=|1−2|=1. 下界与所给方案达到的成本相同,所以 W1=1。这一步证明了最优,而不是只给出一个可能的运输方案。

数轴上半份质量由0移到1,另一半由2移到3
本例的两个运输箭头。每次运输距离为一,合计成本为一。

如果把两分布之间的所有概率质量都独立配对,也得到一种耦合,但平均距离为 (1+3+1+1)/4=3/2,比一更大。求最小值时不能直接把独立耦合当成最优。

与其他距离的区别

两个点质量 δ0 和 δϵ(ϵ>0)的全变差距离为一,因为集合 {0} 能完全区分二者;但 W1(δ0,δϵ)=ϵ。前者衡量集合概率的最大差异,后者使用空间距离,因此 ϵ→0 时所得结论不同。

一维分布常可用分位数计算:若 U 在 (0,1) 上均匀,取广义分位数 Fμ−1(U),Fν−1(U),此共同分位数耦合在适当矩条件下达到 Wp 的最小值。它是特定的一维结论;更高维的最优运输通常不能由逐坐标排序代替。

当极小值可以达到时,相应耦合称为最优耦合;定义里的下确界本身不要求事先知道最优方案。估计数据的瓦瑟斯坦距离时,样本量、距离单位及高维计算代价都会影响结果,距离数值也必须结合实际度量的单位解释。

参考资料

  1. ↑ Cédric Villani,Optimal Transport: Old and New,第 6 章,定义 6.1、6.3,访问于 2026 年 11 月 8 日。
  2. ↑ Advanced Probability,第 5.4 节,维也纳大学概率论讲义,访问于 2026 年 11 月 8 日。