专栏名称: Datawhale
一个专注于AI领域的开源组织,汇聚了众多顶尖院校和知名企业的优秀学习者,聚集了一群有开源精神和探索精神的团队成员。愿景-for the learner,和学习者一起成长。
目录
相关文章推荐
51好读  ›  专栏  ›  Datawhale

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

Datawhale  · 公众号  ·  · 2025-01-27 18:23

正文

请到「今天看啥」查看全文


Datawhale干货

者:Jimmy.DU,Dataw hale成员

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. 设计训练模版,指导基模型在训练过程中遵守设定的指令:

成果:

  1. 推出 DeepSeek-R1-Zero 模型, 无需任何监督微调数据 ,仅通过 RL 进行模型的post-training,在AIME2024、MATH-500等多个Benchmark中达到并且超过OpenAI-o1-0912的水平。
  1. DeepSeek-R1-Zero展示出了 自我进化(self-evolution) 能力,在没有监督数据的情况下,随着强化学习训练进程的深入,模型的思考时间在增加,并自发出现了诸如 reflectio (反射,模型重新审视和重新评估其先前步骤)以及 探索解决问题的替代方法 等更加复杂的行为:
  1. 在DeepSeek-R1-Zero的训练过程中出现了 Aha Moment(顿悟时刻) ,代表RL有可能在人工系统中解锁新的智能水平,为未来更加自主和自适应的模型铺平道路。

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等大参数量非推理模型:

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

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

者:Jimmy.DU,Datawha le成员

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

一起 三连







请到「今天看啥」查看全文