Skip to yearly menu bar Skip to main content


Poster

Improved Rate of First Order Algorithms for Entropic Optimal Transport

Yiling Luo · Yiling Xie · Xiaoming Huo

Auditorium 1 Foyer 114

Abstract: This paper improves the state-of-the-art rate of a first-order algorithm for solving entropy regularized optimal transport. The resulting rate for approximating the optimal transport (OT) has been improved from O~(n2.5/ϵ) to O~(n2/ϵ), where n is the problem size and ϵ is the accuracy level. In particular, we propose an accelerated primal-dual stochastic mirror descent algorithm with variance reduction. Such special designs help us improve the rate compared to other accelerated primal-dual algorithms. We further propose a batch version of our stochastic algorithm, which improves the computational performance through parallel computing.To compare, we prove that the computational complexity of the Stochastic Sinkhorn algorithm is O~(n2/ϵ2), which is slower than our accelerated primal-dual stochastic mirror algorithm. Experiments are done using synthetic and real data, and the results match our theoretical rates.Our algorithm may inspire more research to develop accelerated primal-dual algorithms that have rate O~(n2/ϵ) for solving OT.

Live content is unavailable. Log in and register to view live content