返回 文章 学习 CMS 文章

Ripple-65B Preview:美国开源权重智能体编程新标杆

一款以十分之一成本达到美国开源权重智能体编程 SOTA 的 65B 稠密模型预览版。

Ripple-65B开源权重模型智能体编程模型架构
成长分 / 100 67 综合收获、行动、留存与影响

Ripple-65B Preview:美国开源权重智能体编程新标杆
为什么值得读了解一个从零训练的开源权重模型如何在智能体编程基准上超越多个参数量更大的美国开源模型。

获取低成本、高效率预训练与后训练配方的具体工程细节,包括模型阶梯实验与强化学习技巧。

关键洞察
  1. Ripple-65B Preview 在 TerminalBench 2.1 上得分 80.52,超过 Reflection Beam 501B、Meta Muse Spark Open 816B 等更大规模美国开源权重模型。
  2. 完整研发成本仅 3200 万美元,约为此前智能体编程开源权重 SOTA 成本的十分之一。
  3. 采用现代化稠密 Transformer 而非 MoE,预训练全程维持 40% MFU,且无需自定义内核或量化。
转成行动

深入阅读

正文与原文对照

原文保真覆盖:全文原文字符:7699

Ricursive Intelligence 是一家全栈人工智能实验室,通过协同优化硬件、软件和模型来构建自我改进的系统。

为驱动我们的芯片设计平台,我们从零开始训练了一个语言模型,在智能体编程(agentic coding)上创下新的美国开放权重 SOTA。我们针对训练和服务平台优化了模型架构,带来了显著的效率提升。

今天,我们发布 Ripple-65B Preview,这是驱动我们 AI 芯片设计平台的专有模型的通用变体。该模型在 12 万亿 token 上预训练,支持最长 1M 上下文长度。Ripple-65B Preview 在智能体编程上表现强劲,在 TerminalBench 2.1 上达到美国开放权重最先进水平(80.52),并超越了规模显著更大的美国开放权重模型,如 Reflection Beam 501B(80.1)、Meta Muse Spark Open 816B(76.4)、Thinking Machines Inkling 975B(63.8)和 NVIDIA Nemotron 3 Ultra 550B(56.4)。

Ripple 的稠密模型架构在低延迟 token 生成的同时提供强大的推理性能。该模型较小的参数规模还使其能够在 NVIDIA DGX Spark 等本地设备上高效运行,这些设备相对于数据中心加速器具有高计算 FLOPs 和低内存带宽的特点。

Ripple-65B Preview 已在 Hugging Face、Together AI、GMI Cloud 和 OpenRouter 上提供。Fireworks 将于今天晚些时候上线。

以十分之一成本实现美国开放权重前沿水平

LLM 预训练以昂贵著称。2023 年,OpenAI CEO Sam Altman 表示 GPT-4 的训练成本超过 1 亿美元。次年,Dario Amodei 表示前沿 AI 模型的训练成本可高达 10 亿美元,并预测到 2026 至 2027 年,仅预训练成本就将升至100 亿至 1000 亿美元。

相比之下,我们仅用 3200 万美元就开发并训练了 Ripple-65B Preview,据我们所知,这是此前智能体编程开放权重 SOTA 成本的十分之一。¹ 请注意,3200 万美元是我们从零开始的完整研发成本,包括预训练阶梯开发、数据混合、评估、调试、合成数据生成、后训练配方开发,以及最终的正式预训练和后训练运行。

陡峭上升的强化学习

我们没有投入暴力式 RL 爬坡,而是围绕计算效率,仔细迭代我们的 SFT 和 RL 配方。我们选择不购买任何后训练数据,而是自行设计合成数据流水线、质量过滤和覆盖率指标。

我们的混合 RL 配方使用截断重要性采样、分数居中和线性长度惩罚,并配合高质量长时程合成环境。RL 任务针对我们的模型能力进行调优,随着模型改进,我们持续加入更难的任务。

最终结果是从一个强大的初始 SFT 检查点出发的陡峭 RL 爬坡。我们仅用 140 万次 rollout 就达到了 TerminalBench 2.1 的 80.5 分,rollout 次数比此前 SOTA 的 Reflection Beam 少约 70 倍。

协同设计的模型架构

Ripple-65B Preview 的架构旨在 NVIDIA 硬件上高效训练和服务。一个关键目标是做出能够降低预训练和后训练基础设施复杂性、同时仍能实现强大推理和智能体性能的设计决策。因此,我们选择训练现代化的稠密 Transformer 模型,而不是像其他实验室那样使用混合专家(MoE)。这一决策很快见效,我们能够从零训练出一个强大的模型,并在预训练和后训练全程保持稳定的 MFU(40%),即使只使用简单的分片、没有自定义内核、也没有量化。

Ripple-65B Preview 采用薄而深的模型长宽比,d_model 较窄(5120),共 80 层。为提升推理性能,我们超额配置注意力头数量,同时提高 SWA 与全局层的比例并减小 SWA 窗口大小,以补偿增加的成本。此外,我们加入了FFN 激活裁剪、门控注意力、全局 NoPE和值残差。

细致的模型阶梯带来大幅改进

我们通过一系列细致的模型阶梯实验优化了 Ripple 的预训练配方。

我们的起点是 Olmo 3 的模型阶梯,在此基础上我们在小规模上测试干预措施,重点是提升编程能力。我们的 v1.0 阶梯实现了StepFun 的批量大小和学习率缩放定律,v1.1 实现了我们的新阶梯,可平滑缩放架构维度和模型深度。阶梯 v1.2 从 post-norm 基线切换到 pre+post norm,这在小规模上表现中性,但从 4B 开始显示出改进。阶梯 v1.3 实现了sigmoid 门控注意力。阶梯 v1.4(2-stream mHC)在较小规模上表现良好,但在 4B 规模上没有改进,因此被否决。阶梯 v1.5 切换到全局 NoPE,并加入最终学习率冷却至零,阶梯 v1.6 加入了值残差。在最终版本中,我们切换到自定义预训练数据集,大幅上采样了代码。

综合来看,这些累积的改动将代码上的计算效率(以 dolma_stack CE 衡量)在 4B 模型规模上最多提升了 16 倍。换言之,我们最终的 Ripple 模型阶梯可以用少 16 倍的计算量达到与 Olmo 3 阶梯相似的代码性能。请注意,Olmo 3 阶梯基线的斜率比阶梯 v1.0+ 更平缓,这表明计算效率的提升将随规模增加而增大。

结论

我们构建 Ripple-65B Preview,是为了开始闭合模型与驱动它们的硬件之间的循环。该模型的一个内部变体驱动着我们的 AI 芯片设计平台,我们期待随着我们向前沿扩展,会涌现出怎样的能力。

我们有幸在现有开源贡献的基础上进行构建,包括来自 Allen Institute for AI(AI2)的 Olmo 团队、Prime Intellect、RadixArk 和 NVIDIA 的贡献。强大的开源生态系统也是自我改进的:贡献者增强了生态系统,而生态系统反过来又吸引了更多贡献者。

在 Ricursive,我们正在有选择地招聘杰出的研究人员和工程师,以推动从模型到芯片的全栈 AI 前沿。如果你感兴趣,请查看 https://www.ricursive.com/careers。

¹ 据报道,像 Reflection 这样的美国开放权重参与者承诺每月为 SpaceX Colossus 的计算支付 1.5 亿美元,用于训练他们的模型。Reflection 报告称,Beam 的预训练耗时 4 周,后训练耗时 4 周。