张大妈

北大AI日记 day10

源自小红薯:学不会大模型

02-01 18:26

深入解析DeepSeek R1的后训练流程,探讨如何通过强化学习提升大模型推理能力,揭示RL训练中的关键技术细节和创新点。

北大AI日记 day10智能速览

  • DS-V3直接应用RL得到DeepSeek-R1-Zero

  • Reward设计关注答案准确和格式正确两方面

  • 异步处理reward时间和推理时间重叠

  • Zero模型推理能力随输出长度显著提升

  • 训练中会出现aha-moment改变推理方法

北大AI日记 day10精华内容

探索大模型推理能力的提升路径,DeepSeek R1通过创新的RL训练方法,展现了模型自主学习的潜力。

RL训练架构

在基模的后训练场景中,SFT体现的是人类规定的表达方式,可能限制模型自由的推理过程。而RL恰好给予模型自由探索的空间,有助于逻辑推演。基于这个假设,DeepSeek团队在DS-V3-basemodel上直接应用RL(GRPO)得到DeepSeek-R1-Zero。

训练过程中使用了CoT样式的提示词,让模型按照特定的思考格式展开推理。RL rollout过程调用vllm进行并行推理,充分利用了基础设施的优化。

Reward设计机制

Reward设计关注答案准确和格式正确两个维度,且两者权重相同:Reward_rule = Reward_accuracy + Reward_format。

避免使用神经网络模型作为reward的原因有:容易陷入reward hacking,以及减少计算开销和工程复杂性。答案正确可以通过函数判断,如数学题直接对比答案,代码补全用编译器运行。格式则用正则匹配打分。

关键优化在于:由于reward一般不占用GPU,将打分处理成异步操作,让reward时间和推理时间重叠,大幅提升训练效率。

推理能力演变

Zero模型的训练过程中展现出几个重要特征:随着输出长度变长,推理能力显著提升,这说明模型确实在进行更深层次的思考。

训练中途会出现aha-moment,模型突然改变推理方法,这种突破性变化标志着模型学习的重要进展。

但Zero模型也存在问题:可读性较差,出现多语言混杂现象。因此zero模型仅作为中间产物,其推理能力被用来生成SFT格式的推理数据,经过多轮筛选得到冷启动SFT数据,为下一阶段做准备。

DeepSeek R1的训练流程展现了RL在提升大模型推理能力方面的巨大潜力,通过巧妙的设计让模型自主探索推理路径。这种训练范式不仅提升了模型性能,更为大模型的进化方向提供了新的思路。

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

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

取消
确认
评论举报

最新文章 热门文章