DeepSeek-R1解读:纯强化学习,模型推理能力提升的新范式?

部署运行你感兴趣的模型镜像

 Datawhale干货 

者:Jimmy.DU,Datawhale成员

1. Impressive Points

  1. LLM模型推理能力提升

    1. 在LLM模型post-training中,仅使用强化学习(reinforcement learning,RL) 提升模型推理能力,不再依赖有监督微调训练(supervised fine-tuning,SFT)。

    2. 证明了LLM模型具有自行探索长思维链(chain-of-thought,COT) 的能力。

  2. 端侧模型(小模型)推理能力提升

    1. 相对于使用RL进行训练,基于大模型进行蒸馏(Distillation)的方式,是提升端侧模型推理能力更有效的途径。

2. 纯强化学习,LLM推理能力提升新范式?

2.1 DeepSeek-R1-Zero

核心问题: 当前的post-training流程对于大量监督数据的依赖,监督数据的收集非常耗时:

  1. 当前模型推理性能的提升,需要大量监督数据进行SFT,以作为模型post-training的冷启动。

  2. 当前一些研究已经验证了强化学习在模型推理性能上的有效性,但也依赖监督数据。

解决方案: 探索在没有任何监督数据的情况下,提升LLM模型的推理能力:

  1. 为了节省RL的训练成本,采用群体相对策略优化(GRPO),这个这里就不再多说了,后面专门出一篇文章讲一下GRPO。

  2. 在RL训练过程中,采用Rule-based奖励,主要由两种奖励构成:

    1. Accuracy rewards:评估模型的输出是否正确。

    2. Format rewards:强制模型将其思考过程置于指定标签之间。

  3. 设计训练模版,指导基模型在训练过程中遵守设定的指令:

10258e40527a0d424a93a3567da82f81.png

成果:

  1. 推出DeepSeek-R1-Zero模型,无需任何监督微调数据,仅通过RL进行模型的post-training,在AIME2024、MATH-500等多个Benchmark中达到并且超过OpenAI-o1-0912的水平。

1990672b1305fef617d36053a6bfbd79.png
  1. DeepSeek-R1-Zero展示出了自我进化(self-evolution) 能力,在没有监督数据的情况下,随着强化学习训练进程的深入,模型的思考时间在增加,并自发出现了诸如reflectio(反射,模型重新审视和重新评估其先前步骤)以及探索解决问题的替代方法等更加复杂的行为:

a565da91717482d0d7f0144e3e88c3d7.png
  1. 在DeepSeek-R1-Zero的训练过程中出现了Aha Moment(顿悟时刻),代表RL有可能在人工系统中解锁新的智能水平,为未来更加自主和自适应的模型铺平道路。

916f19869c81ec221f86dd493c6ab7f5.png

2.2 DeepSeek-R1

核心问题:

  1. 相对于完全不使用有监督数据,使用少量高质量数据作为冷启动,是否可以进一步提高推理性能或加速收敛?

  2. 针对DeepSeek-R1-Zero存在的输出内容可读性差的问题进行优化。

解决方案:

  1. 冷启动数据: 使用下述方法构建少量的(约几千条)长COT数据,作为冷启动数据对DeepSeek-V3-Base进行微调:

    1. 以few-shot的长COT prompt作为例子,让DeepSeek-R1-Zero通过反射和验证生成详细的答案;

    2. 将DeepSeek-R1-Zero的结果进行格式化;

    3. 让人工标注人员进行后处理。

  2. Reasoning-oriented Reinforcement Learning: 完成冷启动数据微调后,采用与DeepSeek-R1-Zero一致的强化学习训练过程,同时针对DeepSeek-R1-Zero存在的语言混合,导致模型输出可读性差的问题,在RL训练期间引入语言一致性奖励(目标语言单词在 CoT 中的比例),将推理任务的准确性和语言一致性的奖励结合起来,直接相加作为最终的奖励。

  3. Rejection Sampling and Supervised Fine-Tuning: 当2中的RL过程趋于收敛时,利用checkpoint生产用于下一轮训练的SFT数据。与1中的冷启动数据区别在于,冷启动数据针对推理能力提升,此阶段既包含用于推理能力提升的600k数据,也包含200k推理无关的数据。使用上述约800k样本的精选数据集继续对DeepSeek-V3-Base进行了两个epoch的微调。

  4. Reinforcement Learning for all Scenarios: 为了进一步对齐模型和人类偏好,设计了二级强化学习阶段以同时提高模型的helpfulness(有用性) 和harmlessness(无害性)

    1. helpfulness(有用性):只评估模型最终的结果,而不关注模型的推理过程

    2. harmlessness(无害性):既评估模型最终的结果,也评估模型的推理过程。

3. 端侧模型能力提升:蒸馏>强化学习

基于DeekSeek-R1,文中仅使用SFT对小模型(Qwen、Llama等)进行蒸馏训练得到的模型,性能全面优于GPT-4o-0513等大参数量非推理模型:

6d225382e710890c78f82d6040b1089d.png

同时,直接对小模型进行DeepSeek-R1-Zero同款的强化学习,得到的DeepSeek-R1-Zero-Qwen-32B模型性能弱于蒸馏模型

23ba146af104cc876632d5b362eb3ff5.png

Tips:文中提到将RL应用于蒸馏模型会产生显著的进一步收益,应用方法文中没有详细说明,留给学术界去进一步探索。


作者:Jimmy.DU,Datawhale成员

往期作品:一文详尽之Embedding(向量表示)!

9be5254145172ccc1b2a72e0758778ea.png

一起三连↓

您可能感兴趣的与本文相关的镜像

EmotiVoice

EmotiVoice

AI应用

EmotiVoice是由网易有道AI算法团队开源的一块国产TTS语音合成引擎,支持中英文双语,包含2000多种不同的音色,以及特色的情感合成功能,支持合成包含快乐、兴奋、悲伤、愤怒等广泛情感的语音。

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值