Self-Routing RAG: Binding Selective Retrieval with Knowledge Verbalization
Wu, Di, Gu, Jia-Chen, Chang, Kai-Wei, Peng, Nanyun
–arXiv.org Artificial Intelligence
Selective retrieval improves retrieval-augmented generation (RAG) by reducing distractions from low-quality retrievals and improving efficiency. However, existing approaches under-utilize the inherent knowledge of large language models (LLMs), leading to suboptimal retrieval decisions and degraded generation performance. To bridge this gap, we propose Self-Routing RAG (SR-RAG), a novel framework that binds selective retrieval with knowledge verbalization. SR-RAG enables an LLM to dynamically decide between external retrieval and verbalizing its own parametric knowledge. To this end, we design a multi-task objective that jointly optimizes an LLM on knowledge source selection, knowledge verbalization, and response generation. We further introduce dynamic knowledge source inference via nearest neighbor search to improve the accuracy of knowledge source decision under domain shifts. Fine-tuning three LLMs with SR-RAG significantly improves both their response accuracy and inference latency. Compared to the strongest selective retrieval baseline, SR-RAG reduces retrievals by 29% while improving the performance by 5.1%.
arXiv.org Artificial Intelligence
Apr-1-2025
- Country:
- Africa
- Ethiopia > Addis Ababa
- Addis Ababa (0.04)
- Rwanda > Kigali
- Kigali (0.04)
- Ethiopia > Addis Ababa
- Asia
- Atlantic Ocean (0.04)
- Europe
- Austria > Vienna (0.14)
- Belgium > Brussels-Capital Region
- Brussels (0.04)
- France (0.14)
- Ireland > Leinster
- County Dublin > Dublin (0.04)
- United Kingdom > England (0.04)
- North America
- Canada (0.04)
- Dominican Republic (0.04)
- Haiti (0.46)
- Mexico > Mexico City
- Mexico City (0.04)
- Saint Pierre and Miquelon > Miquelon-Langlade
- Miquelon (0.04)
- United States
- California > Los Angeles County
- Los Angeles (0.14)
- Florida > Miami-Dade County
- Miami (0.04)
- Louisiana > Orleans Parish
- New Orleans (0.05)
- Maryland > Baltimore (0.04)
- California > Los Angeles County
- Oceania > Australia (0.04)
- South America > Colombia
- Meta Department > Villavicencio (0.04)
- Africa
- Genre:
- Research Report (0.64)
- Technology: