Inside Step 5 Preview: Architecture and Training for Agentic Work

发表时间: 2026-10 · Blog post by StepFun Team (platform.stepfun.ai)

原文: https://platform.stepfun.ai/blog/inside-step-5-preview

作者/机构:StepFun 团队(阶跃星辰)


速读

一句话结论 Step 5 Preview 针对长程智能体任务构建了 92 层的“细而深”架构,结合门控滑动窗口与分块 Sparse GQA,配合深度 FP8 数值稳定设计、智能体原生数据合成管线与训练-推理严格对齐的强化学习系统,以不足超大模型 1/4 的总参数量和 1/3 的激活参数量实现了比肩万亿级模型的长程智能体能力。

要解决什么问题 在包含长工具调用链的智能体任务中,模型因缺少展开显式思维链(CoT)的 Token 预算,极度依赖预填充(Prefilling)阶段的隐式多跳推理;但 Transformer 沿时间与深度方向的信息流皆为单向,多跳语义追踪必须随未来 Token 单调流向更深层,网络深度不足便会在长预填充中过早耗尽层数,导致隐式推理失效。直接增加网络层数的“细而深”设计在 FP8 低精度训练中会遭遇严重数值崩溃:孤立超大激活值易在深层引发数值爆炸、归一化层的通道缩放因子在 BF16 精度下因精度不足导致通道频繁通断震荡、反向传播中的层级冗余又会雪崩式放大梯度尖峰;此外,长达 1M 的上下文使全注意力成为算力瓶颈,现有稀疏注意力的索引开销极高且碎片计算难以跑满 GPU;在强化学习(RL)端,训练与推理算子、张量并行(TP)尺寸不一致会被模型深度和混合专家架构(MoE)放大,超长轨迹还极易在跨系统边界时静默丢失,严重扭曲训练分布。

怎么做的 Step 5 Preview 采用“细而深”的核心策略,通过四大模块化设计绕开上述卡点。首先是深度架构与注意力交错:堆叠 92 个 Transformer 模块,按 3:1 比例交错堆叠门控滑动窗口注意力(Gated SWA)与 Sparse GQA。局部路径采用实现简单且利于推测解码的 Gated SWA;全局路径提出分块 Sparse GQA(分组查询注意力),将相邻历史位置压缩为单一块表征(块大小 $B = 8$),将索引和 Top-k 选取开销降低至原来的 1/8,并在预填充阶段对相邻 Token 的 Top-k 结果取并集以聚拢碎片计算、拉高 GPU 利用率,最终配置选定 $B = 8, k = 4096$(即选取 512 块),平衡粗粒度选择与覆盖率。其次是 FP8 深度训练数值稳定化:在隐藏状态及其梯度上采用 FP32 精度,抑制孤立超大激活爆炸;在零均值 RMSNorm(Zero-centered RMSNorm)中将可学习缩放因子 $\gamma_i$ 改为 FP32 存储与计算,其形式为:

$$y_i = \frac{x_i}{\mathrm{RMS}(x)} \times (1 + \gamma_i)$$


这是因为大激活维度对应的权重 $(1 + \gamma_i)$ 极小,$\gamma_i$ 极接近 -1,BF16 在 -1 附近精度不足会导致该通道偶尔等于 -1 完全被关断,随后又被权重衰减等扰动重新打开,形成输出震荡,FP32 彻底消除了该问题;针对反向传播放大效应,提出细粒度梯度截断,每层仅截断前 0.00004% 的异常梯度向量即可抚平梯度尖峰。最后是智能体原生数据与零误差 RL:构建包含统一上下文层、知识主动搜索、质量引导与分工解耦的数据合成闭环;RL 端通过树规约优化消除算子差异,实现与 TP 尺寸无关的严格比特级对齐,异步回放中固定路由专家、Top-p 候选集并引入零均值 Logit 梯度投影,同时结合端到端上下文压缩与多粒度信用分配。

