DynaKRAG: A Unified Framework for Learnable Evidence Control in Multi-Hop Retrieval-Augmented Generation

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, extended

Summary

DynaKRAG is a unified framework for multi-hop Retrieval-Augmented Generation (RAG) that formulates evidence acquisition as state-conditioned control over atomic operations. It addresses the limitation of existing RAG methods that organize operations within method-specific pipelines or predefined control topologies. DynaKRAG employs a validity layer to construct the executable action set and a learned controller to select the next operation, updating the evidence state. Using Qwen2.5-7B-Instruct, it achieved F1 scores of 0.5998 on HotpotQA, 0.5340 on 2Wiki, and 0.3061 on MuSiQue, outperforming the strongest controlled baselines. Ablation studies demonstrated that replacing the learned controller reduced F1 by 3.96–5.78 points, and additional retrieval is not uniformly beneficial, highlighting the value of coordinating retrieval, diagnosis, and gap-directed acquisition.

Key takeaway

For Machine Learning Engineers developing multi-hop RAG systems, DynaKRAG offers a robust approach to overcome fragmented pipelines and improve efficiency. You should consider implementing a state-conditioned control policy that dynamically selects evidence operations, rather than relying on fixed retrieval depths or predefined sequences. This can lead to higher answer quality and reduced token consumption, especially for complex compositional questions, and generalizes across different answer model backbones.

Key insights

DynaKRAG unifies multi-hop RAG by learning state-conditioned control over atomic evidence operations for efficient, adaptive acquisition.

Principles

Method

DynaKRAG uses a hard validity layer to define executable actions for the current evidence state. A learned value model then ranks these valid actions, selecting the most useful operation to update the state.

In practice

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.