Geomstats – a Python package for differential geometry in statistics and machine learning
Nina Miolane (UC Santa Barbara)
Abstract: We introduce Geomstats, an open-source Python package for computations and statistics on nonlinear manifolds that appear in machine learning applications, 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. 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 provide support for different execution backends — namely NumPy, Autograd, PyTorch, and TensorFlow. This talk presents the package, compares it with related libraries, and provides relevant examples. We show that Geomstats provides reliable building blocks to both foster research in differential geometry and statistics and democratize the use of (Riemannian) geometry in statistics and machine learning. The source code is freely available under the MIT license at github.com/geomstats/geomstats.
mathematical softwaresymbolic computationMathematics
Audience: advanced learners
Global Virtual SageDays 112.358
Organizers: | Matthias Köppe*, Jean-Philippe Labbé*, Yuan Zhou* |
*contact for this listing |