Multi-hop knowledge graph reasoning usually adopts the reinforcement learning framework, which makes the model’s performance dependent on accurate rewards. Existing methods compute path rewards using the prior knowledge in the knowledge graph, which is inaccurate due to insufficient knowledge. Using LLM’s vast internal knowledge to compute reward can alleviate the problem, but the introduction of LLM also poses the problem of lost semantic information and high time overhead. This paper proposes the Prior Knowledge Augmentation- based Knowledge Graph Reasoning (PKA-KGR). In PKA-KGR, the In- Context Learning-based semantic enhancement strategy can mitigate the loss of path semantics in LLM processing. This strategy enhances the capture of path semantics by focusing the LLM on the structural pat- terns implicit in the triples through the relevant context. In addition, PKA-KGR uses the path importance-based dynamic reward strategy to reduce the high time overhead caused by frequent LLM calls. This strategy dynamically identifies important paths and prioritizes the use of LLM internal knowledge to compute their rewards, rather than using LLM to compute rewards for all paths. Such a dynamic strategy reduces the number of LLM calls while avoiding model performance degradation. Our method has been evaluated on four datasets of different scales, and the results reveal that our method outperforms existing methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Prior Knowledge Augmentation for Reinforcement Learning-Based Knowledge Graph Reasoning

  • Yiyang Weng,
  • Tong Li,
  • Zifang Tang,
  • Junrui Liu,
  • Zhen Yang

摘要

Multi-hop knowledge graph reasoning usually adopts the reinforcement learning framework, which makes the model’s performance dependent on accurate rewards. Existing methods compute path rewards using the prior knowledge in the knowledge graph, which is inaccurate due to insufficient knowledge. Using LLM’s vast internal knowledge to compute reward can alleviate the problem, but the introduction of LLM also poses the problem of lost semantic information and high time overhead. This paper proposes the Prior Knowledge Augmentation- based Knowledge Graph Reasoning (PKA-KGR). In PKA-KGR, the In- Context Learning-based semantic enhancement strategy can mitigate the loss of path semantics in LLM processing. This strategy enhances the capture of path semantics by focusing the LLM on the structural pat- terns implicit in the triples through the relevant context. In addition, PKA-KGR uses the path importance-based dynamic reward strategy to reduce the high time overhead caused by frequent LLM calls. This strategy dynamically identifies important paths and prioritizes the use of LLM internal knowledge to compute their rewards, rather than using LLM to compute rewards for all paths. Such a dynamic strategy reduces the number of LLM calls while avoiding model performance degradation. Our method has been evaluated on four datasets of different scales, and the results reveal that our method outperforms existing methods.