用轻量强化学习让多智能体检索生成系统更高效
Advancing Multi-Agent RAG Systems with Minimalist Reinforcement Learning
- 将多轮交互拆解为协作子任务,缓解长上下文问题
- 在多个问答基准上表现优于现有方法,提升推理效率
- 适合需要复杂推理的RAG系统开发者使用
配备现代检索增强生成(RAG)系统的大型语言模型常依赖多轮交互来完成复杂推理任务。然而,这种多轮交互导致中间上下文长度随探索深度指数增长,加剧了大模型处理长上下文的能力瓶颈。尤其在依赖上下文学习的RAG系统中,少样本示例也需包含在提示中,进一步加重上下文负担。为此,我们提出Mujica-MyGo:一个统一的高效多轮推理框架。受分治思想启发,Mujica将多轮交互分解为协同子交互,有效缓解长上下文问题;同时开发了轻量高效的强化学习算法MyGO(Minimalist Policy Gradient Optimization),无需依赖上下文学习即可在复杂RAG流程中实现有效后训练,并提供理论保障其收敛至最优策略。在涵盖文本语料库与知识图谱的多个问答基准上的实证评估表明,Mujica-MyGo表现出色。
原文摘要 · Abstract (English)
Large Language Models (LLMs) equipped with modern Retrieval-Augmented Generation (RAG) systems often employ multi-turn interaction pipelines to interface with search engines for complex reasoning tasks. However, such multi-turn interactions inevitably produce long intermediate contexts, as context length grows exponentially with exploration depth. This leads to a well-known limitation of LLMs: their difficulty in effectively leveraging information from long contexts. This problem is further amplified in RAG systems that depend on in-context learning, where few-shot demonstrations must also be included in the prompt, compounding the context-length bottleneck. To address these challenges, we propose Mujica-MyGo, a unified framework for efficient multi-turn reasoning in RAG. Inspired by the divide-and-conquer principle, we introduce Mujica (Multi-hop Joint Intelligence for Complex Question Answering), a multi-agent RAG workflow that decomposes multi-turn interactions into cooperative sub-interactions, thereby mitigating long-context issues. To eliminate the dependency on in-context learning, we further develop MyGO (Minimalist Policy Gradient Optimization), a lightweight and efficient reinforcement learning algorithm that enables effective post-training of LLMs within complex RAG pipelines. We provide theoretical guarantees for MyGO's convergence to the optimal policy. Empirical evaluations across diverse question-answering benchmarks, covering both text corpora and knowledge graphs, show that Mujica-MyGO achieves superior performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。