Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Jia Ao Sun, Hao Yu, Fabrizio Gotti, Fengran Mo, Yihong Wu, Yuchen Hui, Zhan Su, Lingfeng Xiao, Jian-Yun Nie• 2025

Related benchmarks

TaskDatasetResultRank
Knowledge Base Question AnsweringWebQSP Freebase (test)--
60
Knowledge Base Question AnsweringGrailQA Freebase (test)
Hits@186.9
48
Knowledge Base Question AnsweringCWQ Freebase (test)--
38
Knowledge Graph Question AnsweringSimpleQA Freebase-based (test)
Hits@184.8
31
Knowledge Graph Question AnsweringQALD-10en Wikidata-based (test)
Hits@174.4
31
Knowledge Graph Question AnsweringQALD-9 Wikidata (test)
Exact Match Accuracy79.4
6
Showing 6 of 6 rows

Other info

Follow for update