分布式异步强化学习框架训练32B大模型:INTELLECT-2

INTELLECT-2 模型详解

一、模型概述

INTELLECT-2 是一个拥有 320 亿参数的语言模型,其训练采用了一种创新的方式,即通过社区贡献的分布式、无需许可的 GPU 资源进行强化学习训练。该模型基于 qwen2 架构构建,因此与 vllm 或 sglang 等流行库和推理引擎具有良好的兼容性。它在训练过程中运用了 prime-rl 这一分布式异步强化学习框架,并结合了基于可验证奖励的 GRPO 方法以及一些提升训练稳定性的改进措施。在训练过程中,INTELLECT-2 特别强调了长度控制预算,建议在指令后附加"Think for 10000 tokens before giving a response."的提示,以实现最佳效果。不过,由于训练时间不足以完全掌握长度控制目标,除了 10,000 之外,2,000、4,000、6,000 和 8,000 等目标长度也能带来较好的结果,因为这些长度在训练过程中也有所涉及。

二、模型架构

INTELLECT-2 以 QwQ-32B 为基础模型,该模型在数学和编码能力方面取得了显著提升,但在 IFEval 上的表现略有下降,这可能与训练数据不够多样化以及对数学和编码的过度专注有关。在架构方面,INTELLECT-2 采用 qwen2 架构,这种架构设计使其能够与多个流行的库和推理引擎实现兼容,从而为模型的应用提供了更大的灵活性和便利性。

三、训练方法

训练 INTELLECT-2 时,采用了 prime-rl 框架,这一框架专为分布式异步强化学习而设计。训练过程中,运用了基于可验证奖励的 GRPO 方法,并且对训练稳定性进行了一些改进。训练数据集主要包括可验证的数学和编码任务,这些任务构成了模型训练的基础,帮助模型在相关领域取得了性能提升。

四、训练数据集

INTELLECT-2 的训练数据集为 PrimeIntellect/Intellect-2-RLDataset,涵盖了各种数学和编码任务,这些任务经过精心设计和验证,以确保模型能够在这些关键领域获得有效的训练,从而提升其在数学和编码方面的能力。

五、性能表现

在性能方面,INTELLECT-2 在多个评估基准上与 QwQ-32B 和其他模型进行了对比。具体来说,在 AIME24、AIME25、LiveCodeBench (v5)、GPQA-Diamond 和 IFEval 等评估基准上,INTELLECT-2 的表现如下:

  • 在 AIME24 上,INTELLECT-2 达到了 78.8 的分数,相较于 QwQ-32B 的 76.6 有所提升。

  • 在 AIME25 上,INTELLECT-2 的分数为 64.9,略高于 QwQ-32B 的 64.8。

  • 在 LiveCodeBench (v5) 上,INTELLECT-2 取得了 67.8 的成绩,高于 QwQ-32B 的 66.1。

  • 在 GPQA-Diamond 上,INTELLECT-2 的分数为 66.8,与 QwQ-32B 的 66.3 相比略有进步。

  • 然而,在 IFEval 上,INTELLECT-2 的分数为 81.5,低于 QwQ-32B 的 83.4。

与其他模型相比,如 Deepseek-R1 在 AIME24 上获得了 78.6 分,在 AIME25 上为 65.1 分,在 LiveCodeBench (v5) 上为 64.1 分,在 GPQA-Diamond 上为 71.6 分,在 IFEval 上为 82.7 分。这些数据表明 INTELLECT-2 在数学和编码能力上有一定的优势,但在 IFEval 上的表现仍有提升空间。

六、核心技术总结

相关推荐
亚马逊云开发者2 小时前
【Bedrock AgentCore】AI Agent 回答不一致怎么办?双 Memory 架构实现服务标准化(附完整代码)
大数据·人工智能·架构
悟纤3 小时前
Seedance 2.0 API 已上线 | 支持「人像视频生成」|支持100并发 | 满血版 [灵龙AI API]
人工智能·音视频·seedance 2.0
Ulyanov3 小时前
Apache Kafka在雷达仿真数据流处理中的应用
分布式·python·kafka·apache·雷达电子战
男孩李3 小时前
什么是workbuddy
人工智能·语言模型
人工智能AI技术3 小时前
终身学习 Agent:积累知识、不遗忘、可进化
人工智能
DFCED3 小时前
突发!Sora 之父 Bill Peebles 离职:OpenAI 理想主义的又一次落幕
人工智能·大模型·agent·sora
_Evan_Yao3 小时前
RAG中的“Chunk”艺术:我试过10种切分策略后总结的结论
java·人工智能·后端·python·软件工程
拾薪3 小时前
[SuperPower] Brainingstorm - 流程控制架构分析
网络·人工智能·ai·架构·superpower·brainstorming
AI自动化工坊3 小时前
SemaClaw开源框架实践指南:从提示工程到安全工程的AI代理基础设施变革
人工智能·开源·ai agent·semaclaw
今天你TLE了吗3 小时前
LLM到Agent&RAG——AI概念概述 第二章:提示词
人工智能·笔记·后端·学习