开源22万条DeepSeek R1的高质量数据!你也能复现DeepSeek了

DeepSeek 在海内外搅起的惊涛巨浪,余波仍在汹涌。 当中国大模型撕开硅谷的防线之后,在预设中总是落后半拍的中国 AI 军团,这次竟完成了一次反向技术输出,引发了全球范围内复现 DeepSeek 的热潮。 DeepSeek-R1 虽然开源,但也没有完全开源,训练数据、训练脚本等关键信息并未完全公布。

DeepSeek 在海内外搅起的惊涛巨浪,余波仍在汹涌。

当中国大模型撕开硅谷的防线之后,在预设中总是落后半拍的中国 AI 军团,这次竟完成了一次反向技术输出,引发了全球范围内复现 DeepSeek 的热潮。

DeepSeek-R1 虽然开源,但也没有完全开源,训练数据、训练脚本等关键信息并未完全公布。

不过,有技术报告,相当于拥有着复现 R1 的指导方针,已经有不少执行力强的团队用小模型见证「aha moment」了。在浩浩荡荡的复刻大军中,最令人瞩目的,当数 Hugging Face 领衔的 Open R1 项目。

Open R1 宣称要做到完全开放复现 DeepSeek-R1,补齐 DeepSeek 所有未公开的技术细节。Open R1 项目刚刚启动不过几周,他们已经完成了:

  • GRPO 实现

  • 训练与评估代码

  • 用于合成数据的生成器

图片

项目地址:https://github.com/huggingface/open-r1

在开源社区的众志成城,Open R1 更是动作迅速。今天,他们发布了 OpenR1-Math-220k 数据集,又补全了一块 DeepSeek R1「碎片」—— 合成数据。

图片

                              OpenR1-Math-220k 数据集概览

数据集链接:https://huggingface.co/datasets/open-r1/OpenR1-Math-220k

DeepSeek R1 的一个重要优势在于它能够将高级推理能力迁移到较小的模型中。DeepSeek 团队生成了 60 万条推理数据,在 Qwen 和 Llama 等开源模型上证明了这种迁移能力。即使不使用强化学习,直接从 R1 模型进行迁移也能实现强大的推理性能。

然而,这些合成数据仅 DeepSeek 可见,未对其他团队开放阅读权限。

OpenR1-Math-220k 数据集就是来补上这块空缺的。具体而言,Open R1 团队使用 DeepSeek R1 生成了 80 万条推理轨迹,经过筛选和验证后得到了 22 万条高质量数据。

这些数据可以用来支持更小的模型,来达到媲美 DeepSeek R1 的效果。比如在 OpenR1-Math-220k 数据集上训练出来的 Qwen-7B-Math-Instruct,达到了与 DeepSeek-Distill-Qwen-7B 相当的性能

下面就让我们来看看 OpenR1-Math-220k 数据集的特点,以及它是怎么诞生的:

自动过滤正确答案

本地高效生成

目前,开源社区已经发布了 OpenThoughts-114k、Bespoke-Stratos-17k、Dolphin-R1 和 LIMO 等多个推理数据集。

然而,为了进行推理公众公开发布,这促使社区独立重建类似的数据集。另外,社区已经发布了多个开放数据集,包括 OpenThoughts-114k、Bespoke-Stratos-17k、Dolphin-R1 和 LIMO。

为了整理 OpenR1-Math-220k,Open R1 还和开发了广受欢迎的 NuminaMath-CoT 数据集开发团队 Numina 进行了合作。

那么,与现有数据集相比,Open R1 的数据集有什么新特点呢?

  • 80 万条 R1 推理轨迹:使用 DeepSeek R1 为 40 万个问题各生成了两个答案,最终经过筛选后保留了 22 万个具有正确推理轨迹的问题。

  • 本地运行 512 个 H100:没有依赖 API,而是在计算集群上利用 vLLM 和 SGLang 本地运行生成任务,每天可以生成 18 万条推理轨迹。

  • 基于 NuminaMath 1.5:专注于数学推理公式,为 NuminaMath 1.5(NuminaMath-CoT 数据集的改进版本)中的问题生成答案。

  • 自动过滤:Open R1 团队通过数学验证,只保留至少有一个正确答案的问题,还让 Llama3.3-70B-Instruct 作为「判官」,以筛选出更多正确的样本,特别是那些因格式错误而无法通过基于规则的解析器验证的答案。

  • 在 OpenR1-Math-220k 训练出来的 Qwen-7B-Math-Instruct,达到了与 DeepSeek-Distill-Qwen-7B 相当的性能。

数据集分为两个部分:

  • default(94k 问题):这部分数据在经过监督微调(SFT)后表现最佳。

  • extended(131k 问题):这部分数据包含额外的 NuminaMath 1.5 数据源,例如 cn_k12,提供了更多的推理公式。(研究发现这个子集在经过监督微调后的性能低于默认数据集,可能是因为 cn_k12 中的问题相对简单。)

图片

