arXiv:2504.05216cs.IRcs.AI2025-04中稿 · CIKM 2026被引 6

用语言模型提升密集检索效果,让大模型更懂查询与文档的匹配关系。

Unleashing the Power of LLMs in Dense Retrieval with Query Likelihood Modeling

  • 通过最大化查询似然,增强大模型在检索中的语义理解能力。
  • 在MS MARCO和BEIR数据集上优于现有基于大模型的检索方法。
  • 适合关注大模型检索性能优化的研究者和工程师。

密集检索是信息检索中的关键任务,为重排序和生成增强等下游任务提供基础。近期,大语言模型(LLMs)展现出强大的语义理解能力,吸引了密集检索领域的研究兴趣。尽管LLMs作为解码式生成模型在语言生成方面表现优异,但因缺乏对后续标记的关注,难以建模全局信息。受经典基于词的语言模型(如查询似然,QL)的启发,我们提出通过最大化QL来发挥LLMs的生成优势。不同于直接用QL进行文档排序,我们设计了一个辅助任务——最大化查询似然,以增强检索器骨干网络,进而支持后续对比学习。我们提出了新模型LLM-QL,包含两个关键组件:注意力块(AB)阻止预测标记关注文档结束标记之前的文档标记;文档混淆(DC)在预测时掩码文档部分标记。在域内(MS MARCO)和域外(BEIR)数据集上的评估表明,LLM-QL优于其他基于大模型的检索器。全面分析也验证了其有效性和各组件的作用。

原文摘要 · Abstract (English)

Dense retrieval is a crucial task in Information Retrieval (IR), serving as the basis for downstream tasks such as re-ranking and augmenting generation. Recently, large language models (LLMs) have demonstrated impressive semantic understanding capabilities, making them attractive to researchers focusing on dense retrieval. While LLMs, as decoder-style generative models, excel in language generation, they often fall short in modeling global information due to a lack of attention to subsequent tokens. Drawing inspiration from the classical word-based language modeling approach for IR, specifically the query likelihood (QL) model, we aim to leverage the generative strengths of LLMs through QL maximization. Rather than employing QL estimation for document ranking, we propose an auxiliary task of QL maximization to enhance the backbone for subsequent contrastive learning of the retriever. We introduce our model, LLM-QL, which incorporates two key components: Attention Block (AB) and Document Corruption (DC). AB blocks the attention of predictive tokens to the document tokens before the document's ending token, while DC corrupts a document by masking a portion of its tokens during prediction. Evaluations on the in-domain (MS MARCO) and out-of-domain dataset (BEIR) indicate LLM-QL's superiority over other LLM-based retrievers. Furthermore, comprehensive analyses also validate the efficacy of LLM-QL and its components.

密集检索大模型查询似然对比学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。