为多智能体系统设计了可解耦训练的分层强化学习方法,提升复杂任务推理能力。
Multi-Agent Deep Research: Training Multi-Agent Systems with M-GRPO
- 采用分层优势计算与轨迹对齐机制,解决多智能体异步、异构问题。
- 在GAIA等真实基准上超越单智能体和固定子代理模型,样本效率更高。
- 适合需要多个专用智能体协同完成复杂工具调用的任务场景。
多智能体系统在通用推理任务中表现良好,但缺乏专业领域训练限制了其准确性。现有方法对所有智能体统一训练大语言模型,因不同智能体分布差异导致性能受限。因此,为各智能体分别训练独立大语言模型是下一步关键。然而该方法带来优化挑战:智能体运行频率不同,轨迹涉及不同数量的子智能体调用,且常部署于独立服务器,破坏端到端梯度流。为此,我们提出M-GRPO,一种面向主智能体(规划者)与多个子智能体(多轮工具执行器)的垂直多智能体系统的分层组相对策略优化扩展。M-GRPO为上下层智能体分别计算组相对优势,保持层级信用分配;引入轨迹对齐方案,在子智能体调用数量变化时仍生成固定大小批次。采用解耦训练管道,各智能体在独立服务器运行,仅通过共享存储交换最小统计量,实现无需跨服务器反向传播的可扩展训练。在真实世界基准(如GAIA、XBench-DeepSearch、WebWalkerQA)上的实验表明,M-GRPO始终优于单智能体GRPO及冻结子智能体的多智能体GRPO,展现出更强稳定性与样本效率。结果表明,对齐异质轨迹并解耦专业化智能体间的优化,能有效提升工具增强型推理任务表现。
原文摘要 · Abstract (English)
Multi-agent systems perform well on general reasoning tasks. However, the lack of training in specialized areas hinders their accuracy. Current training methods train a unified large language model (LLM) for all agents in the system. This may limit the performances due to different distributions underlying for different agents. Therefore, training multi-agent systems with distinct LLMs should be the next step to solve. However, this approach introduces optimization challenges. For example, agents operate at different frequencies, rollouts involve varying sub-agent invocations, and agents are often deployed across separate servers, disrupting end-to-end gradient flow. To address these issues, we propose M-GRPO, a hierarchical extension of Group Relative Policy Optimization designed for vertical Multi-agent systems with a main agent (planner) and multiple sub-agents (multi-turn tool executors). M-GRPO computes group-relative advantages for both main and sub-agents, maintaining hierarchical credit assignment. It also introduces a trajectory-alignment scheme that generates fixed-size batches despite variable sub-agent invocations. We deploy a decoupled training pipeline in which agents run on separate servers and exchange minimal statistics via a shared store. This enables scalable training without cross-server backpropagation. In experiments on real-world benchmarks (e.g., GAIA, XBench-DeepSearch, and WebWalkerQA), M-GRPO consistently outperforms both single-agent GRPO and multi-agent GRPO with frozen sub-agents, demonstrating improved stability and sample efficiency. These results show that aligning heterogeneous trajectories and decoupling optimization across specialized agents enhances tool-augmented reasoning tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。