arXiv:2410.13765cs.CLcs.IR2024-10NAACL被引 38

用知识图谱增强大模型查询扩展,提升复杂查询的文本与关系检索效果

Knowledge-Aware Query Expansion with Large Language Models for Textual and Relational Retrieval

  • 将知识图谱中的文档关系注入大模型,生成更符合用户意图的查询扩展
  • 在三个跨领域数据集上优于现有最优基线,尤其在关系型查询上提升显著
  • 适合需要同时理解语义和结构关系的智能搜索系统开发者

大语言模型(LLM)已被用于生成查询扩展以提升信息检索效果。近期研究尝试将初始检索结果提供给LLM,使其生成更贴近文档语料的扩展,但多数方法仅关注查询与目标文档之间的文本相似性,忽视了文档间的关系。对于“找一款高评分、适合野生动物摄影且兼容尼康F卡口镜头的相机”这类兼具文本与关系需求的半结构化查询,现有方法可能生成语义相关但结构无关的扩展。为此,本文提出一种知识感知查询扩展框架,通过知识图谱(KG)引入结构化文档关系,并利用文档文本作为节点丰富表示,采用基于文档的关系过滤策略实现知识感知检索(KAR)。在三个不同领域的数据集上进行的大量实验表明,该方法在文本与关系双重要求的半结构化检索任务中显著优于当前最佳基线。

原文摘要 · Abstract (English)

Large language models (LLMs) have been used to generate query expansions augmenting original queries for improving information search. Recent studies also explore providing LLMs with initial retrieval results to generate query expansions more grounded to document corpus. However, these methods mostly focus on enhancing textual similarities between search queries and target documents, overlooking document relations. For queries like "Find me a highly rated camera for wildlife photography compatible with my Nikon F-Mount lenses", existing methods may generate expansions that are semantically similar but structurally unrelated to user intents. To handle such semi-structured queries with both textual and relational requirements, in this paper we propose a knowledge-aware query expansion framework, augmenting LLMs with structured document relations from knowledge graph (KG). To further address the limitation of entity-based scoring in existing KG-based methods, we leverage document texts as rich KG node representations and use document-based relation filtering for our Knowledge-Aware Retrieval (KAR). Extensive experiments on three datasets of diverse domains show the advantages of our method compared against state-of-the-art baselines on textual and relational semi-structured retrieval.

知识图谱查询扩展大模型信息检索

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