← 返回 2026-09-15

ZGCM-1:面向数学与智能体搜索的全开源高效基础模型 ZGCM-1: A Fully Open and Extremely Efficient Foundation Model for Math and Agentic Search

Jiyan He, Guang Liang, Hao Liu, Haoxiang Guan, Jinbo Sun, Junyi Guo, Wenjun Feng, Yantai Xie, Yifei Shen, Bin Shao, Chuyang Wei, Kai Chen, Kexin Zhou, Minghang Zhu, Shuxin Zheng, Tie-Yan Liu, Taine Zhao, Wenhui Zhu, Xueyin Xu, Xiaoqing Zhang, Yatao Li, Yuxuan Ren 📅 2026-09-11 👍 103 2026-11-02 18:30
AI原生研发 FP8训练 开源基础模型 数学推理 智能体搜索 混合注意力 长上下文 高效训练

7B全开源模型靠深度思考加工具调用双引擎与极致训练效率,在数学和智能体搜索上比肩大型前沿模型。

前置知识

滑动窗口注意力(SWA)与混合注意力

让每个 token 只与相邻 128 个 token 计算注意力,使 KV 缓存保持恒定大小;再每隔若干层插入一个全局注意力层保证长程信息流动。ZGCM-1 采用 5 层局部对 1 层全局的交错比例(27 层门控 SWA + 5 层全局),全局层位于第 6/12/18/24/30 层。

论文的架构章节与全部效率结论(3.94 倍吞吐、6.4 倍 KV 缩减)都建立在 SWA 与全局比例的取舍之上,不懂 SWA 无法理解架构实验。

FP8 混合精度训练与 TWEO

FP8 用 8 位浮点执行矩阵乘法(前向 E4M3、反向 E5M2),按 1024 步历史最大值更新缩放因子;TWEO 是抑制中间激活异常值的正则化手段,防止 FP8 训练数值发散。二者结合使预训练稳定维持在约 585 TFLOP/s/GPU(约 60% BF16 等效 MFU)。

4.2 倍预训练加速中 FP8 贡献约 1.5 倍,而其稳定性机制(Finding 2)是论文系统-算法协同设计的关键论据。

Muon 优化器

一种对矩阵参数做谱缩放更新的优化器:动量 0.9、谱范数缩放、五步 Newton-Schulz 迭代,标量参数仍用 Adam;学习率 2e-4、权重衰减 0.1。论文实测其对 AdamW 有约 1.8 倍的步数-损失效率,SFT 阶段采用逐层 Muon 变体。

Muon 是 4.2 倍加速中最大的单项因子(1.8 倍),也是复现训练稳定性与超参配置的核心。

MDP 中间训练(MDP Mid-Training)

把智能体交互轨迹重构为马尔可夫决策过程式样本:任务描述、交互历史与当前环境反馈构成状态 $s_t$,模型在原始 next-token 目标下预测合适的下一动作 $a_t$。与复现完整轨迹相比,它提供状态条件下逐步决策的密集监督,且仍在全序列因果语言建模目标内完成。

这是本文中期训练最核心的方法创新,是理解 ZGCM-1 智能体能力为何能在预训练阶段就形成的钥匙。

GRPO(组相对策略优化)

一种免价值网络的强化学习算法:每个问题采样一组回答,用组内奖励的相对差异构造优势函数;配合动态采样过滤零方差组、参考策略 KL 正则控制漂移、温和长度惩罚抑制超长截断噪声。本文 rollout 允许 65,536 个新生成 token、总上下文 98,304。

后训练阶段数学与代码的增益全部来自这一算法,其采样规模(384 或 3072 条轨迹/步)决定了长推理训练的可行性。

think/no-think 混合微调

同一套权重同时学习两种模式:带显式推理链的 think 样本与直接给出答案的 no-think 样本,由系统提示或运行时预算切换。论文发现 think 监督正向迁移到 no-think:AIME 2025 从 10.00% 升至 43.33%,但 HumanEval+ 反降 7.32。

论文评测分 think 与 no-think 两种模式报告(Table 2 与 Table 16),不理解该机制就无法解读结果表。

研究动机

前沿级推理与深度智能体搜索长期被视为千亿参数系统的专属领地。以 Qwen3-235B-A22B、Kimi-K2、GLM-5.1 为代表的前沿模型依赖海量算力,训练动辄上万卡时,把算力受限的高校与个人研究者挡在门外——这构成第一重'规模壁垒'。第二重是'透明度壁垒':多数有竞争力的模型只开放权重而非完全开源,上游数据过滤配方、中期训练课程、长上下文扩展日程、多轮智能体轨迹等关键环节仍是黑箱。例如 DeepSeek-R1-0528-Qwen3-8B 这类蒸馏模型虽在 AIME 2024 达 83.33%,但其教师模型与数据管线均不公开,研究者既不能复现也难以做受控实验,无法系统研究训练动力学与紧凑模型的容量极限,导致开源生态在推理与智能体方向持续落后于闭源系统。

