张大妈

LLM之Agent(四十四)|使用 GRPO 算法训练多智能体系统用于复杂任务规划

源自知乎:Arron

02-20 12:52

长期任务中,多智能体系统易因规划失误导致失败。GRPO算法通过强化学习机制,让智能体在组内比较中学习,显著提升任务规划的准确性和稳定性,有效减少幻觉与偏离轨道问题。

LLM之Agent(四十四)|使用 GRPO 算法训练多智能体系统用于复杂任务规划智能速览

  • GRPO算法通过比较组内多次尝试结果,强化优于平均表现的策略,而非单次动作评分。

  • 多智能体系统的表现高度依赖规划阶段,规划错误是导致任务偏离轨道和幻觉的主要原因。

  • 训练数据合并了DeepMath-103K与Natural Questions数据集,形成包含18.2万样本的多样化训练集。

  • 系统架构包含规划、执行、验证三个核心智能体,协同完成多步推理任务。

  • 训练采用小型可训练模型(Qwen 1.5B)与大型固定模型(Qwen 2.5B, GPT-4o)结合的策略。

  • 通过GRPO训练后,智能体在工具使用和推理步骤上的表现得到改善,任务成功率提升。

LLM之Agent(四十四)|使用 GRPO 算法训练多智能体系统用于复杂任务规划精华内容

要真正理解GRPO如何优化多智能体系统,需要深入其实现细节,从数据准备到核心训练循环,一步步构建一个能自我优化的智能体架构。

挑战与GRPO解法

现有的多智能体Agent系统在处理长期任务时,普遍存在一个核心挑战:系统依赖推理,且每个组件在每次交互时都从零开始,缺乏预训练。这导致在任务执行的任何阶段,都可能出现规划失误或工具调用错误,且错误会不断累积。

GRPO(广义强化策略优化)算法为这一问题提供了创新的解决思路。它摒弃了传统强化学习对单个动作进行评分的机制,转而要求智能体对同一问题进行多次尝试,形成一个“组”。随后,系统会比较组内所有尝试的结果,并智能地强化那些表现优于组内平均水平的策略。通过这种组内比较与优胜劣汰的机制,GRPO能够更有效地引导智能体学习正确的规划和执行方式。

多智能体架构设计

一个高效的多智能体系统需要清晰的架构分工。该实现方案将问题拆解为规划、执行和验证三个核心阶段,分别由不同的子智能体负责。

Planner智能体负责分析当前状态(包括原始查询、历史记录和可用工具),并决定下一步应该执行哪个工具以及具体目标。Executer智能体则将高层次的子目标转化为精确的Python代码并执行,实现与外部世界的交互。执行完毕后,Verifier智能体会对当前状态进行全面审查,判断任务是否已经完成,并发出停止或继续的信号。整个流程通过一个Memory类记录所有历史,为后续决策提供上下文,确保智能体能够记住自己的行为和结果。

数据准备与工具集成

为了让智能体学会处理多样化任务,训练数据集的构建至关重要。该方案将DeepMath-103K数据集(包含10.3万个数学问题及答案)与Natural Questions数据集(包含真实用户查询及维基百科答案)进行合并与标准化处理。最终形成了一个包含182,190个样本的综合训练集,并用Parquet格式高效存储,确保了训练的多样性和效率。

智能体的能力来源于其工具集。系统中集成了多个关键工具,例如:能在沙盒环境中安全执行LLM生成代码的Python_Coder_Tool;利用Google Gemini API进行实时网络搜索的Google_Search_Tool;以及针对特定URL内容进行深度检索的RAG_Tool。这些工具通过统一的BaseTool接口进行管理,使得Planner智能体可以灵活调用。

GRPO训练核心实现

训练环节的设计是整个方案的精髓。为了兼顾效率与效果,训练采用了大小模型协同的策略:使用一个参数量较小的Qwen 1.5B模型作为可训练的策略模型,并应用LoRA技术进行高效微调,仅训练1.1%的参数(约1670万)。同时,选用更强大的Qwen 2.5B模型作为固定的环境模型(模拟执行者和验证者),并使用GPT-4o作为奖励模型来评估最终答案的正确性。

训练循环遵循GRPO流程:策略模型生成规划,环境模型执行并生成轨迹,奖励模型给出二元奖励(成功1.0,失败0.0)。随后,系统计算组内轨迹的相对优势,并基于PPO(近端策略优化)损失函数来更新策略模型的权重,通过梯度下降逐步优化智能体的规划能力。

训练效果与验证

在训练过程中,可以清晰地观察到模型的进步。随着训练步数的增加,PPO损失函数值呈现下降趋势,而每一组轨迹的平均奖励分数则稳步上升。这表明策略模型正在逐步学会生成更有效的规划,从而获得更高的任务成功率。

训练完成后,将微调好的模型部署为vLLM服务器,并使用同样复杂的查询进行测试。结果显示,经过GRPO训练的智能体在工具使用的合理性、推理步骤的逻辑性以及最终答案的准确性上,相比于未训练的基线模型都有显著改善,有效减少了先前出现的错误规划和偏离轨道的现象。

通过GRPO训练,多智能体系统在复杂任务规划上的表现得到显著提升,有效降低了错误率。这为构建更可靠、更自主的AI智能体开辟了新路径,未来能否在更多开放世界任务中取得突破?

内容由AI生成
0
扫一下,分享更方便,购买更轻松
0评论

当前文章无评论,是时候发表评论了
提示信息

取消
确认
评论举报

最新文章 热门文章