小智头像图片
AI动态资讯 2025年04月26日
0 收藏 0 点赞 44 浏览 2099 个字
摘要 :

​​李飞飞团队复现R1框架,训练Agent深度思考​​: 什么开源算法自称为DeepSeek-R1(-Zero) 框架的第一个复现? 新强化学习框架RAGEN,作者包括DeepSeek前员工Zihan Wang……

哈喽!伙伴们,我是小智,你们的AI向导。欢迎来到每日的AI学习时间。今天,我们将一起深入AI的奇妙世界,探索“​​李飞飞团队复现R1框架,训练Agent深度思考​​”,并学会本篇文章中所讲的全部知识点。还是那句话“不必远征未知,只需唤醒你的潜能!”跟着小智的步伐,我们终将学有所成,学以致用,并发现自身的更多可能性。话不多说,现在就让我们开始这场激发潜能的AI学习之旅吧。

​​李飞飞团队复现R1框架,训练Agent深度思考​​

​​李飞飞团队复现R1框架,训练Agent深度思考​​:

什么开源算法自称为DeepSeek-R1(-Zero) 框架的第一个复现?

新强化学习框架RAGEN,作者包括DeepSeek前员工Zihan Wang、斯坦福李飞飞团队等,可训练Agent在行动中深度思考。

​​李飞飞团队复现R1框架,训练Agent深度思考​​

论文一作Zihan Wang在DeepSeek期间参与了Deepseek-v2和Expert Specialized Fine-Tuning等工作,目前在美国西北大学读博。

他在介绍这项工作时上来就是一个灵魂提问:为什么你的强化学习训练总是崩溃?

而RAGEN正是探讨了使用多轮强化学习训练Agent时会出现哪些问题 ,以及如何解决这些问题。

​​李飞飞团队复现R1框架,训练Agent深度思考​​

通过大量实验,研究团队发现了训练深度推理型Agent的三大难点:

Echo Trap(回声陷阱):多轮强化学习中,模型过度依赖局部收益的推理,导致行为单一化、探索能力衰退,从而影响长期收益。
数据质量:Agent生成的交互数据直接影响强化学习的效果。合理的数据应该具有多样性、适度的交互粒度和实时性。比如在单个任务上多试几次,每轮限制5-6个动作,并保持rollout的频繁更新。
缺乏推理动机:如果没有精心设计的奖励函数,Agent很难学会多轮任务中持续的推理能力。甚至会出现表面看起来能完成任务,实际上只是匹配了固定模式的假象。下一步的关键在于建立更细粒度、面向解释的奖励机制。

在交互式随机环境中训练推理Agent
RAGEN是一个模块化的Agent训练和评估系统,基于StarPO(State-Thinking-Actions-Reward Policy Optimization)框架,通过多轮强化学习来优化轨迹级别的交互过程,由两个关键部分组成:

MDP Formulation
将Agent与环境的交互表述为马尔可夫决策过程 (MDP),其中状态和动作是token序列,从而允许在环境动态上推理。

​​李飞飞团队复现R1框架,训练Agent深度思考​​

StarPO:通过轨迹级优化强化推理
StarPO是一个通用的强化学习框架,用于优化Agent的整个多轮交互轨迹,在两个阶段之间交替进行,支持在线和离线学习。

Rollout阶段:

给定初始状态,该模型会生成多条轨迹。在每一步中,模型都会接收轨迹历史记录并生成推理引导的动作。

…reasoning process… action

环境接收动作并返回反馈(奖励和下一个状态)。

​​李飞飞团队复现R1框架,训练Agent深度思考​​

Update阶段:多回合轨迹优化

生成轨迹后,训练优化预期奖励。StarPO并非采用逐步优化的方式,而是使用重要性采样来优化整个轨迹。这种方法能够在保持计算效率的同时实现长远推理。

StarPO支持PPO、GRPO等多种优化策略。

​​李飞飞团队复现R1框架,训练Agent深度思考​​
​​李飞飞团队复现R1框架,训练Agent深度思考​​

除提出算法外,RAGEN论文中还重点介绍了通过研究推理稳定性和强化学习动态得出的6点主要发现。

6点主要发现

发现1:多轮训练引入了新的不稳定模式

像PPO和GRPO这样的单轮强化学习方法的adaptations在Agent任务中有效,但经常会崩溃。PPO中的“批评者”或许可以**延缓不稳定性,但无法阻止推理能力的下降,这凸显了在Agent任务中对专门的稳定性进行改进的必要性。

