RouterKGQA: Specialized--General Model Routing for Constraint-Aware Knowledge Graph Question Answering

Bo Yuan, Hexuan Deng, Xuebo Liu, Min Zhang
3/20/2026
cs.CLcs.DBcs.IR

Official code: Oldcircle/RouterKGQA (3 stars, MIT, last commit 2026-03-20).

Abstract

Knowledge graph question answering (KGQA) is a promising approach for mitigating LLM hallucination by grounding reasoning in structured and verifiable knowledge graphs. Existing approaches fall into two paradigms: retrieval-based methods utilize small specialized models, which are efficient but often produce unreachable paths and miss implicit constraints, while agent-based methods utilize large general models, which achieve stronger structural grounding at substantially higher cost. We propose RouterKGQA, a framework for specialized--general model collaboration, in which a specialized model generates reasoning paths and a general model performs KG-guided repair only when needed, improving performance at minimal cost. We further equip the specialized with constraint-aware answer filtering, which reduces redundant answers. In addition, we design a more efficient general agent workflow, further lowering inference cost. Experimental results show that RouterKGQA outperforms the previous best by 3.57 points in F1 and 0.49 points in Hits@1 on average across benchmarks, while requiring only 1.15 average LLM calls per question. Codes and models are available at https://github.com/Oldcircle/RouterKGQA.

View on arXivView PDF

Code Implementations(1)

MIT
33PythonMar 15, 20267 months agoMIT

Similar Papers

Cite this paper

@article{yuan2026routerkgqa,
  title  = {RouterKGQA: Specialized--General Model Routing for Constraint-Aware Knowledge Graph Question Answering},
  author = {Bo Yuan and Hexuan Deng and Xuebo Liu and Min Zhang},
  year   = {2026},
  eprint = {2603.20017},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.20017v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.20017.svg)](https://codeofpaper.com/paper/2603.20017)

Discussion