arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1602.01376math.NAcs.DScs.MScs.NA
收藏

Inv-ASKIT:一种并行快速核矩阵求解器

Inv-ASKIT: A Parallel Fast Diret Solver for Kernel Matrices

The University of Texas at Austin(德克萨斯大学奥斯汀分校);Institute for Computational Engineering and Science(计算工程与科学研究所)

Chenhan D. Yu, William B. March, Bo Xiao, George Biros

AI总结:

本文提出Inv-ASKIT方法,通过并行计算快速求解大规模核矩阵的近似因子分解,提升计算效率与可扩展性。

摘要:

我们提出了一种并行算法,用于计算N×N核矩阵的近似因子分解。一旦构造出该分解(工作量为N log²N),就可以用N logN的工作量解决线性系统。核矩阵表示度量空间中点对的相互作用。它们出现在机器学习、逼近理论和计算物理中。核矩阵通常密集(矩阵乘法与N平方成正比)且病态(求解可能需要数百次Krylov迭代)。因此,快速的矩阵乘法和因子分解算法对于可扩展性至关重要。最近我们引入了ASKIT,一种新的近似核矩阵的方法,类似于N体方法。在此,我们介绍Inv-ASKIT,一种基于ASKIT的因子分解方案。我们描述了新方法,推导了复杂度估计,并进行了其准确性和可扩展性的实证研究。我们报告了在共享内存和分布式内存并行上的真实世界数据集的结果,包括“COVTYPE”(0.5M点,54维)、“SUSY”(4.5M点,8维)和“MNIST”(2M点,784维)。在我们最大的运行中,我们使用4096个Sandy-Bridge核心,在64维点生成的32M×32M密集矩阵上进行了近似分解。据我们所知,这些结果在多个数量级上改进了现有技术。

英文摘要:

We present a parallel algorithm for computing the approximate factorization of an $N$-by-$N$ kernel matrix. Once this factorization has been constructed (with $N \log^2 N $ work), we can solve linear systems with this matrix with $N \log N $ work. Kernel matrices represent pairwise interactions of points in metric spaces. They appear in machine learning, approximation theory, and computational physics. Kernel matrices are typically dense (matrix multiplication scales quadratically with $N$) and ill-conditioned (solves can require 100s of Krylov iterations). Thus, fast algorithms for matrix multiplication and factorization are critical for scalability. Recently we introduced ASKIT, a new method for approximating a kernel matrix that resembles N-body methods. Here we introduce INV-ASKIT, a factorization scheme based on ASKIT. We describe the new method, derive complexity estimates, and conduct an empirical study of its accuracy and scalability. We report results on real-world datasets including "COVTYPE" ($0.5$M points in 54 dimensions), "SUSY" ($4.5$M points in 8 dimensions) and "MNIST" (2M points in 784 dimensions) using shared and distributed memory parallelism. In our largest run we approximately factorize a dense matrix of size 32M $\times$ 32M (generated from points in 64 dimensions) on 4,096 Sandy-Bridge cores. To our knowledge these results improve the state of the art by several orders of magnitude.

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

↑