用小型神经网络做查询,比传统向量匹配更精准。
Hypencoder: Hypernetworks for Information Retrieval
- 用超网络生成查询专属的打分函数,替代固定向量。
- 在880万文档中检索仅需58毫秒,性能远超基线模型。
- 适合需要高精度、复杂查询理解的搜索场景。
现有信息检索系统主要依赖向量内积判断查询与文档相关性,限制了打分表达能力。本文提出新范式:不将查询表示为向量,而是使用一个小神经网络作为可学习的查询特定相关性函数,该函数接收文档向量输入并输出标量相关性分数。通过超网络(hypernetwork)生成该小网络的权重,构建称为Hypencoder的查询编码器。在域内搜索任务上,Hypencoders显著优于强健的密集检索模型,甚至超越参数量大一个数量级的重排序模型。在包含“舌尖效应”和指令跟随等难题的硬任务上,性能差距进一步扩大。此外,我们实现了一种近似搜索算法,在880万文档的语料库中检索耗时低于60毫秒,验证了方法的实用性。
原文摘要 · Abstract (English)
Existing information retrieval systems are largely constrained by their reliance on vector inner products to assess query-document relevance, which naturally limits the expressiveness of the relevance score they can produce. We propose a new paradigm; instead of representing a query as a vector, we use a small neural network that acts as a learned query-specific relevance function. This small neural network takes a document representation as input (in this work we use a single vector) and produces a scalar relevance score. To produce the small neural network we use a hypernetwork, a network that produces the weights of other networks, as our query encoder. We name this category of encoder models Hypencoders. Experiments on in-domain search tasks show that Hypencoders significantly outperform strong dense retrieval models and even surpass reranking models and retrieval models with an order of magnitude more parameters. To assess the extent of Hypencoders' capabilities, we evaluate on a set of hard retrieval tasks including tip-of-the-tongue and instruction-following retrieval tasks. On harder tasks, we find that the performance gap widens substantially compared to standard retrieval tasks. Furthermore, to demonstrate the practicality of our method, we implement an approximate search algorithm and show that our model is able to retrieve from a corpus of 8.8M documents in under 60 milliseconds.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。