本文的目标是本文要在完全透明的开放科学范式下,从零训练一个 7.39B 的稠密基础模型 ZGCM-1,并证明其核心论题:紧凑模型虽受静态参数容量约束,但可以通过'深思熟虑的内部推理 + 主动的外部工具调用'双引擎突破这一限制——不靠被动记忆开放网络,而是在 256K 上下文中自主检索网络证据、操作终端、分析去符号化的二进制程序。具体目标有三:(1) 构建端到端高效开源训练管线,在学术级算力下把 16K 预训练时间-损失相对 BF16/AdamW 基线加速约 4.2 倍;(2) 在数学推理与智能体搜索上与数量级更大的前沿模型正面竞争;(3) 公开预训练、中期训练、后训练全部阶段权重、中间检查点、训练代码、分阶段数据配方、W&B 日志与评测套件。

与已有工作不同的是,本文的独特切入是三条线同时展开。其一,把'知识外置'作为设计公理:7B 参数不用于记忆网页内容,而用于承载慢思考与主动求索能力,知识本身留在模型之外按需获取,这直接决定了架构(256K 混合注意力)、中期训练(智能体轨迹进入预训练语料)与后训练(深度研究/软件工程/终端三支数据)的取舍。其二,把交互轨迹重构为 MDP 状态-动作转移,在全序列因果语言模型目标内部注入逐步决策监督,而不是像常见做法那样把智能体数据推迟到 SFT/RL 阶段。其三,首次在基础模型技术报告中对'AI 原生研发'给出量化自评:九名核心成员按 L1-L5 五级量表对 11 类任务打分(共 99 份评分),坦承架构与学习算法设计仅达 L2、从未超过 L3,而实验监控与部署工程已达 L4;同时以 8 条经验发现公开全部教训而非只报喜。

核心方法

ZGCM-1 是一条四段式全栈管线。第一段架构与系统协同设计:32 层、7.39B 参数的 decoder-only Transformer(隐藏维 4096、SwiGLU 中间维 11008、GQA 32 查询头/8 KV 头、部分 RoPE 比例 0.33),以 5:1 比例交错 128-token 门控滑动窗口注意力(27 层)与全局注意力(5 层);矩阵参数用 Muon、标量用 Adam,矩阵乘用混合 FP8 加 TWEO 异常值抑制,在 H100 上稳定跑到约 585 TFLOP/s/GPU(约 60% BF16 等效 MFU)。第二段通用预训练:0.99T token 的课程式 Stage 1 加 3.20T 的 Stage 2。第三段中期训练:600.51B token 按 16K、64K、256K 三阶段渐进扩窗,混入重构为 MDP 的智能体轨迹。第四段后训练:492 万条 SFT 数据(激进质量剪枝约 50%)加 GRPO 混合强化学习。整条管线使 16K 预训练时间-损失加速约 4.2 倍,并通过 AI 原生研发流程让智能体参与数据处理、实验、评估与部署全周期。

核心创新是把'小模型如何变强'的答案从堆参数转向双引擎机制。作者的前提是:7B 模型无法被动记住开放网络,但参数可以承载两件事——慢思考(长链推理)与主动求索(工具使用)。这带来两个本质区别。其一,中期训练把交互轨迹改写为 MDP 单步决策样本:任务描述加交互历史加环境反馈构成状态 $s_t$,模型在全序列因果目标下预测下一动作 $a_t$,使逐步决策监督直接进入预训练,而非像既有工作只在后训练稀疏地学智能体行为。其二,实验证明长上下文智能体能力应在中间训练阶段打下根基:短 SFT 探针显示中期训练把 MATH-500 从 35.83% 提到 74.12%、IFEval Strict 从 29.57% 提到 72.64%,且后训练只需中等长度 SFT 数据即可激活 256K 行为(Finding 7),绕开了收集 256K 长推理轨迹的高昂成本。配套的 5:1 门控 SWA、FP8+Muon+TWEO 让这一切在学术算力下可行。

方法步骤详情

