TY - GEN
T1 - Temporal Knowledge Graph-Driven Multi-Agent Collaborative Retrieval Augmented Generation Framework
AU - Niu, Yichen
AU - Ai, Ying
AU - Wan, Xueyao
AU - Gao, Shiyu
AU - Zhao, Yue
AU - Kuang, Jiyuan
AU - Liu, Jianxing
N1 - Publisher Copyright:
© 2025 IEEE.
PY - 2025
Y1 - 2025
N2 - 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.
AB - 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.
KW - LLMs
KW - Multi-agent Collaboration
KW - Retrieval Augmented Generation
KW - Temporal Knowledge Graph
UR - https://www.scopus.com/pages/publications/105040990754
U2 - 10.1109/CAC67268.2025.11487625
DO - 10.1109/CAC67268.2025.11487625
M3 - 会议稿件
AN - SCOPUS:105040990754
T3 - Proceedings - 2025 China Automation Congress, CAC 2025
SP - 5256
EP - 5261
BT - Proceedings - 2025 China Automation Congress, CAC 2025
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 2025 China Automation Congress, CAC 2025
Y2 - 26 September 2025 through 28 September 2025
ER -