Coupling matrix manifolds assisted optimization for optimal transport problems

Shi, D., Gao, J., Hong, X. ORCID: https://orcid.org/0000-0002-6832-2298, Choy, S. T. B. and Wang, Z. (2021) Coupling matrix manifolds assisted optimization for optimal transport problems. Machine Learning, 110. pp. 533-558. ISSN 1573-0565 doi: 10.1007/s10994-020-05931-2

Abstract/Summary

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
URI	https://reading-clone.eprints-hosting.org/id/eprint/94589
Identification Number/DOI	10.1007/s10994-020-05931-2
Refereed	Yes
Divisions	Science > School of Mathematical, Physical and Computational Sciences > Department of Computer Science
Publisher	Springer
Download/View statistics	View download statistics for this item