arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.00730math.OCcs.LG

Faster Unbalanced Optimal Transport: Translation invariant Sinkhorn and 1-D Frank-Wolfe

DMA, ENS, PSL(巴黎文理研究大学高等师范学校数学系);LIGM, UPEM(巴黎东马恩拉瓦莱大学加斯帕尔·蒙日信息实验室);CNRS, DMA, ENS, PSL(法国国家科学研究中心、巴黎文理研究大学高等师范学校数学系)

Thibault Séjourné, François-Xavier Vialard, Gabriel Peyré

英文摘要:

Unbalanced optimal transport (UOT) extends optimal transport (OT) to take into account mass variations to compare distributions. This is crucial to make OT successful in ML applications, making it robust to data normalization and outliers. The baseline algorithm is Sinkhorn, but its convergence speed might be significantly slower for UOT than for OT. In this work, we identify the cause for this deficiency, namely the lack of a global normalization of the iterates, which equivalently corresponds to a translation of the dual OT potentials. Our first contribution leverages this idea to develop a provably accelerated Sinkhorn algorithm (coined 'translation invariant Sinkhorn') for UOT, bridging the computational gap with OT. Our second contribution focusses on 1-D UOT and proposes a Frank-Wolfe solver applied to this translation invariant formulation. The linear oracle of each steps amounts to solving a 1-D OT problems, resulting in a linear time complexity per iteration. Our last contribution extends this method to the computation of UOT barycenter of 1-D measures. Numerical simulations showcase the convergence speed improvement brought by these three approaches.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