流程分五步。第一步配方搜索:用 0.3B 代理模型跑约 30B token,以损失与知识/代码/数学评测信号迭代调整混合比。第二步通用预训练:Stage 1(0.99T)按词汇复杂度对通用文本由易到难排序,代码与数学独立交错(编码 BPB 从 1.99 降至 0.81);Stage 2(3.20T)提高代码与数学占比并引入专门推理数据。第三步中期训练:从 2.86T 去重池采样 600.51B,依次在 16K(180B)、64K(240B)、256K(180.51B)上下文训练,智能体数据占比从 1.5% 增至 3.3%,交互轨迹重构为 MDP 状态-动作对,每步约 12.6M token;分段路线最终损失 1.16 优于直上 256K 的 1.19。第四步 SFT:492 万条样本经分层质量过滤,剪掉约 50% 反而更优(六基准均值 67.78 升至 68.83),think/no-think 混合、assistant-only 损失、通用与智能体数据联合交错。第五步 GRPO:数学用答案对错二值奖励、代码用通过测试比例,配合 KL 正则与温和长度惩罚。

技术新颖性

技术新颖性体现在四点。(1) 注意力配置由受控实验选出:同预算 10B token、8 张 H100 下横评全注意力、Flash-GQA、MLA 与 1:1/3:1/5:1 三种 SWA——SWA 5:1 吞吐最高(9,566 token/s/GPU)且尾损失 1.93 与全注意力持平,加速比随上下文从 4K 的 1.13 倍增至 256K 的 3.94 倍;KV 缓存 5.0 GiB 优于线性注意力 GDN 3:1 的 7.0 GiB 与全注意力的 32.0 GiB。(2) 效率增益按可分离因子显式分解:SWA 1.4 倍、FP8 1.5 倍、Muon 1.8 倍、Pre-LN 1.1 倍,相乘约 4.2 倍,每个因子可独立检验。(3) MDP 中期训练在不放弃全序列因果目标的前提下叠加步级监督,兼顾全轨迹上下文与局部决策密度。(4) 公开 8 条带数据支撑的经验发现——课程排序对代码/数学失效、SFT 质量胜过数量、长 CoT 过量损害指令遵循、think 训练正向迁移 no-think 等——这种连负结果与 trade-off 都完整披露的写法在技术报告中少见。

Hybrid attention architecture of ZGCM-1.
Figure 3: Hybrid attention architecture of ZGCM-1.
Architecture experiments: (a) quality-throughput trade-off; (b) throughput speedup vs context length; (c) KV cache memory footprint.
Figure 4: Architecture experiments: (a) quality-throughput trade-off; (b) throughput speedup vs context length; (c) KV cache memory footprint.
General Pre-Training is organized into two data stages, followed by three Mid-Training stages with progressively longer context lengths.
Figure 5: General Pre-Training is organized into two data stages, followed by three Mid-Training stages with progressively longer context lengths.
Data mixture composition shown as donut charts for the two General Pre-Training stages and the three Mid-Training context stages.
Figure 6: Data mixture composition shown as donut charts for the two General Pre-Training stages and the three Mid-Training context stages.
Mid-Training data processing pipeline.
Figure 7: Mid-Training data processing pipeline.
AI-driven self-iterating data governance pipeline.
Figure 8: AI-driven self-iterating data governance pipeline.
Token-level cross-entropy (CE) loss and learning-rate schedule for the staged Mid-Training run.
Figure 9: Token-level cross-entropy (CE) loss and learning-rate schedule for the staged Mid-Training run.
Supervised fine-tuning data curation pipeline.
Figure 10: Supervised fine-tuning data curation pipeline.
Training dynamics of the two length-specific SFT variants (64K and 256K).
Figure 11: Training dynamics of the two length-specific SFT variants (64K and 256K).
AI-native R&D workflow: each researcher directs agents across seven core stages of model development.
Figure 13: AI-native R&D workflow: each researcher directs agents across seven core stages of model development.
Capability trajectories across General Pre-Training and Mid-Training milestones after rapid fine-tuning.
Figure 16: Capability trajectories across General Pre-Training and Mid-Training milestones after rapid fine-tuning.

实验结果

核心结论:ZGCM-1-7B 在 7B-8B 档 14 项推理基准平均第一。数学上 AIME 2026 得 75.00%(同档次优 71.67%)、MATH-500 97.13%、HMMT 2025 70.42%(次优仅 61.50%);AIME 2024/2025 与最强 R1 蒸馏 8B 互有胜负。HumanEval+ 90.24% 最佳,但知识任务落后:MMLU 73.88%(Olmo-3 为 85.40%)、GPQA 47.87%(R1 蒸馏 60.10%)。智能体侧比肩数十倍大的模型:WebWalkerQA 63.09% 超 Kimi-K2(63.00%)与 Claude 4 Sonnet(61.70%);BrowseComp 19.43% 远超 Kimi-K2 的 14.10%。Binary Function Search 50 任务命中 62%(31/50),是 Qwen3-8B(12%)的 5 倍多、逼近 GLM-5.1 的 66%。效率上 16K 预训练时间-损失加速约 4.2 倍;ACE 最强为数学 98.18%、最弱为指令遵循 49.74%。

