arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.13402math.NAcs.MScs.NA

在 GPU 上计算秩揭示分解的高效算法

Efficient algorithms for computing rank-revealing factorizations on a GPU

Nathan Heavner, Chao Chen, Abinand Gopal, Per-Gunnar Martinsson

AI总结:

本文提出两种基于随机投影的 GPU 秩揭示分解算法,将大部分运算转化为高效的矩阵乘法,在保持接近 SVD 低秩近似误差的同时实现了一个数量级的加速。

摘要:

诸如奇异值分解和列主元 QR 分解等标准秩揭示分解在 GPU 上难以高效实现。其主要困难在于标准算法无法将大部分运算转化为 Level-3 BLAS 操作。本文提出了两种替代算法,用于计算形如 $A = U T V^*$ 的秩揭示分解,其中 $U$ 和 $V$ 为正交矩阵,$T$ 为三角矩阵。这两种算法均采用随机投影技术,将大部分浮点运算转化为在 GPU 上极其高效的矩阵乘法。数值实验表明,与经过精细调优的 GPU SVD 实现相比,这些算法实现了一个数量级的加速,同时其低秩近似误差接近 SVD。

英文摘要:

Standard rank-revealing factorizations such as the singular value decomposition and column pivoted QR factorization are challenging to implement efficiently on a GPU. A major difficulty in this regard is the inability of standard algorithms to cast most operations in terms of the Level-3 BLAS. This paper presents two alternative algorithms for computing a rank-revealing factorization of the form $A = U T V^*$, where $U$ and $V$ are orthogonal and $T$ is triangular. Both algorithms use randomized projection techniques to cast most of the flops in terms of matrix-matrix multiplication, which is exceptionally efficient on the GPU. Numerical experiments illustrate that these algorithms achieve an order of magnitude acceleration over finely tuned GPU implementations of the SVD while providing low-rank approximation errors close to that of the SVD.

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

↑