Temporal knowledge graph (TKG) logical reasoning aims to predict future facts by learning logical correlations from TKGs. However, conventional Retrieval-Augmented Generation (RAG) methods rely on unstructured text as the retrieval space, failing to capture and utilize implicit logical correlations within the retrieved content. Furthermore, the conventional rule confidence calculation methods significantly restrict the generalization of the model when temporal logical rules are incorporated. To address these challenges, We propose a novel TKG reasoning framework RuleRAG (integration Rule with RAG). It validates the confidence of temporal logical rules using large language models (LLMs), and applies the rules and the query as input to LLMs. Experimental results demonstrate that RuleRAG outperforms state-of-the-art models.

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

Rules-Guided Retrieval-Augmented Generation for Temporal Knowledge Graph Reasoning

  • Kaijia Xu,
  • Lin Liu,
  • Hailong Wang,
  • Ruixin Shi,
  • Tianyuan Niu,
  • Anqi Ren

摘要

Temporal knowledge graph (TKG) logical reasoning aims to predict future facts by learning logical correlations from TKGs. However, conventional Retrieval-Augmented Generation (RAG) methods rely on unstructured text as the retrieval space, failing to capture and utilize implicit logical correlations within the retrieved content. Furthermore, the conventional rule confidence calculation methods significantly restrict the generalization of the model when temporal logical rules are incorporated. To address these challenges, We propose a novel TKG reasoning framework RuleRAG (integration Rule with RAG). It validates the confidence of temporal logical rules using large language models (LLMs), and applies the rules and the query as input to LLMs. Experimental results demonstrate that RuleRAG outperforms state-of-the-art models.