Open R1 团队表示,希望这个可扩展的、高质量的推理数据生成过程,能够启发代码生成等数学之外的领域。

数据生成

为了构建数据集,OpenR1 团队让 DeepSeek R1 为来自 NuminaMath 1.5 的 40 万个问题生成答案。他们遵循了 DeepSeek 技术报告中推荐的参数设置,并在提示词前添加了以下指令:

Please reason step by step, and put your final answer within \boxed{}.

为了确保生成过程的高效性,团队将每次生成的 tokens 限制设置为 16k。经过分析发现,只有 75% 的问题能够在 8k tokens 内解决,而大多数剩余问题需要完整的 16k tokens。

最初,他们使用 vLLM 进行推理,每个 H100 节点每秒可以生成 15 个答案,并且相关生成脚本已分享在 OpenR1 仓库中。最近,他们又开始尝试使用 SGLang,每个 H100 节点每秒可以生成 25 个答案(速度提升了近两倍),这使得 512 个 H100 节点上每天能生成 30 万个问题的答案。

为了在后续的过滤和优化过程中提供更大的灵活性,团队为每个问题生成了两个答案 —— 有时甚至生成四个。这样一来,不仅复刻出了类似于 DeepSeek R1 允许进行拒绝采样的方法,还能使数据集能够适用于如 DPO 等偏好优化方法。

数据生成脚本:https://github.com/huggingface/open-r1/tree/main/slurm。

数据过滤

为了确保数据集中只包含高质量且正确的推理结果,Open R1 团队设计了一套数学验证系统,用于自动比对 LLM 生成的复杂数学表达式答案与数据集中的标准答案。

在这个过程中,OpenR1 团队发现大约 55% 的问题至少有一个正确答案。然而,NuminaMath 1.5 数据集中有很多答案是空的,或者格式不符合验证标准,这都给自动验证带来了困难。

为了解决这些问题,Open R1 团队先是对 Math-Verify 工具进行了改进,使其能够处理更多不常见的答案格式,再使用 Llama-3.3-70B-Instruct 模型进行二次评估。

具体来说,对于那些被 Math-Verify 判定为错误的答案,使用 Llama-3.3-70B-Instruct 模型重新评估,识别实际上正确但因格式问题被错判的答案。最终,他们找回了 2.5 万条被「误判」的数据。

优化 Math-Verify 工具:对 Math-Verify 工具进行了改进,使其能够处理更多不常见的答案格式。

让 Llama-3.3-70B-Instruct 「作判官」的提示词如下:

You are a mathematical answer validator. You will be provided with a mathematical problem and you need to compare the answer in the reference solution, and the final answer in a model's solution to determine if they are equivalent, even if formatted differently.

PROBLEM:

{problem}

REFERENCE SOLUTION:

{answer}

MODEL'S SOLUTION:

{generation}

Focus ONLY on comparing the final mathematical answer provided by the model while ignoring differences in:

- Formatting (e.g., \\boxed{{}} vs plain text)

- Multiple choice formatting (e.g., "A" vs full solution)

- Order of coordinate pairs or solutions

- Equivalent mathematical expressions or notation variations

- If the model's answer is nonsense, return "Verdict: AMBIGUOUS"

Start with a brief explanation of your comparison (2-3 sentences). Then output your final answer in one of the following formats:

- "Verdict: EQUIVALENT"

- "Verdict: DIFFERENT"

- "Verdict: AMBIGUOUS"

对于那些包含多个正确答案的数据行,团队尝试使用奖励模型(RM)作为最终筛选器来选择最佳答案。具体操作如下:

首先,从每个包含多个正确答案的数据行中,去掉(<think>…</think>),提取最终答案;第二,将问题和提取的答案输入到配置了 vLLM 的 Qwen/Qwen2.5-Math-RM-72B 模型中,获取每个答案的评分;接着,根据模型评分,对每个包含多个正确答案的数据行排名,选择排名最高的答案纳入训练数据集。

遗憾的是,消融实验表明,这种方法并没有比随机选择一个正确答案带来更好的模型性能。Open R1 团队的判断是,可能在使用奖励模型评分时,不仅要考虑最终答案,还要包括推理过程。

Open R1 训练出来的新模型 VS DeepSeek-Distill-Qwen-7B

Open R1 在 OpenR1-Math-220k 的基础上,对 Qwen2.5-Math-Instruct 进行了 3 轮微调,学习率为 5e-5。

为了将上下文长度从 4k 扩展到 32k,他们将 RoPE 频率提高到 300k。训练遵循线性学习率调度,其中包含 10% 的预热阶段。

下表展示了在 lighteval 上 OpenR1-Qwen-7B、DeepSeek-Distill-Qwen-7B 和 OpenThinker-7B 的性能对比,可以看出在数学成绩上,OpenR1-Qwen-7B 和 DeepSeek-Distill-Qwen-7B 差距不是非常明显。

图片

