Processing math: 100%
Skip to yearly menu bar Skip to main content


Poster

A Truncated Newton Method for Optimal Transport

Mete Kemertas · Amir-massoud Farahmand · Allan Jepson

Hall 3 + Hall 2B #378
[ ] [ Project Page ]
Thu 24 Apr 7 p.m. PDT — 9:30 p.m. PDT

Abstract: Developing a contemporary optimal transport (OT) solver requires navigating trade-offs among several critical requirements: GPU parallelization, scalability to high-dimensional problems, theoretical convergence guarantees, empirical performance in terms of precision versus runtime, and numerical stability in practice. With these challenges in mind, we introduce a specialized truncated Newton algorithm for entropic-regularized OT. In addition to proving that locally quadratic convergence is possible without assuming a Lipschitz Hessian, we provide strategies to maximally exploit the high rate of local convergence in practice. Our GPU-parallel algorithm exhibits exceptionally favorable runtime performance, achieving high precision orders of magnitude faster than many existing alternatives. This is evidenced by wall-clock time experiments on 24 problem sets (12 datasets × 2 cost functions). The scalability of the algorithm is showcased on an extremely large OT problem with n106, solved approximately under weak entopric regularization.

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