发现2:Agent强化学习中的模型崩溃体现为训练过程中的“回声陷阱”

早期智能体会以多样化的符号推理做出反应,但训练后会陷入确定性、重复性的模板。模型会收敛到固定的措辞,这表明强化学习可能会强化表面模式而非一般推理,并形成阻碍长期泛化的“回声陷阱”。

发现3:崩溃遵循类似的动态,可以通过指标预测

奖励的标准差和熵通常会在性能下降之前发生波动,而梯度范数的峰值通常标志着不可逆崩溃的临界点。这些指标提供了早期指标,并激发了对稳定策略的需求。

发现4:基于不确定性的过滤提高了训练的稳定性和效率
基于奖励方差过滤训练数据可以有效对抗“回声陷阱”。仅保留高度不确定的训练实例可以延迟或防止跨任务崩溃,并提高数据效率。

发现5:任务多样性、行动预算和推出频率影响数据质量

多样化的任务实例能够实现更好的策略对比和跨环境泛化。合适的行动预算能够提供充足的规划空间,并避免过长序列引入的噪声。Up-to-date rollouts能够确保优化目标与当前策略行为保持一致。

发现6:如果没有精心的奖励设计,推理行为就无法产生

虽然符号推理在弱监督下的单轮任务中自然出现,但在多轮环境中,如果没有明确鼓励可解释的中间推理步骤的奖励设计,它就无法持续存在。

团队观察到,即使有结构化的提示,如果奖励信号仅关注最终结果,推理能力也会在训练过程中逐渐衰退。这表明如果没有细致的奖励塑造,智能体可能会倾向于走捷径,完全绕过推理。

One More Thing

同团队还有另一个项目VAGEN,使用多轮强化学习训练多模态Agent。

VAGEN 引入了回合感知推理交互链优化 (TRICO) 算法,通过两项关键创新扩展了传统的RICO方法:选择性token屏蔽,跨轮credit分配。

与传统的Agent强化学习相比,VAGEN不会平等对待轨迹中的所有token,而是重点优化最关键的决策token并在交互过程中创建更细致的奖励结构,更适合多模态Agent

​​李飞飞团队复现R1框架,训练Agent深度思考​​
RAGEN、VAGEN代码均已开源,感兴趣的团队可以跑起来了。

嘿,伙伴们,今天我们的AI探索之旅已经圆满结束。关于“​​李飞飞团队复现R1框架,训练Agent深度思考​​”的内容已经分享给大家了。感谢你们的陪伴,希望这次旅程让你对AI能够更了解、更喜欢。谨记,精准提问是解锁AI潜能的钥匙哦!如果有小伙伴想要了解学习更多的AI知识,请关注我们的官网“AI智研社”,保证让你收获满满呦!

微信打赏二维码 微信扫一扫

支付宝打赏二维码 支付宝扫一扫

版权: 转载请注明出处:https://www.ai-blog.cn/15491.html

相关推荐

三星李在镕出击:与Meta、高通共商AI合作大计: 在科技行业不断探索创新边界的当下,三星、Meta和…

小智头像图片
143

加州拟立法:儿童AI聊天需定期提示“非人类”: 儿童AI依赖频发 加州率先启动立法监管 近日,加州议…

小智头像图片
165

深圳AI教育大放异彩:超200万学生畅享智慧课堂: 在教育领域的创新探索中,深圳一直走在前列。如今…

小智头像图片
133

人类照片伪装AI图夺季军 艺术界掀起“真假之辩”: 一场“测试”引发的风波:真实照片混入AI图像赛 近…

小智头像图片
75

抖音AI美食怪兽爆火:6条视频1700万播放的创意奇迹: 在抖音这个充满创意的平台上,总是有各种新奇…

小智头像图片
33

360AI发布会陷“1元侵权案” 演示图片版权争议引关注: 新品发布遇“意外插曲” 演示图片被指侵权 6月1…

小智头像图片
150

小米小爱同学接入豆包大模型,AI交互再升级: 在人工智能技术飞速发展的当下,科技巨头之间的合作…

小智头像图片
84

复旦评测:阿里千问、讯飞星火高考数学超GPT-4o: 高考数学成大模型“新考场” 国内双雄击败国际顶流…

小智头像图片
146
发表评论
暂无评论

还没有评论呢,快来抢沙发~

助力原创内容

快速提升站内名气成为大牛

扫描二维码

手机访问本站

二维码
vip弹窗图片