本周 AIME 2025 开赛,来自苏黎世联邦理工学院的研究人员用新题来测评了多款模型,却发现各种模型的数学能力下降了 10-20 个百分点之间。有研究者发现这些「全新」的 AIME 题可能已经在互联网论坛泄露,这有可能造成意外的过拟合问题,这也凸显了新鲜测试数据的困境。

同时,开源社区也从多个角度探索了 GRPO,有多个研究实验室表明,大约 1000 个高质量的训练样本可能就足以在现有的开源模型中引发推理能力

启示和新的问题

LLM 是否需要使用自然语言进行推理?

图片

论文链接:https://arxiv.org/pdf/2502.05171

马里兰大学的一篇论文表明,通过使用循环语言模型,可以在潜在空间中隐式推理,从而在测试时扩展计算能力,这类似于 Meta 的 Coconut。这些方法的优势在于它们的计算效率更高:通过探索潜在空间,无需生成大量「思考」token 即可获得高性能。

图片

是否要转向更小、高质量的推理数据?

尽管 DeepSeek R1 使用了 600k 推理轨迹进行蒸馏,但最新研究表明,复杂的推理能力并非单纯通过大规模训练在语言模型中实现,而是可以通过少量精心设计的样本达成

s1K 数据集就是一个很好的例子。它包含 1000 个经过精心挑选的数学问题,以及从 Gemini Flash 蒸馏出的推理轨迹。在选择问题时,研究者注重难度、多样性和质量。通过在 s1K 数据集上对 Qwen2.5-32B-Instruct 进行微调,研究者成功使其在竞赛数学基准测试中超过了 OpenAI 的 o1-preview,最高提升了 27%。

另一个数据集 LIMO 也进一步验证了这一理念。它仅使用 817 个训练样本,就在 AIME 和 MATH 基准测试中取得了出色的表现。LIMO 的作者推测,当模型在预训练阶段已经积累了丰富的知识后,可能只需要少量结构良好的样本,就能解锁高级推理能力。

CoT 长度:预算强制与奖励塑造

Qwen2.5-32B-Instruct 模型在 s1K 数据集上微调后表现出色,其中一个关键因素是采用了「预算强制」。这是一种测试时的计算技术,通过在模型生成中添加「等待」token 来延长推理时间,或者添加「结束思考」的 token 来截断推理。

这种方法使研究者能够灵活调整模型的思考时间,并发现随着思考时间的增加,模型在不同数学基准测试中的准确性也随之提高。

同样,Yeo 等人在研究《Demystifying Long Chain-of-Thought Reasoning in LLMs》中探讨了思维链(CoT)长度对模型性能的影响。他们引入了一种名为「余弦奖励」的新奖励函数,用于在正确生成时激励较短的 CoT,在错误生成时激励较长的 CoT。这种奖励机制在模型的最大上下文大小有限且平均响应长度可能失控的情况下,能够稳定强化学习训练。

图片

论文链接:https://arxiv.org/pdf/2502.03373

此外,当模型在处理难题时出现奖励劫持的迹象(即通过重复而非真正解决问题来增加 CoT 长度),研究者还会采用重复惩罚机制,以避免模型陷入无效的循环推理。

Open R1 团队表示,现在 GRPO 已经在 TRL 中顺利运行,他们正在开展一系列广泛的实验,以了解哪些超参数和奖励函数对训练的影响最大。

那就让我们静待 Open R1 的好消息吧。

参考链接:

https://huggingface.co/blog/open-r1/update-2

https://x.com/_lewtun/status/1889002019316506684

https://huggingface.co/open-r1

相关资讯

Hugging Face 推出号称“世界上最小的视觉语言模型”SmolVLM-256M

Hugging Face 上周发布了两款全新多模态模型 SmolVLM-256M 和 SmolVLM-500M,其中 SmolVLM-256M 号称是世界上最小的视觉语言模型(Video Language Model)。

arXiv联手Hugging Face,论文链接到开源Demos一步到位

arXiv 作为超大规模的预印本网站,从诞生之初就造福了无数科研工作者。而它存在的意义,远不止让大家免费看论文这么简单。近日,Hugging Face 宣布与 arXiv 合作开发了一项新功能,即 Hugging Face Spaces 通过 Demos 选项卡与 arXivLabs 集成,该选项卡包含指向社区或作者自己创建的 Demos 的链接。用户只需将鼠标点到 Demos 选项卡,就可以查看源代码并试用。

世界最大开源 AI 社区 Hugging Face 涉足开源机器人领域,前特斯拉科学家领衔

据外媒 VentureBeat 报道,AI 创企 Hugging Face 设立开源机器人项目,前特斯拉科学家 Remi Cadene 领衔。Remi Cadene 的 X 平台动态显示,这位前特斯拉人形机器人 Optimus 团队成员加入 Hugging Face 并启动了一个开源机器人项目,招募愿在巴黎建造“真正的机器人”的工程师。参考IT之家以往报道,Hugging Face 目前主营业务均为软件形态,包括开源 AI 模型库和 AI 助手 Hugging Chat Assistants。此次组建开源机器人团队