arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1708.08640math.NAcs.NA
收藏

快速、准确且可扩展的稀疏耦合矩阵-张量分解方法

Fast, Accurate, and Scalable Method for Sparse Coupled Matrix-Tensor Factorization

Seoul National University(首尔大学)

Dongjin Choi, Jun-Gi Jang, U Kang

AI总结:

本文提出S3CMTF方法,通过利用数据稀疏性、捕捉因素间关系及锁自由并行SGD更新,实现快速、准确且可扩展的耦合矩阵-张量分解,实验表明其在速度和精度上均优于现有方法。

摘要:

如何在快速、准确且可扩展的方式下从张量和矩阵数据中捕捉隐藏属性?耦合矩阵-张量分解(CMTF)是提取张量和矩阵潜在因素的主要工具。随着现实数据规模和维度爆炸性增长,设计准确且高效的CMTF方法变得尤为重要。然而,现有CMTF方法存在精度不足、运行缓慢和可扩展性有限的问题。本文提出S3CMTF,一种快速、准确且可扩展的CMTF方法。S3CMTF通过利用现实张量的稀疏性实现高速,通过捕捉因素间关系实现高精度。此外,S3CMTF通过锁自由并行SGD更新在多核共享内存系统中实现进一步加速。我们提出了两种方法,S3CMTF-naive和S3CMTF-opt。S3CMTF-naive是S3CMTF的基本版本,S3CMTF-opt通过利用中间数据提高速度。我们理论和实验证明S3CMTF是最快的,优于现有方法。实验结果表明,S3CMTF比现有方法快11~43倍,精度高2.1~4.1倍。S3CMTF在数据条目数和核心数上具有线性可扩展性。此外,我们还将S3CMTF应用于结合3个额外矩阵的Yelp推荐张量数据,以发现有趣属性。

英文摘要:

How can we capture the hidden properties from a tensor and a matrix data simultaneously in a fast, accurate, and scalable way? Coupled matrix-tensor factorization (CMTF) is a major tool to extract latent factors from a tensor and matrices at once. Designing an accurate and efficient CMTF method has become more crucial as the size and dimension of real-world data are growing explosively. However, existing methods for CMTF suffer from lack of accuracy, slow running time, and limited scalability. In this paper, we propose S3CMTF, a fast, accurate, and scalable CMTF method. S3CMTF achieves high speed by exploiting the sparsity of real-world tensors, and high accuracy by capturing inter-relations between factors. Also, S3CMTF accomplishes additional speed-up by lock-free parallel SGD update for multi-core shared memory systems. We present two methods, S3CMTF-naive and S3CMTF-opt. S3CMTF-naive is a basic version of S3CMTF, and S3CMTF-opt improves its speed by exploiting intermediate data. We theoretically and empirically show that S3CMTF is the fastest, outperforming existing methods. Experimental results show that S3CMTF is 11~43 times faster, and 2.1~4.1 times more accurate than existing methods. S3CMTF shows linear scalability on the number of data entries and the number of cores. In addition, we apply S3CMTF to Yelp recommendation tensor data coupled with 3 additional matrices to discover interesting properties.

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

↑