arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1812.06360cs.LGstat.ML
收藏

最大内积搜索的Bandit方法

A Bandit Approach to Maximum Inner Product Search

University of Michigan, Ann Arbor(密歇根大学安娜堡分校)

Rui Liu, Tianyi Wu, Barzan Mozafari

摘要:

最大内积搜索(MIPS)的次线性时间近似算法已经有了大量的研究。为了实现快速查询时间,最先进的技术需要大量的预处理,当后续查询的数量不足以分摊成本时,这可能是一个负担。此外,现有的方法没有能力直接控制其近似结果的次优性与理论保证。在本文中,我们提出了MIPS的第一近似算法,不需要任何预处理,并允许用户控制和约束结果的次优性。我们将MIPS视为最佳手臂识别问题,并引入了一个新的强盗设置,可以充分利用MIPS的特殊结构。我们的方法在合成和真实世界的数据集上都优于最先进的方法。

英文摘要:

There has been substantial research on sub-linear time approximate algorithms for Maximum Inner Product Search (MIPS). To achieve fast query time, state-of-the-art techniques require significant preprocessing, which can be a burden when the number of subsequent queries is not sufficiently large to amortize the cost. Furthermore, existing methods do not have the ability to directly control the suboptimality of their approximate results with theoretical guarantees. In this paper, we propose the first approximate algorithm for MIPS that does not require any preprocessing, and allows users to control and bound the suboptimality of the results. We cast MIPS as a Best Arm Identification problem, and introduce a new bandit setting that can fully exploit the special structure of MIPS. Our approach outperforms state-of-the-art methods on both synthetic and real-world datasets.

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

↑