Efficient Federated Search for Retrieval-Augmented Generation
Large language models (LLMs) have demonstrated remarkable capabilities across various domains but remain susceptible to hallucinations and inconsistencies, limiting their reliability. Retrieval-augmented generation (RAG) mitigates these issues by grounding model responses in external knowledge sources. Existing RAG workflows often leverage a single vector database, which is impractical in the common setting where information is distributed across multiple repositories. We introduce RAGRoute, a novel mechanism for federated RAG search. RAGRoute dynamically selects relevant data sources at query time using a lightweight neural network classifier. By not querying every data source, this approach significantly reduces query overhead, improves retrieval efficiency, and minimizes the retrieval of irrelevant information. We evaluate RAGRoute using the MIRAGE and MMLU benchmarks and demonstrate its effectiveness in retrieving relevant documents while reducing the number of queries. RAGRoute reduces the total number of queries up to 77.5% and communication volume up to 76.2%.
Cite this item
Guerraoui, R., Kermarrec, A.-M., Petrescu, D. A., Pires, R., Randl, M. B. M., & de Vos, M. (2025, March 31). Efficient Federated Search for Retrieval-Augmented Generation. EuroMLSys ’25 Proceedings of Th 2025 5th Workshop on Machine Learning and Systems [Forthcoming Publication]. 5th Workshop on Machine Learning and Systems (EuroMLSys), Rotterdam, The Netherlands, March 31, 2025. https://doi.org/10.1145/3721146.3721942
Click Copy to copy the citation, or select the text manually.
euromlsys25-final14.pdf
Main Document
Published version
openaccess
N/A
938.58 KB
Adobe PDF
53596c1c0aadf73535118087d5aa2950