Work

Fast Computation of Optimal Transport via Entropy-Regularized Extragradient Methods

Year: 2025
Type: article
Abstract: . Efficient computation of the optimal transport distance between two distributions serves as an algorithm subroutine that empowers various applications. This paper develops a scalable first-order optimization-based method that computes optimal transport to within [Formula: see text] additive accuracy with runtime [Formula: see text], where [Formula: see text] denotes the dimension of the probability distributions of interest. Our algorithm achieves the state-of-the-art computational guarantees among all first-order methods, while exhibiting favorable numerical performance compared to classical algorithms like Sinkhorn and Greenkhorn. Underlying our algorithm designs are two key elements: (a) converting the original problem into a bilinear minimax problem over probability distributions; (b) exploiting the extragradient idea—in conjunction with entropy regularization and adaptive learning rates—to accelerate convergence.
Language: English
Open Access status: green
OpenAlex ID: W4411432493
FWCI: 1.827
Cites: 28
Cited by: 2
Related to: 10