Composition of our supervised fine-tuning corpus.
Table 1: Composition of our supervised fine-tuning corpus.
Thinking-mode benchmark results (%) for ZGCM-1-7B and comparable-scale models.
Table 2: Thinking-mode benchmark results (%) for ZGCM-1-7B and comparable-scale models.
Web-environment deep-research results (%).
Table 3: Web-environment deep-research results (%).
Binary Function Search results on 50 tasks.
Table 4: Binary Function Search results on 50 tasks.
Levels of AI autonomy in R&D workflows (L1-L5).
Table 5: Levels of AI autonomy in R&D workflows (L1-L5).
Distributed-training configurations on H100 GPUs.
Table 6: Distributed-training configurations on H100 GPUs.
Five-checkpoint window averages during General Pre-Training.
Table 7: Five-checkpoint window averages during General Pre-Training.
Bits per byte (BPB) after 10B training tokens (curriculum vs random).
Table 8: Bits per byte (BPB) after 10B training tokens (curriculum vs random).
Mid-training context mixtures.
Table 9: Mid-training context mixtures.
Ablation of SFT data filtering.
Table 10: Ablation of SFT data filtering.
Tool schema for Deep Research trajectories.
Table 11: Tool schema for Deep Research trajectories.
No-think performance before and after mixed think/no-think SFT.
Table 12: No-think performance before and after mixed think/no-think SFT.
Evaluated ACE taxonomy, category scope, and probe allocation.
Table 13: Evaluated ACE taxonomy, category scope, and probe allocation.
Selected no-think benchmark results (%) for ZGCM-1-7B and selected 7B-8B models.
Table 16: Selected no-think benchmark results (%) for ZGCM-1-7B and selected 7B-8B models.
Per-benchmark ranks for ZGCM-1-7B and models at comparable scale across 14 reasoning benchmarks.
Figure 1: Per-benchmark ranks for ZGCM-1-7B and models at comparable scale across 14 reasoning benchmarks.
Binary Function Search workflow and Ghidra command interface.
Figure 12: Binary Function Search workflow and Ghidra command interface.
AI4AI autonomy across R&D tasks, based on core contributor assessments.
Figure 14: AI4AI autonomy across R&D tasks, based on core contributor assessments.
查看结构化数据
任务指标本文基线提升
AIME 2026 竞赛数学 pass@1(%,32 次运行均值) 75.00 Qwen3-8B 71.67;DeepSeek-R1-0528-Qwen3-8B 69.17 +3.33 / +5.83 个百分点
HMMT 2025 竞赛数学 pass@1(%) 70.42 DeepSeek-R1-0528-Qwen3-8B 61.50 +8.92 个百分点
MATH-500 pass@1(%) 97.13 MiniCPM4.1-8B 96.32 +0.81 个百分点
WebWalkerQA 开放网页深度研究 准确率(%) 63.09 Kimi-K2 63.00;Claude 4 Sonnet 61.70;Qwen3-235B-A22B 59.60 +0.09 / +1.39 / +3.49 个百分点
BrowseComp 深度网页搜索 准确率(%) 19.43 Kimi-K2 14.10;Claude 4 Sonnet 12.20 +5.33 / +7.23 个百分点
Binary Function Search 二进制函数定位 精确地址命中率(%,50 题) 62.00 Qwen3-8B 12.00;GPT-4o 18.00;GLM-5.1 66.00 对同规模 +50.00 个百分点;对 GLM-5.1 落后 4.00 个百分点
16K 预训练时间-损失 相对加速比 约 4.2 倍 OLMo 3 风格 7B BF16/AdamW 基线 SWA 1.4 倍 × FP8 1.5 倍 × Muon 1.8 倍 × Pre-LN 1.1 倍
256K 上下文推理 KV 缓存 显存(GiB,batch=1,bf16) 5.0 全注意力 32.0;GDN 3:1 线性注意力 7.0 6.4 倍缩减
SFT 数据质量剪枝消融 六基准平均分 68.83(仅保留 1.145M 高质量样本) 最简处理全量 2.08M 样本得 67.78 样本少 44.9% 而均值 +1.05

