GeAR: Graph-enhanced Agent for Retrieval-augmented Generation

Shen, Zhili, Diao, Chenxin, Vougiouklis, Pavlos, Merita, Pascual, Piramanayagam, Shriram, Graux, Damien, Tu, Dandan, Jiang, Zeren, Lai, Ruofei, Ren, Yang, Pan, Jeff Z.

arXiv.org Artificial Intelligence 

Retrieval-augmented generation systems rely on effective document retrieval capabilities. By design, conventional sparse or dense retrievers face challenges in multi-hop retrieval scenarios. In this paper, we present GeAR, which advances RAG performance through two key innovations: (i) graph expansion, which enhances any conventional base retriever, such as BM25, and (ii) an agent framework that incorporates graph expansion. Our evaluation demonstrates GeAR's superior retrieval performance on three multi-hop question answering datasets. Additionally, our system achieves state-of-the-art results with improvements exceeding 10% on the challenging MuSiQue dataset, while requiring fewer tokens and iterations compared to other multi-step retrieval systems.