DynaKRAG: A Unified Framework for Learnable Evidence Control in Multi-Hop Retrieval-Augmented Generation
Summary
DynaKRAG is a novel unified framework designed for learnable evidence control in multi-hop retrieval-augmented generation (RAG). It addresses the limitation of existing methods that use predefined pipelines for operations like iterative retrieval, query reformulation, and evidence critique. DynaKRAG formulates evidence acquisition as state-conditioned control, where a validity layer constructs executable actions and a learned controller selects the next operation, updating the evidence state. Using Qwen2.5-7B-Instruct, DynaKRAG achieved F1 scores of 0.5998 on HotpotQA, 0.5340 on 2Wiki, and 0.3061 on MuSiQue, surpassing the strongest controlled baseline across all three benchmarks. Experiments showed that replacing the learned controller reduced F1 by 3.96-5.78 points, and removing sufficiency feedback negatively impacted performance, demonstrating the value of coordinating diverse evidence operations.
Key takeaway
For Machine Learning Engineers optimizing multi-hop RAG systems, consider adopting a learned, state-conditioned control framework like DynaKRAG. Your current predefined pipelines may limit performance, as coordinating retrieval, diagnosis, and gap-directed acquisition under an evolving evidence state significantly improves F1 scores. Prioritize integrating sufficiency feedback and dynamically controlling retrieval steps, recognizing that additional retrieval is not always beneficial and can be counterproductive.
Key insights
DynaKRAG unifies multi-hop RAG operations via a learned, state-conditioned control policy for dynamic evidence acquisition.
Principles
- Sufficiency feedback is critical for multi-hop RAG performance.
- Additional retrieval is not uniformly beneficial.
Method
DynaKRAG formulates multi-hop evidence acquisition as state-conditioned control. A validity layer defines executable actions, and a learned controller selects the next operation, updating the evidence state.
In practice
- Implement state-conditioned policies for RAG operations.
- Integrate sufficiency feedback into multi-hop RAG.
Topics
- Multi-hop RAG
- Evidence Control
- State-Conditioned Policies
- Qwen2.5-7B-Instruct
- Natural Language Generation
- Question Answering Benchmarks
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.