局限与改进

作者承认四条局限:(1) 参数知识边界——纯闭卷、依赖记忆的任务天然落后于巨型模型,MMLU 73.88 对 Olmo-3 的 85.40、GPQA 47.87 对 R1 蒸馏的 60.10;(2) 推理冗长与指令遵循互相牵制,重推理监督诱发冗长偏置,ACE 指令类别仅 49.74%;(3) 通用软件与终端智能体尚幼稚——SWE-bench Verified Mini50 仅解 2/50(4.0%),Terminal-Bench 2.0 仅 2/89(2.25%);(4) 环境与协议脆弱,严重依赖 schema 对齐与稳定反馈。我的补充观察:miniF2F 仅 2.87%、ARC-AGI-1 仅 3.50%,形式化证明与抽象推理几乎未解决;BFS 评测仅 50 题,统计功效有限;Web 结果依赖搜索 API 与裁判模型(Qwen3-30B-A3B),跨论文可比性存疑;RL 章节只给配置而未报告消融增益数字;AI4AI 的 99 份评分是主观序数判断;no-think 模式下 HumanEval+ 因混合训练反降 7.32 个点。

独立分析的弱点

独立分析出的弱点与改进方向:(1) 知识型任务短板——7.39B 参数上限决定 MMLU/GPQA 差距,可加入教师蒸馏或把检索前移进预训练;(2) 形式化定理证明近乎空白(miniF2F 2.87%),应接入 Lean/Isabelle 执行器反馈做专用 RL;(3) 仓库级软件工程(4.0%)与终端任务(2.25%)暴露交互式 RL 的缺失——目前 RL 增益集中在数学/代码单域结果奖励,需在可执行沙箱内做多轮端到端 RL;(4) think 到 no-think 迁移有代价(HumanEval+ -7.32、MMLU -2.60),混合配比需 ACE 式诊断信号自动闭环校准;(5) 4.2 倍加速是四因子相乘的估计而非单一受控 A/B,Pre-LN 的 1.1 倍仅来自探索性 7B 证据;(6) FP8 稳定性依赖 TWEO,缺少去掉 TWEO 的失败模式分析;(7) BFS 评测仅 50 题且未报方差,应扩到数百题并给置信区间;(8) Web 深度研究结果强依赖第三方搜索 API 时效与行为,需报告重跑稳定性。

未来方向

作者提出四个方向:(1) 把混合 SWA、Muon 优化与 MDP 中期训练配方推广到稀疏 MoE 骨干,在保持低推理 FLOPs 的同时扩大参数容量与领域专精;(2) 从 token 级 SFT 与结果奖励转向真实执行沙箱(终端、网页、编译器)内的多轮交互式端到端 RL,直指其最弱的 SWE/Terminal 能力;(3) 统一预训练表征与在线自主检索,使模型在检测到参数不确定性时动态触发搜索子程序,正面回应闭卷知识边界;(4) 把 AI4AI 智能体体系从集群运维、数据治理、原子评测扩展到自主假设提出、自动内核优化与闭环合成环境设计。基于其成果还可延伸:用 ACE 的 183 项原子能力做自动课程生成,让 SFT 数据配比由诊断信号闭环驱动;把 BFS 的 11,767 题池与 Ghidra 工具链发展为二进制分析领域的标准智能体基准;利用其公开的全阶段检查点与 W&B 日志,系统研究'MDP 中期训练对下游 RL 可塑性的因果贡献'这类社区长期缺乏数据的问题。

复现评估

复现友好度在同类工作中属最高档:GitHub(zgcagi/ZGCM-1)与 HuggingFace 公开预训练、中期训练、后训练三阶段权重、中间检查点、训练代码与配置、分阶段数据集与配方、W&B 遥测日志和全部评测 harness,Binary Function Search 数据集(11,767 题、529 项目)承诺公开。评测协议高度透明:think 模式温度 1.0、top-p 1.0、非智能体基准报 32 次 pass@1 均值,附录 C 完整给出 Web 研究与 BFS 的交互协议(Serper 搜索 + Jina Reader 抽取 + Qwen3-30B-A3B 裁判;Ghidra gcmd 六动作接口)。算力方面:架构对比实验 8 张 H100 即可复现;中期训练与 SFT 规模中等(每步约 12.6M token,SFT 仅 19.46B 打包 token);但完整 4.19T token 预训练需要大规模 H100 集群,学术团队难以整体重跑。总体难度评估:推理复现容易(权重开放),架构与配方级复现中等,全量预训练复现困难。

相似论文推荐