Search-on-Graph: Iterative Informed Navigation for Large Language Model Reasoning on Knowledge Graphs
About
Large language models (LLMs) augmented with knowledge graphs (KGs) offer a promising approach for knowledge-intensive reasoning. Central to this approach is the selection of appropriate reasoning paths in the KG. Yet, existing methods face a common limitation: reasoning path selection is often performed by separate modules using criteria that are only weakly connected to the reasoning requirements. This often results in selecting incorrect relations or premature pruning of relevant paths. We propose Search-on-Graph (SoG), a method that strengthens the connection between path selection and reasoning by having the LLM itself select which relations to follow, informed by both the available KG structure and the complete reasoning history. SoG follows an \textit{observe-think-navigate} paradigm: at each step, the LLM observes the relational connections available at the current entity, reasons about which path best advances toward answering the question, and navigates accordingly. This context-aware navigation fully exploits the LLM's reasoning capabilities rather than relying on independent selection modules with surrogate criteria. Experiments on six knowledge graph question answering (KGQA) benchmarks demonstrate that SoG outperforms state-of-the-art methods while requiring no task-specific fine-tuning and generalizing across different KG schemas.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Knowledge Base Question Answering | WebQSP Freebase (test) | -- | 60 | |
| Knowledge Base Question Answering | GrailQA Freebase (test) | Hits@186.9 | 48 | |
| Knowledge Base Question Answering | CWQ Freebase (test) | -- | 38 | |
| Knowledge Graph Question Answering | SimpleQA Freebase-based (test) | Hits@184.8 | 31 | |
| Knowledge Graph Question Answering | QALD-10en Wikidata-based (test) | Hits@174.4 | 31 | |
| Knowledge Graph Question Answering | QALD-9 Wikidata (test) | Exact Match Accuracy79.4 | 6 |