解决多智能体大模型强化学习训练不稳问题,提升推理与搜索性能。
Dr. MAS: Stable Reinforcement Learning for Multi-Agent LLM Systems
- 按每个智能体自身奖励统计归一化优势,稳定梯度。
- 数学与搜索任务上相比基线提升4.6%~15.2%,消除梯度突增。
- 支持异构模型部署,适合需高效多智能体协作的场景。
多智能体大模型系统通过角色分工实现高级推理与工具使用,但其强化学习(RL)后训练仍存在可靠性难题。本文理论分析发现,将群体式RL扩展至多智能体大模型时,全局归一化基线可能偏离各智能体的奖励分布,导致梯度范数不稳定。基于此,提出Dr. MAS:一种简单稳定的多智能体大模型强化学习训练方法。该方法采用个体化修正策略,即按每个智能体自身的奖励统计归一化优势,有效校准梯度尺度,显著提升训练稳定性,兼具理论与实证支持。此外,Dr. MAS构建了端到端训练框架,支持可扩展编排、灵活的每智能体模型服务与优化配置,以及共享的LLM执行后端资源调度。在Qwen2.5与Qwen3系列模型上,于多智能体数学推理与多轮搜索基准测试中,相比基础GRPO方法,平均准确率提升5.6%(math, avg@16)、pass率提升4.6%(math, pass@16),搜索任务分别提升15.2%与13.1%,同时大幅消除梯度尖峰。在异构智能体-模型分配下仍保持高效,提升系统整体效率。
原文摘要 · Abstract (English)
Multi-agent LLM systems enable advanced reasoning and tool use via role specialization, yet reliable reinforcement learning (RL) post-training for such systems remains difficult. In this work, we theoretically pinpoint a key reason for training instability when extending group-based RL to multi-agent LLM systems. We show that under GRPO-style optimization, a global normalization baseline may deviate from diverse agents' reward distributions, which ultimately leads to gradient-norm instability. Based on this finding, we propose Dr. MAS, a simple and stable RL training recipe for multi-agent LLM systems. Dr. MAS uses an agent-wise remedy: normalizing advantages per agent using each agent's own reward statistics, which calibrates gradient scales and dramatically stabilizes training, both theoretically and empirically. Beyond the algorithm, Dr. MAS provides an end-to-end RL training framework for multi-agent LLM systems, supporting scalable orchestration, flexible per-agent LLM serving and optimization configs, and shared resource scheduling of LLM actor backends. We evaluate Dr. MAS on multi-agent math reasoning and multi-turn search benchmarks using Qwen2.5 and Qwen3 series models. Dr. MAS achieves clear gains over vanilla GRPO (e.g., +5.6\% avg@16 and +4.6\% pass@16 on math, and +15.2\% avg@16 and +13.1\% pass@16 on search) while largely eliminating gradient spikes. Moreover, it remains highly effective under heterogeneous agent-model assignments while improving efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。