141
v1v2v3v4 (latest)

Accelerated Sinkhorn Algorithms for Partial Optimal Transport

Nghia Thu Truong
Qui Phu Pham
Quang Nguyen
Dung Luong
Mai Tran
Main:3 Pages
7 Figures
Bibliography:2 Pages
1 Tables
Appendix:7 Pages
Abstract

Partial Optimal Transport (POT) addresses the problem of transporting only a fraction of the total mass between two distributions, making it suitable when marginals have unequal size or contain outliers. While Sinkhorn-based methods are widely used, their complexity bounds for POT remain suboptimal and can limit scalability. We introduce Accelerated Sinkhorn for POT (ASPOT), which integrates alternating minimization with Nesterov-style acceleration in the POT setting, yielding a complexity of O(n7/3ε5/3)\mathcal{O}(n^{7/3}\varepsilon^{-5/3}). We also show that an informed choice of the entropic parameter γ\gamma improves rates for the classical Sinkhorn method. Experiments on real-world applications validate our theories and demonstrate the favorable performance of our proposed methods.

View on arXiv
Comments on this paper