cuRegOT

新的 GPU 求解器 cuRegOT 加速了机器学习的最优传输

研究人员开发了 cuRegOT，这是一种新的 GPU 加速求解器，旨在克服大规模机器学习应用中最优传输 (OT) 的计算挑战。该求解器通过引入摊销符号分析和异步迭代生成等优化，解决了现有方法（如 Sinkhorn 算法和稀疏加低秩拟牛顿法）的局限性。数值实验表明，cuRegOT 在各种基准任务上的表现显著优于当前最先进的 GPU 求解器。