统一分发函数的MATLAB代码
Matlab
的
Wasserstein
距离代码
这是一个紧凑的
Matlab
代码,用于计算一维概率分布的
1-和
2-Wasserstein
距离。
有关
Wasserstein
距离的一般定义,请参阅
。
此实现基于以下事实:对于给定的分布u和v
,1-Wasserstein
距离可以写为
和
2-Wasserstein
距离为
这里
和
参考u和v的累积密度函数,以及由下式定义的相应伪逆累积密度函数
代码假定(与
不同)
u和v是离散且均匀的概率分布。
在这种情况下,存在样本使得任何u分布的随机变量满足所有k
。
这些样本是函数的输入,让我们不失一般性地假设它们越来越排序,然后累积分布函数及其伪逆由阶跃函数给出
该代码已在
Matlab
R2017a
中进行了测试,并针对(在
1-Wasserstein
距离的情况下)和
.
后一种代码允许计算一般的
p-Wasserstein
距离,但在计算上比所描述情况的当前代码更复杂。
一些参考
Carrillo,
JA
和
G.
Toscani。
“非线性扩散方程的
Wasserstein
度量和大时间
2021-08-06 16:44:06
4KB
系统开源
1