Geomstats: A Python Package for Riemannian Geometry in Machine Learning

Abstract

We introduce Geomstats, an open-source Python toolbox for computations and statistics on nonlinear manifolds, such as hyperbolic spaces, spaces of symmetric positive definite matrices, Lie groups of transformations, and many more. We provide object-oriented and extensively unit-tested implementations. Among others, manifolds come equipped with families of Riemannian metrics, with associated exponential and logarithmic maps, geodesics and parallel transport. Statistics and learning algorithms provide methods for estimation, clustering and dimension reduction on manifolds. All associated operations are vectorized for batch computation and provide support for different execution backends, namely NumPy, PyTorch and TensorFlow, enabling GPU acceleration. This paper presents the package, compares it with related libraries and provides relevant code examples. We show that Geomstats provides reliable building blocks to foster research in differential geometry and statistics, and to democratize the use of Riemannian geometry in machine learning applications. The source code is freely available under the MIT license at \url{geomstats.ai}.

References in corpus (6)

• Geometric deep learning: going beyond Euclidean data
• Manopt, a Matlab toolbox for optimization on manifolds
• Popularity versus Similarity in Growing Networks
• Shape Analysis on Lie Groups with Applications in Computer Animation
• Möbius deconvolution on the hyperbolic plane with application to impedance density estimation
• Statistics on the (compact) Stiefel manifold: Theory and Applications

Cited by in corpus (14)

• Riemannian adaptive stochastic gradient algorithms on matrix manifolds
• Manifolds.jl: An Extensible Julia Framework for Data Analysis on Manifolds
• Fréchet Mean Set Estimation in the Hausdorff Metric, via Relaxation
• A reduced parallel transport equation on Lie Groups with a left-invariant metric
• Polynomial Chaos Expansions on Principal Geodesic Grassmannian Submanifolds for Surrogate Modeling and Uncertainty Quantification
• Approximation and bounding techniques for the Fisher-Rao distances between parametric statistical models
• Universal Prototype Transport for Zero-Shot Action Recognition and Localization
• Generalizing Spatial Transformers to Projective Geometry with Applications to 2D/3D Registration
• HypLL: The Hyperbolic Learning Library
• Geometric Machine Learning for Channel Covariance Estimation in Vehicular Networks
• On the Estimation of Centre of Mass in Periodic Systems
• Geometric Learning of Hidden Markov Models via a Method of Moments Algorithm
• ICLR 2021 Challenge for Computational Geometry & Topology: Design and Results
• Computing CNN Loss and Gradients for Pose Estimation with Riemannian Geometry