MoG: Mixture of Experts for Graph-based Retrieval-Augmented Generation
About
Retrieval-augmented generation is intensively studied to ground large language models on external evidence. However, retrieving from a unified knowledge base could inevitably introduce irrelevant information that may mislead generation for complex reasoning. Inspired by the conditional computation of mixture of experts (MoE), where a router sparsely selects specialized experts alongside shared ones for each input, we propose \textbf{M}ixture \textbf{o}f experts for \textbf{G}raph-based Retrieval-Augmented Generation, i.e., \textbf{MoG}. It organizes knowledge into two core components: (i) diverse, always-accessible hub graphs that encode semantically and structurally central knowledge and provide contextual clues for expert activation, and (ii) sparsely activated expert graphs that contain domain-specific evidence. MoG first accesses hub graphs to identify general evidence and derive contextual clues. Then, a topology-aware router dynamically activates a limited set of expert graphs conditioned on the query, thereby confining retrieval to a focused evidence subspace. Extensive experiments on challenging benchmarks show that MoG consistently outperforms strong baselines, with over 20\% relative improvement on MuSiQue. Our code is available in https://github.com/DEEP-PolyU/MoG.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Information Retrieval | HotpotQA | Recall95.4 | 31 | |
| Multi-hop Question Answering | HotpotQA | Top-5 Accuracy82 | 12 | |
| Multi-hop Question Answering | 2Wiki | Top-5 Accuracy84.2 | 12 | |
| Multi-hop Question Answering | MuSiQue | Top-5 Accuracy55.1 | 12 | |
| Multi-hop Question Answering | HotpotQA | LLM Accuracy86.7 | 9 | |
| Multi-hop Question Answering | 2Wiki | LLM Acc85.7 | 9 | |
| Multi-hop Question Answering | MuSiQue | LLM Accuracy66 | 9 | |
| Multi-hop Question Answering | GraphRAG-Bench | LLM Accuracy84.3 | 9 | |
| Information Retrieval | 2Wiki | Recall95.4 | 3 | |
| Information Retrieval | MuSiQue | Recall88.7 | 3 |