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

GraphRAG-Router: Learning Cost-Efficient Routing over GraphRAGs and LLMs with Reinforcement Learning

About

Graph-based retrieval-augmented generation (GraphRAG) has recently emerged as a powerful paradigm for knowledge-intensive question answering, especially for tasks that require structured evidence organization and multi-hop reasoning. However, existing GraphRAG systems are typically built in a one-size-fits-all manner, relying on a fixed retrieval framework and a single, often large and costly, generator LLM for all queries. This static design limits their ability to adapt to the complexity of varying questions and often incurs unnecessary computational cost. To fill in the gap, we propose GraphRAG-Router, a cost-efficient framework that adopts a hierarchical routing strategy to coordinate heterogeneous GraphRAGs and generator LLMs. Specifically, GraphRAG-Router is first warmed up through supervised fine-tuning and then optimized with a two-stage reinforcement learning procedure, whose second stage introduces a curriculum cost-aware reward to encourage difficulty-aware and economical generator allocation. Extensive experiments on six general-domain and multi-hop QA benchmarks show that GraphRAG-Router consistently outperforms state-of-the-art baselines, reducing the overuse of large LLMs by nearly 30% while maintaining strong generalization capability.

Dongzhe Fan, Chuanhao Ji, Zimu Wang, Tong Chen, Qiaoyu Tan• 2026

Related benchmarks

TaskDatasetResultRank
Multi-hop Question Answering2Wiki--
215
Multi-hop Question AnsweringMuSiQue--
209
Multi-hop QAHotpotQA
Exact Match46.1
143
Question AnsweringTriviaQA--
71
Multi-hop QA2WikiMultihopQA
Exact Match (EM)52.3
67
Question AnsweringNQ
F1 Score (NQ)52.5
64
General QAPopQA
Exact Match (EM)36.8
58
Multi-hop Question AnsweringHotpotQA
F158.9
54
General QATriviaQA
EM67.2
48
General Question AnsweringNQ (Natural Questions)
EM42.6
32
Showing 10 of 16 rows

Other info

Follow for update