Geometric Characteristics of the Wasserstein Metric on SPD(n) and Its Applications on Data Processing

Entropy (Basel). 2021 Sep 14;23(9):1214. doi: 10.3390/e23091214.

Abstract

The Wasserstein distance, especially among symmetric positive-definite matrices, has broad and deep influences on the development of artificial intelligence (AI) and other branches of computer science. In this paper, by involving the Wasserstein metric on SPD(n), we obtain computationally feasible expressions for some geometric quantities, including geodesics, exponential maps, the Riemannian connection, Jacobi fields and curvatures, particularly the scalar curvature. Furthermore, we discuss the behavior of geodesics and prove that the manifold is globally geodesic convex. Finally, we design algorithms for point cloud denoising and edge detecting of a polluted image based on the Wasserstein curvature on SPD(n). The experimental results show the efficiency and robustness of our curvature-based methods.

Keywords: Wasserstein metric; curvature; image edge detecting; point cloud denoising; symmetric positive-definite matrix.