Memory-augmented Query Reconstruction for LLM-based Knowledge Graph Reasoning
Large language models (LLMs) have achieved remarkable performance on knowledge graph question answering (KGQA) tasks by planning and interacting with knowledge graphs.However, existing methods often confuse tool utilization with knowledge reasoning, harming readability of model outputs and giving rise to hallucinatory tool invocations, which hinders the advancement of KGQA.To address this issue, we propose Memoryaugmented Query Reconstruction for LLMbased Knowledge Graph Reasoning (MemQ) to decouple LLM from tool invocation tasks using LLM-built query memory.By establishing a memory module with explicit descriptions of query statements, the proposed MemQ facilitates the KGQA process with natural language reasoning and memoryaugmented query reconstruction.Meanwhile, we design an effective and readable reasoning to enhance the LLM's reasoning capability in KGQA.Experimental results demonstrate that MemQ achieves state-of-the-art performance on widely used benchmarks WebQSP and CWQ.
Authors
- Kehai Chen (ORCID: https://orcid.org/0000-0002-4346-7618)
- Mingze Xu (ORCID: https://orcid.org/0009-0004-5130-5741)
- Sheng Wang (ORCID: https://orcid.org/0000-0003-2696-2171)
- Min Zhang (ORCID: https://orcid.org/0000-0002-8673-5491)
- Tiejun Zhao (ORCID: https://orcid.org/0000-0003-4659-4935)
- Xun Zhou (ORCID: https://orcid.org/0000-0002-6539-4130)
- Muyun Yang (ORCID: https://orcid.org/0000-0002-5940-0266)
- Guang Liang (ORCID: https://orcid.org/0009-0005-1603-3223)
Publication Details
- Published
- 2025-01-01
- DOI
- https://doi.org/10.18653/v1/2025.findings-acl.1234
- Primary Topic
- Semantic Web and Ontologies
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- Natural Science Foundation of Guangdong Province