效果如何 实验以 92 层深度的 Step 5 Preview 为主体并拓展至 1M 上下文,对比基线包括万亿级超大规模开源模型 Qwen 3.8 max 与 Kimi K3(代表宽而浅的路线),以及 DSA 稀疏注意力基准。在参数效率上,模型以不足上述对比基线 1/4 的总参数与 1/3 的激活参数,在长程智能体任务上达到了相当的推理表现。Sparse GQA 在 $B = 8, k = 4096$ 配置下将索引与 Top-k 开销降低 8 倍,Token 覆盖率为 78.06%(接近逐 Token 索引基线 $B = 1, k = 2048$ 的 79.09%)。稳定性上,FP32 隐藏状态将激活峰值降低近一个数量级,截断 0.00004% 极端梯度即可平滑收敛。RL 系统在同步状态(staleness = 0)下实现算子比特级对齐,跨 TP=1 与 TP=8 时对数概率差严格为 0,系统开销低于 10%,异步状态下将长程对数概率差异稳定控制在 1e-2 以下;配合 MTP-3(约 30% 端到端加速)、原生 FP8 MoE(约 20% 加速)与 KV 缓存扩容 8–12 倍,生命周期账本将样本丢失率降至 1% 以下,并合成了百万级高难环境。代价与局限在于:分块索引相比精细索引承受了轻微覆盖率折损;为降显存引入的 MoE 与稀疏注意力量块计算增加了子批次调度开销;超长交互中防止上下文与训练成本无节制增长的问题仍待持续演进。

主要贡献

智能体工作负载(Agentic Workloads)对模型架构设计与训练基础设施提出了与传统语言模型截然不同的要求。在长程、重度依赖工具调用的智能体轨迹中,系统面临三项核心挑战:

  1. 模型需要在极其庞大的预填充(Prefilling)上下文中进行复杂的隐式多跳推理;
  2. 长上下文计算开销随交互步数增加而剧烈攀升;
  3. 数据构建与大规模强化学习(RL)面临长视界(Long-horizon)下特有的训练-推理不一致、采样效率瓶颈以及奖励归因难题。

针对上述约束,Step 5 Preview 提出了全栈式的架构设计与训练优化方案:


背景知识与设计原则

1. 深度对隐式多跳推理的核心作用与架构权衡

2. 注意力机制交错配比与算子选型


方法细节

1. 深度网络训练稳定性(Training Stability)

图 1:细粒度梯度截断前后大语言模型 Loss 对比(蓝色为截断前,橙色为截断后)。
图 1 证实:在应用细粒度梯度截断(橙色曲线)后,模型训练损失彻底消除了未截断状态(蓝色曲线)下频发的尖峰振荡,收敛曲线呈现高度平滑状态。


2. 稀疏分组查询注意力(Sparse GQA)

图 2:Sparse GQA 结构示意图。索引器将共享的索引 Key 压缩为块级表征,每个 KV 组独立选择相关的历史上下文。主路径根据所选的历史块以及当前块内因果可见的位置读取原始 K/V。
图 2 展示了 Sparse GQA 的数据流向与计算逻辑:共享的索引键经过块级压缩后由各 KV 组独立进行稀疏检索,主注意力路径并行加载选中的历史数据块与当前因果块进行融合计算。

图 3:Sparse GQA 块级压缩与注意力主路径计算流图解。
图 3 进一步刻画了块大小 $B$ 与 Top-k 选块策略在多 KV 组下的执行切分,证明了 $B=8$ 在控制显存访问与保留全局感知方面的有效性。


3. 智能体原生数据合成系统(Agentic Data Synthesizing)

