arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2004.04667cs.LGcs.MS
收藏

Geomstats:一个用于机器学习中黎曼几何的Python包

Geomstats: A Python Package for Riemannian Geometry in Machine Learning

Inria(法国国家信息与自动化研究所)
Nina Miolane, Alice Le Brigant, Johan Mathe, Benjamin Hou, Nicolas Guigui, Yann Thanwerdas, Stefan Heyder, Olivier Peltre, Niklas Koep, Hadi Zaatiti, Hatem Hajr… 展开作者

Nina Miolane, Alice Le Brigant, Johan Mathe, Benjamin Hou, Nicolas Guigui, Yann Thanwerdas, Stefan Heyder, Olivier Peltre, Niklas Koep, Hadi Zaatiti, Hatem Hajri, Yann Cabanes, Thomas Gerald, Paul Chauchat, Christian Shewmake, Bernhard Kainz, Claire Donnat, Susan Holmes, Xavier Pennec

摘要:

我们介绍Geomstats,一个开源Python工具箱,用于非线性流形的计算和统计,例如双曲空间,对称正定矩阵空间,变换李群等等。我们提供面向对象和广泛的单元测试的实现。其中,流形配备了黎曼度量的家庭,与相关的指数和对数映射,测地线和平行运输。统计学和学习算法为流形上的估计、聚类和降维提供了方法。所有相关的操作都是矢量化的,用于批量计算,并为不同的执行后端提供支持,即NumPy,PyTorch和TensorFlow,从而实现GPU加速。本文介绍了该软件包,并将其与相关库进行了比较,并提供了相关的代码示例。我们表明,Geomstats提供了可靠的构建块,以促进微分几何和统计学的研究,并在机器学习应用程序中民主化黎曼几何的使用。源代码可以在\url{geomstats.ai}的MIT许可下免费获得。

英文摘要:

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}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