Geomstats: a python package for riemannian geometry in machine learning
File(s)19-027.pdf (1.18 MB)
Published version
OA Location
Author(s)
Type
Journal Article
Abstract
We introduce Geomstats, an open-source Python package 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. 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. This paper presents the package, compares it with related libraries, and provides relevant code 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 machine learning applications. The source code is freely available under the MIT license at geomstats.ai.
Date Issued
2020-12-01
Date Acceptance
2020-11-06
Citation
Journal of Machine Learning Research, 2020, 21 (223), pp.1-9
ISSN
1532-4435
Publisher
JMLR
Start Page
1
End Page
9
Journal / Book Title
Journal of Machine Learning Research
Volume
21
Issue
223
Copyright Statement
©2020 Nina Miolane, Nicolas Guigui, Alice Le Brigant, Johan Mathe, Benjamin Hou, Yann Thanwerdas, Stefan
Heyder, Olivier Peltre, Niklas Koep, Hadi Zaatiti, Hatem Hajri, Yann Cabanes, Thomas Gerald, Paul Chauchat,
Christian Shewmake, Daniel Brooks, Bernhard Kainz, Claire Donnat, Susan Holmes and Xavier Pennec.
License: CC-BY 4.0, see https://creativecommons.org/licenses/by/4.0/. Attribution requirements are provided
at http://jmlr.org/papers/v21/19-027.html
Heyder, Olivier Peltre, Niklas Koep, Hadi Zaatiti, Hatem Hajri, Yann Cabanes, Thomas Gerald, Paul Chauchat,
Christian Shewmake, Daniel Brooks, Bernhard Kainz, Claire Donnat, Susan Holmes and Xavier Pennec.
License: CC-BY 4.0, see https://creativecommons.org/licenses/by/4.0/. Attribution requirements are provided
at http://jmlr.org/papers/v21/19-027.html
License URL
Identifier
https://www.jmlr.org/papers/v21/19-027.html
Subjects
Science & Technology
Technology
Automation & Control Systems
Computer Science, Artificial Intelligence
Computer Science
differential geometry
Riemannian geometry
statistics
machine learning
manifold
MANIFOLDS
Artificial Intelligence & Image Processing
08 Information and Computing Sciences
17 Psychology and Cognitive Sciences
Publication Status
Published