Accessibility navigation

Coupling matrix manifolds assisted optimization for optimal transport problems

Shi, D., Gao, J., Hong, X., Choy, S. T. B. and Wang, Z. (2021) Coupling matrix manifolds assisted optimization for optimal transport problems. Machine Learning. ISSN 1573-0565

[img] Text - Accepted Version
· Restricted to Repository staff only until 1 January 2022.


It is advisable to refer to the publisher's version if you intend to cite from this work. See Guidance on citing.

To link to this item DOI: 10.1007/s10994-020-05931-2


Optimal transport (OT) is a powerful tool for measuring the distance between two probability distributions. In this paper, we develop a new manifold named the coupling matrix manifold (CMM), where each point on CMM can be regarded as a transportation plan of the OT problem. We firstly explore the Riemannian geometry of CMM with the metric expressed by the Fisher information. These geometrical features of CMM have paved the way for developing numerical Riemannian optimization algorithms such as Riemannian gradient descent and Riemannian trust region algorithms, forming an essential optimization method for all types of OT problems. The proposed method is then applied to solve several OT problems studied by recent literature. For the classic OT problem and its entropy regularized variant, the OT solution generated from our method is comparable to that from the classic algorithms (i.e. Linear programming and Sinkhorn algorithms), while for other types of non-entropy regularized OT problems our method outperforms other state-of-the-art algorithms which don’t incorporate the geometric information of the OT feasible space.

Item Type:Article
Divisions:Science > School of Mathematical, Physical and Computational Sciences > Department of Computer Science
ID Code:94589

University Staff: Request a correction | Centaur Editors: Update this record

Page navigation