1: \begin{abstract}
2:
3: %{\bf Brahim's original abstract}
4: %Optimal transport distances play a major role in machine learning and computer vision, and research on algorithms to compute them is a hot topic in these disciplines with numerous recent advances. In particular, regularizing the Optimal transport problem with an entropic penalization proved to have several computational advantages, and led to the well-known Sinkhorn algorithm and its greedy version Greenkhorn. In this paper, we propose a general framework of stochastic solvers for this kind of problems: Stochastic Sinkhorn. We show how Greenkhorn is a particular instance of this framework and we prove convergence rates for several other cases. We also present numerical experiments where a simple and intuitive choice of parameters for Stochastic Sinkhorn leads to a slight improvement in short-term speed compared to regular Greenkhorn for regimes of low penalization.
5: %
6: %{\bf Rob Alternative:}
7:
8: Optimal transport (OT) distances are finding evermore applications in machine learning and computer vision, but their wide spread use in larger-scale problems is impeded by their high computational cost. In this work we develop a family of fast and practical stochastic algorithms for solving the optimal transport problem with an entropic penalization. This work extends the recently developed Greenkhorn algorithm, in the sense that, the Greenkhorn algorithm is a limiting case of this family. We also provide a simple and general convergence theorem for all algorithms in the class, with rates that match the best known rates of Greenkorn and the Sinkhorn algorithm, and conclude with numerical experiments that show under what regime of penalization the new stochastic methods are faster than the aforementioned methods.
9: \end{abstract}