arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.14974math.OCcs.LGcs.MScs.NAmath.NA

Automatic differentiation for Riemannian optimization on low-rank matrix and tensor-train manifolds

Marchuk Institute of Numerical Mathematics of the Russian Academy of Sciences(俄罗斯科学院马尔丘克计算数学研究所);HSE University(高等经济学院);Skolkovo Institute of Science and Technology(斯科尔科沃科学技术学院)

Alexander Novikov, Maxim Rakhuba, Ivan Oseledets

英文摘要:

In scientific computing and machine learning applications, matrices and more general multidimensional arrays (tensors) can often be approximated with the help of low-rank decompositions. Since matrices and tensors of fixed rank form smooth Riemannian manifolds, one of the popular tools for finding low-rank approximations is to use Riemannian optimization. Nevertheless, efficient implementation of Riemannian gradients and Hessians, required in Riemannian optimization algorithms, can be a nontrivial task in practice. Moreover, in some cases, analytic formulas are not even available. In this paper, we build upon automatic differentiation and propose a method that, given an implementation of the function to be minimized, efficiently computes Riemannian gradients and matrix-by-vector products between an approximate Riemannian Hessian and a given vector.

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

↑