通过种子扩展策略高效检索知识图谱中的多跳关系。
SeedER: Seed-and-Expand Retrieval from Knowledge Graphs
- 先用轻量方法选核心节点,再用强化学习策略逐步扩展。
- 在少于10个候选节点下,召回率显著高于现有方法。
- 适合需要快速精准检索的复杂问答系统使用。
知识图谱(KG)提供了丰富的关系知识表示,但其不规则结构使检索困难:中心节点扩展迅速,密集嵌入方法难以处理多跳组合查询。现有基于代理的图遍历方法虽表达能力强,但大规模检索成本过高。我们提出SeedER(种子-扩展检索),一种利用知识图谱结构的迭代式低开销检索框架。SeedER首先通过轻量级密集检索和实体检索生成一组紧凑的核心节点,然后通过强化学习训练的图感知策略选择性扩展该集合。该设计将全局推理分解为可复用的局部决策,实现高效发现查询相关节点的同时严格控制扩展成本。我们分析了密集检索在组合查询上的理论局限,并从组合泛化与图约束子模优化角度证明了SeedER的优势。实验表明,相较于强基线方法,SeedER在更小候选集(<10个)下显著提升召回率,适合作为知识密集型推理系统的首阶段检索器。
原文摘要 · Abstract (English)
Knowledge graphs (KGs) offer a rich representation for relational knowledge, but their irregular structure makes retrieval challenging: ego-graph expansion grows rapidly, and dense embedding methods struggle with multi-hop compositional queries. Existing agent-based graph exploration approaches, while expressive, are often too expensive for large-scale retrieval. We introduce SeedER (Seed-and-Expand Retrieval), a retrieval framework that explicitly leverages KG structure through iterative, low-cost expansion. SeedER first seeds a compact set of core nodes using lightweight dense and entity-based retrieval, then selectively expands this set via a learned graph-aware policy trained with reinforcement learning. This design decomposes global reasoning into reusable local decisions, enabling efficient discovery of query-relevant nodes while tightly controlling expansion cost. We show theoretical limitations of dense retrieval on compositional graph queries, and establish advantages of SeedER from both compositional generalization and graph-constrained submodular optimization perspectives. Empirically, SeedER substantially improves recall with compact candidate sets over strong dense and graph-augmented baselines, making it an effective first-stage retriever for knowledge-intensive reasoning systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。