arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1507.05910cs.LGcs.CLstat.ML
收藏

聚类对近似最大内积搜索是有效的

Clustering is Efficient for Approximate Maximum Inner Product Search

École Normale Supérieure(巴黎高等师范学院);Université de Montréal(蒙特利尔大学);Twitter Cortex(推特Cortex部门);Université de Sherbrooke(舍布鲁克大学)

Alex Auvolat, Sarath Chandar, Pascal Vincent, Hugo Larochelle, Yoshua Bengio

摘要:

有效的最大内积搜索(MIPS)是一个重要的任务,具有广泛的适用性,在推荐系统和分类与大量的类。在最近的文献中,已经研究了基于局部敏感哈希(LSH)以及基于树的解决方案的解决方案,以在次线性时间内执行近似MIPS。在本文中,我们比较这些解决近似MIPS的另一个非常简单的方法,基于变体的k-均值聚类算法。具体来说,我们建议训练球形k-均值,减少MIPS问题的最大余弦相似性搜索(MCSS)后。两个标准的推荐系统基准测试以及大词汇量词嵌入的实验表明,这种简单的方法产生更高的加速比,相同的检索精度,比目前最先进的基于哈希和基于树的方法。当查询被噪声破坏时,这种简单的方法也会产生更鲁棒的检索。

英文摘要:

Efficient Maximum Inner Product Search (MIPS) is an important task that has a wide applicability in recommendation systems and classification with a large number of classes. Solutions based on locality-sensitive hashing (LSH) as well as tree-based solutions have been investigated in the recent literature, to perform approximate MIPS in sublinear time. In this paper, we compare these to another extremely simple approach for solving approximate MIPS, based on variants of the k-means clustering algorithm. Specifically, we propose to train a spherical k-means, after having reduced the MIPS problem to a Maximum Cosine Similarity Search (MCSS). Experiments on two standard recommendation system benchmarks as well as on large vocabulary word embeddings, show that this simple approach yields much higher speedups, for the same retrieval precision, than current state-of-the-art hashing-based and tree-based methods. This simple method also yields more robust retrievals when the query is corrupted by noise.

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

↑