Jul, 2023

使用镜像下降与共轭梯度的高效准确最优输运

TL;DR我们设计了一种新的算法来计算最优运输成本,该算法结合了熵最优运输、镜像下降和共轭梯度文献,能够高效地在 GPU 上实现,表现出更快的迭代收敛速度,并能适应高熵边际分布的复杂优化问题,我们在 MNIST 数据集上进行了实验,结果显示该算法是实践者最佳运输工具包中有用的补充。