用分组强化学习优化大模型多智能体搜索系统,提升效率与稳定性。
End-to-End Optimization of LLM-Driven Multi-Agent Search Systems via Heterogeneous-Group-Based Reinforcement Learning
- 通过异构分组策略评估相对优势,替代传统大型价值网络。
- 在多个任务上超越基线模型,成功率提升12%以上,内存消耗降低40%。
- 适合需要高效协同的大模型应用开发人员或研究者参考。
大型语言模型(LLM)虽具通用性,但受限于静态知识截止和单次推理中行为控制困难,在复杂现实场景中部署受限。多智能体搜索系统(MASS)通过任务分解与检索增强的协作方式,缓解这些问题。然而,为各智能体角色优化LLM仍依赖耗时的提示工程或监督微调,亟需自动化端到端训练。现有多智能体强化学习(MARL)方法如多智能体近端策略优化(MAPPO)通常依赖大型评价网络来评估联合动作,导致训练不稳定且内存开销高。本文提出多智能体异构分组策略优化(MHGPO),通过在异构多智能体回放序列间估计相对优势来更新策略,将优化重点从局部智能体表现转向全局系统成功。进一步研究了三种分组回放采样策略,在样本效率与优化质量间实现权衡。实验表明,MHGPO能捕捉隐式智能体依赖关系,显著优于强基线,在任务性能和计算效率上均表现更优。
原文摘要 · Abstract (English)
Large language models (LLMs) are versatile, yet their deployment in complex real-world settings is limited by static knowledge cutoffs and the difficulty of producing controllable behavior within a single inference. Multi-agent search systems (MASS), which coordinate specialized LLM agents equipped with search tools, mitigate these issues via task decomposition and retrieval-augmented problem solving. However, optimizing LLMs for agent-specific roles remains labor-intensive with prompt engineering or supervised fine-tuning, motivating automated end-to-end training. Existing multi-agent reinforcement learning (MARL) methods such as Multi-Agent Proximal Policy Optimization (MAPPO) typically depend on large critic networks to evaluate joint actions, leading to instability and high memory costs. We introduce Multi-Agent Heterogeneous Group Policy Optimization (MHGPO), which updates policies by estimating relative advantages across heterogeneous groups of multi-agent rollouts, shifting the optimization focus from local agent performance to global system success. We further study three group rollout sampling strategies to trade off sample efficiency and optimization quality. Experiments show that MHGPO captures implicit inter-agent dependencies and consistently outperforms strong baselines in both task performance and computational efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。