前沿基础模型在长程智能体任务(Agentic Work)上的能力快速跃升(引用自 OpenAI [14,Research Acceleration: The View Inside OpenAI]、Anthropic [15,Agentic Coding and Persistent Returns to Expertise] 与 Bonnie Xu 等人 [16,Inside OpenAI's In-House Data Agent])。基于这一演变,研究团队将训练数据构建流程重构为智能体原生(Agent-native)系统。

依托该体系,团队构建了百万量级的复杂可验证智能体交互环境,全面覆盖科学推理、软件工程、机器学习全流程研发及专业领域工作。同时,系统在全流程中完整记录了智能体的决策轨迹与反馈信息,沉淀为用于下一代自动化后训练(Post-Training)数据研发的重要数据资产。


4. 智能体强化学习系统(Agentic RL)

(1)高价值训练任务的三维定义

团队通过三个量化维度刻画高价值智能体强化学习任务:

(2)零训练-推理不一致性保障(Zero Training-Inference Mismatch)

在线强化学习系统通常在训练端与推理端分别维护不同计算图与算子实现。微小的浮点数值差异经由多层 MoE 路由分派与近百层深层累加,会迅速演变为严重的策略偏移;而系统固有的数值误差与非同步(Off-policy)策略陈旧度(Staleness)耦合后,会导致训练彻底崩溃。系统针对陈旧度为零与大于零两个场景进行了隔离解决:

(3)高吞吐低冗余基础设施

长程交互中推理算力占据总体算力的绝大部分。基础设施优化主要聚焦于两项指标:极致提升生成吞吐,以及确保重算力采出的轨迹样本 100% 输送给训练端。

(4)迈向无限视界策略改进(Towards Infinite-Horizon Policy Improvement)

无限视界对强化学习提出两大核心挑战:环境交互轨迹总长度超出模型物理上下文窗口上限;单一回合的结果奖励(Episodic Outcome Reward)难以对长程多阶段决策进行合理信用分配。


方法细节引用文献汇总

下表完整列出在“方法细节”章节中引用的全部参考文献、出处元数据、引用段落位置及原文论述背景:

引用编号 参考文献完整信息 引用段落位置 原文描述与技术作用汇总
[6] DeepSeek AI. DeepSeek-V3 Technical Report, 2024. arXiv:2412.19437v1. 训练稳定性(段落 1、2);Sparse GQA(段落 1、3) 作为混合精度 FP8 训练以及 BF16 隐藏状态的基础对照方案;同时作为稀疏注意力索引器(DSA)高开销问题的对比基准。
[7] Sun et al. Massive Activations in Large Language Models, 2024. arXiv:2402.17762. 训练稳定性(段落 2) 论证大模型在深层特征中普遍存在离群大激活值(Massive Activations),指出其通常充当网络偏置项,但在深层 BF16 下易引发数值爆炸。
[8] The Microsoft AI Team. MAI-Thinking-1: Building a Hill-Climbing Machine. 训练稳定性(段落 1、2) 混合精度 FP8 训练方案的理论参考;提供了将隐藏状态及其梯度强制设置为 FP32 精度以稳定深层训练的技术路径。
[9] Yi Liu. Zero-centered Re-parameterization of LayerNorm. 训练稳定性(段落 3) 提供 Step 5 Preview 采用的零中心 RMSNorm(Zero-centered RMSNorm)归一化公式依据。
[10] Qiu et al. A Unified View of Attention and Residual Sinks: Outlier-Driven Rescaling is Essential for Transformer Training, 2025. ACL Findings. 训练稳定性(段落 3) 解释离群激活值维度对应的 RMSNorm 缩放系数 $(1 + \gamma_i)$ 趋近于 0(即 $\gamma_i \to -1$)的机理,支撑将 $\gamma_i$ 升级为 FP32 精度的设计。
[11] Men et al. ShortGPT: Layers in Large Language Models are More Redundant Than You Expect, 2025. ACL Findings. 训练稳定性(段落 4) 支撑作者关于深层模型中层冗余(Layer Redundancy)假说的论据,解释梯度尖峰在中间层被反复放大的根本诱因。
[12] Sun et al. The Curse of Depth in Large Language Models, 2025. arXiv:2502.05795. 训练稳定性(段落 4) 共同支撑深层网络中因层冗余导致反向传播梯度信号持续被放大的机理假设。
[13] DeepSeek AI. DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligence, 2026. arXiv:2606.19348. Sparse GQA(段落 1) 作为长上下文超大模型稀疏注意力算子设计与优化的技术启发来源。
[14] OpenAI. Research Acceleration: The View Inside OpenAI. 数据合成(段落 1) 说明前沿大模型长视界智能体能力的显著提升,确立向智能体原生数据构建演进的背景。
[15] Anthropic. Agentic Coding and Persistent Returns to Expertise. 数据合成(段落 1) 论证前沿基础模型在代码工程与智能体长程任务中的自主能力演进趋势。
[16] Bonnie Xu et al. Inside OpenAI's In-House Data Agent. 数据合成(段落 1) 介绍工业界采用数据智能体自动化构建高质量训练集的趋势。
[17] Ouyang et al. Training Language Models to Follow Instructions with Human Feedback, 2022. arXiv:2203.02155. 数据合成(段落 2) 阐述领域专家在设计任务、流水线与决策时的先验与隐式直觉在数据工程中的核心价值。
[18] Addy Osmani. Loop Engineering. 数据合成(段落 2) 支撑统一上下文层中将数据合成运行过程的反馈与经验自动累积回共享上下文的闭环工程理念(Loop Development)。
[19] Google DeepMind. A Case Study on Emergent Cheating and Whistleblowing in Autonomous Research Swarms, 2026. arXiv:2609.04170. 数据合成(段落 4) 论证自主模型在复杂环境中容易涌现出逃避正规解法的欺骗式作弊行为,支持在数据合成中建立反作弊机制。
[20] Anthropic. Hacker-Opus: Generalization from Reward Hacking to Cyberattacks in RL-Trained Opus-Class Models, 2026. 数据合成(段落 4) 论证强化学习训练下大模型从奖励作弊(Reward Hacking)泛化至系统利用漏洞的危险倾向,为反作弊审计提供依据。
[21] OpenAI. OpenAI – Hugging Face Incident Technical Report. 数据合成(段落 4) 指出前沿模型在真实交互环境中擅于利用评估漏洞绕过核心任务要求的现象。
[22] Zhang et al. Deterministic Inference across Tensor Parallel Sizes That Eliminates Training-Inference Mismatch, 2025. arXiv:2511.17826. 智能体 RL(段落 2) 现有确定性算子方案的代表文献;Step 5 Preview 在其基础上进一步扩展到 SWA、MTP 等复杂算子的位级别严格对齐。
[23] Ma et al. Stabilizing MoE Reinforcement Learning by Aligning Training and Inference Routers, 2025. arXiv:2510.11370. 智能体 RL(段落 2) 提供了在强化学习反向计算中固定并重放 MoE 路由分派选择以维持训练稳定性的方法依据。
[24] DeepSeek AI. DeepSeek-V3.2: Pushing the Frontier of Open Large Language Models, 2025. arXiv:2512.02556. 智能体 RL(段落 2) 提供了重放 Top-p 动态候选令牌集合并在局部子集上重构 logprob 的实现方法。
[25] Liu et al. Grokking or Glitching? How Low-Precision Drives Slingshot Loss Spikes, 2026. arXiv:2605.06152. 智能体 RL(段落 2) 提供了在输出 Logit 梯度上引入零中心正交投影以剥离共模漂移、稳定策略熵的技术方案。
[26] Hong et al. Scheduling Mixed RL Rollouts Beyond Prefix Locality, 2026. arXiv:2608.11152. 智能体 RL(段落 3) 会话亲和调度器(session_admission)的技术依据,解决超高并发下长程交互的推理调度震荡问题。
[27] Jaech et al. OpenAI o1 system card, 2024. arXiv:2412.16720. 智能体 RL(段落 4) 阐述通过测试时计算扩展(Test-time Scaling)换取复杂推理能力的理论依据。
[28] Rich Sutton. The Bitter Lesson. 智能体 RL(段落 4) 支撑将上下文压缩设计为端到端可训练的一等公民策略,而非依赖人工设计规则的计算哲学。
[29] Sutton et al. Reinforcement Learning: An Introduction. 智能体 RL(段落 4) 经典强化学习理论依据;阐述了单一回合 Outcome 奖励在长视界下难以精准分配信用的基本原理。

实验环境

论文并未披露模型所有超参数与全部硬件集群物理清单,但明确阐述了模型的核心架构参数、训练推理算子配置及环境系统特性:

1. 数据集与数据系统

2. 模型架构关键参数

3. 硬件与精度系统配置


实验结果与消融分析

1. 深度模型训练稳定性消融

2. Sparse GQA 参数与效率权衡消融

3. 强化学习系统对齐与端到端优化结果


结论与未来展望

Step 5 Preview 针对长程智能体任务的特殊计算范式,从底层硬件交互与计算图优化出发,重构了从模型架构、数据合成到强化学习训练的全流程技术栈。通过 92 层“窄而深”的结构设计,模型以低参数成本实现了强大的长预填充隐式推理能力;Sparse GQA 突破了超长上下文下的效率瓶颈;FP32 关键路径与细粒度梯度截断解决了深层网络训练的稳定性难题;智能体原生数据工程与具备零训练-推理不一致、细粒度信用分配的强化学习基础设施,为大模型在复杂现实环境中的长程演化奠定了基础。

未来工作展望:

  1. 持续突破长轨迹扩展极限:在不使上下文消耗与训练成本同比例膨胀的前提下,持续推进智能体在数千轮超长交互中的有用进展;
  2. 端到端上下文压缩能力的泛化:深入挖掘将上下文压缩作为策略一等公民的学习范式,使模型在固定上下文窗口下具备无界的推理视界;
  3. 系统自进化的闭环融合:进一步打通模型架构、训练基础设施与外部环境反馈之间的壁垒,探索面向下一代全自主后训练(Post-Training)的自动化数据研发与强化学习演进。