Skip to main navigation Skip to search Skip to main content

Temporal Knowledge Graph-Driven Multi-Agent Collaborative Retrieval Augmented Generation Framework

  • School of Astronautics, Harbin Institute of Technology
  • School of Information Science and Engineering, Harbin Institute of Technology Weihai

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Retrieval augmented generation (RAG) expands large language models' knowledge by incorporating external evidence, but conventional RAG frameworks often struggle with reasoning over time-stamped events and managing noise from overly broad retrieval. To address these challenges, we propose TKG-MRAG, a temporal knowledge graph-driven multi-agent collaborative RAG framework that tackles both issues. In the indexing phase, LLM extracts 〈entity, relation, timestamp〉 triples from raw text to build a temporal knowledge graph and constructs two-layer indexing structure (the temporal index and the semantic index) to prevent reasoning errors arising from time conflicts. During retrieval, we designed a multi-agent collaborative retrieval mechanism: a temporal planning agent decomposes the query into a sub-query reasoning sequence; a causal retriever agent prunes the temporal subgraph and selects candidate nodes; and a reflective refiner agent ensures consistency, eliminates redundancy, and outputs a compact evidence subgraph. Finally, the original query, sub-query reasoning sequence, and refined subgraph are fed into an LLM to generate the answer. Experiments on the Time-LongQA benchmark demonstrate that TKG-MRAG significantly outperforms baseline methods including Naive-RAG, Light-RAG, and Graph-RAG, achieving superior accuracy in temporal question answering tasks.

Original languageEnglish
Title of host publicationProceedings - 2025 China Automation Congress, CAC 2025
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages5256-5261
Number of pages6
ISBN (Electronic)9798331589677
DOIs
StatePublished - 2025
Externally publishedYes
Event2025 China Automation Congress, CAC 2025 - Harbin, China
Duration: 26 Sep 202528 Sep 2025

Publication series

NameProceedings - 2025 China Automation Congress, CAC 2025

Conference

Conference2025 China Automation Congress, CAC 2025
Country/TerritoryChina
CityHarbin
Period26/09/2528/09/25

Keywords

  • LLMs
  • Multi-agent Collaboration
  • Retrieval Augmented Generation
  • Temporal Knowledge Graph

Fingerprint

Dive into the research topics of 'Temporal Knowledge Graph-Driven Multi-Agent Collaborative Retrieval Augmented Generation Framework'. Together they form a unique fingerprint.

Cite this