KG-Hopper: Empowering Compact Open LLMs with Knowledge Graph Reasoning via Reinforcement Learning
Organizations: Department of Computer Science and Engineering, Chalmers University of Technology and University of Gothenburg, SE-41296 Gothenburg, Sweden
Abstract
Large Language Models (LLMs) demonstrate impressive natural language capabilities but often struggle with knowledge-intensive reasoning tasks. Knowledge Base Question Answering (KBQA), which leverages structured Knowledge Graphs (KGs) exemplifies this challenge due to the need for accurate multi-hop reasoning. Existing approaches typically perform sequential reasoning steps guided by predefined pipelines, restricting flexibility and causing error cascades due to isolated reasoning at each step. To address these limitations, we propose KG-Hopper, a novel Reinforcement Learning (RL) framework that empowers compact open LLMs with the ability to perform integrated multi-hop KG reasoning within a single inference round. Rather than reasoning step-by-step, we train a Reasoning LLM that embeds the entire KG traversal and decision process into a unified ``thinking'' stage, enabling global reasoning over cross-step dependencies and dynamic path exploration with backtracking. Experimental results on eight KG reasoning benchmarks show that KG-Hopper, based on a 7B-parameter LLM, consistently outperforms larger multi-step systems (up to 70B) and achieves competitive performance with proprietary models such as GPT-3.5-Turbo and GPT-4o-mini, while remaining compact, open, and data-efficient. The code is publicly available at: https://github.com/Wangshuaiia/KG-Hopper.
Figures & tables
| Method | Short Description | Pros | Cons |
| Model-KG Integration Strategies | |||
| Prompt-only LLM | LLM answers with parametric knowledge only | Easy to use; no dependencies | Poor on multi-hop or unseen queries |
| LLM + KG (no training) | Uses KG retrieval tools, no adaptation | Quick to integrate KG; flexible | No reasoning adaptation; tool orchestration needed |
| LLM + KG (SFT) | LLM fine-tuned with KG reasoning traces | More accurate; learns tool use | Requires labeled CoT data; brittle over long chains |
| LLM + KG (RL) | RL-trained LLM with KG traversal tools | Strong generalization; fewer examples needed | Requires reward design |
| Inference and Reasoning Modes | |||