E^2GraphRAG: Streamlining Graph-based RAG for High Efficiency and Effectiveness
About
Graph-based RAG methods like GraphRAG have shown promising global understanding of the knowledge base by constructing hierarchical entity graphs. However, they often suffer from inefficiency and rely on manually pre-defined query modes, limiting practical use. In this paper, we propose E^2GraphRAG, a streamlined graph-based RAG framework that improves both Efficiency and Effectiveness. During the indexing stage, E^2GraphRAG constructs a summary tree with large language models and an entity graph with SpaCy based on document chunks. We then construct bidirectional indexes between entities and chunks to capture their many-to-many relationships, enabling fast lookup during both local and global retrieval. For the retrieval stage, we design an adaptive retrieval strategy that leverages the graph structure to retrieve and select between local and global modes. Experiments show that E^2GraphRAG achieves up to 10 times faster indexing than GraphRAG and 100 times speedup over LightRAG in retrieval while maintaining competitive QA performance.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multi-hop Question Answering | HotpotQA | LLM Judge Score65.7 | 72 | |
| Multi-hop Question Answering | 2WikiMultihopQA | String Accuracy57.2 | 44 | |
| Multi-hop Question Answering | MuSiQue | String Accuracy26.1 | 44 | |
| Question Answering | medical | GPT Accuracy58 | 31 | |
| Question Answering | G-bench Novel | Accuracy53.82 | 25 | |
| Multi-hop Question Answering | G-Novel | LLM Accuracy54.28 | 20 | |
| Multi-hop Question Answering | G-Medical | LLM Accuracy60.24 | 20 | |
| Multi-hop Question Answering | Multi-hop QA Suite (HotpotQA, 2Wiki, MuSiQue, G-Medical, G-Novel) | Average Score49.57 | 20 | |
| Multi-hop Question Answering | HotpotQA | SubEM61 | 17 | |
| Multi-hop Question Answering | 2WikiMultiHop | SubEM54.3 | 17 |