AAAIDec, 2023

关于部分最优输运:修正 Sinkhorn 和高效梯度方法的不可行性

TL;DR本文研究了两个非平衡度量之间的部分最优输运(Partial Optimal Transport,POT)问题及其在颜色转移或领域适应等各种人工智能任务中的应用。我们首先通过理论和实验证明了现有的 Sinkhorn 算法在 POT 问题上的不可行性,进而提出了一种新的 POT 算法来解决这一问题,并且提供了几种近似 POT 问题的一阶方法,其中包括了近似解在 ε 范围内的 Adaptive Primal-Dual Accelerated Gradient Descent(APDAGD)算法以及具有最佳计算复杂度的 Dual Extrapolation 算法。同时,我们还展示了 POT 相比标准 OT 的灵活性以及我们算法在两个非平衡边缘分布的实际应用中的实用性。