Explore then Determine: A GNN-LLM Synergy Framework for Reasoning over Knowledge Graph

Guangyi Liu,Yongqi Zhang,Yong Li,Quanming Yao
2024-06-03
Abstract:The task of reasoning over Knowledge Graphs (KGs) poses a significant challenge for Large Language Models (LLMs) due to the complex structure and large amounts of irrelevant information. Existing LLM reasoning methods overlook the importance of compositional learning on KG to supply with precise knowledge. Besides, the fine-tuning and frequent interaction with LLMs incur substantial time and resource costs. This paper focuses on the Question Answering over Knowledge Graph (KGQA) task and proposes an Explore-then-Determine (EtD) framework that synergizes LLMs with graph neural networks (GNNs) for reasoning over KGs. The Explore stage employs a lightweight GNN to explore promising candidates and relevant fine-grained knowledge to the questions, while the Determine stage utilizes the explored information to construct a knowledge-enhanced multiple-choice prompt, guiding a frozen LLM to determine the final answer. Extensive experiments on three benchmark KGQA datasets demonstrate that EtD achieves state-of-the-art performance and generates faithful reasoning results.
Computation and Language
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: the challenges faced by large language models (LLMs) when performing reasoning on knowledge graphs (KG). Specifically, LLMs have the following problems when handling KGQA (Question Answering over Knowledge Graph) tasks: 1. **Complex structure and a large amount of irrelevant information**: The complex structure of the knowledge graph and a large amount of irrelevant information make it difficult for LLMs to reason effectively. 2. **Lack of the importance of combinatorial learning**: Existing methods overlook the importance of performing combinatorial learning on the knowledge graph to provide accurate knowledge. 3. **High time and resource costs**: The time and resource costs caused by fine - tuning and frequent interactions are high. To solve these problems, the paper proposes a framework named "Explore - then - Determine (EtD)", which combines graph neural networks (GNNs) and large language models (LLMs) to perform reasoning on the knowledge graph more effectively. ### Main features of the EtD framework: - **Explore stage**: Use a lightweight GNN to explore promising candidate answers and fine - grained knowledge related to the question, and adaptively filter out irrelevant information. - **Determine stage**: Use the explored information to construct a knowledge - enhanced multiple - choice prompt to guide the frozen LLM to determine the final answer. In this way, the EtD framework can perform reasoning on the knowledge graph more efficiently and accurately, and reduces the need for training and fine - tuning of the LLM. ### Specific contributions of the paper: 1. Propose a novel "Explore - then - Determine" framework, which combines the precision of combinatorial learning and the powerful ability of language understanding. 2. Design a lightweight GNN model empowered by LLM for accurate and efficient KG exploration, and propose a knowledge - enhanced multiple - choice prompt to ensure the accuracy of the answer. 3. Through extensive experimental verification, this method has achieved state - of - the - art performance on three benchmark KGQA datasets, demonstrating its efficient and interpretable reasoning ability. In summary, this paper aims to solve the key challenges in existing KGQA methods by combining the advantages of GNN and LLM, thereby improving the accuracy and efficiency of reasoning.