最优传输性能研究
数据结构与算法
2020-07-07 v1 机器学习
摘要
我们研究了高效计算最优传输（OT）距离的问题，该问题等价于二部图中的节点容量受限最小费用最大流问题。我们比较了在多个领域数据上计算 OT 距离的运行时，例如几何形状的合成数据、文档中词元的嵌入以及图像中的像素。我们表明，在实践中，诸如网络单纯形和基于增广路径的组合方法可以持续优于基于数值矩阵缩放的方法（如 Sinkhorn [Cuturi'13] 和 Greenkhorn [Altschuler et al'17]），即使在低精度情形下，也可实现高达数个数量级的加速。最后，我们提出了一种改进经典 Kuhn-Munkres 算法的新的组合算法。
引用
@article{arxiv.2005.01182,
title = {A Study of Performance of Optimal Transport},
author = {Yihe Dong and Yu Gao and Richard Peng and Ilya Razenshteyn and Saurabh Sawlani},
journal= {arXiv preprint arXiv:2005.01182},
year = {2020}
}