← 返回 2026-11-02

三思而后行:Metis-1.6 技术报告 Think Twice: The Metis-1.6 Technical Report

Giulianno Vollmer 📅 2026-10-27 👍 5 2026-11-08 18:30
GRPO强化学习 Mamba-2 循环深度 混合专家(MoE) 训练失误复盘 语言模型

单人1.02T token训练的双遍循环MoE模型登上数据与算力帕累托前沿,并如实复盘全部训练失误

前置知识

混合专家模型 (Mixture-of-Experts, MoE)

MoE 把每层 FFN 换成多个并行“专家”网络,路由器为每个 token 选 top-k 个专家计算、其余跳过,从而总参数量大而每 token 计算量不变。本文用细粒度专家加 1 个常开共享专家,负载均衡靠无辅助损失的偏置更新加 router z-loss,专家在压缩潜空间中工作。

Metis-1.6 的每一层都以潜空间 MoE 收尾:Logos 存储 10.23B 参数但每 token 只激活 2.36B,“以 1/12 的数据打平大厂模型”的效率结论完全建立在这个机制上,不懂 MoE 就无法理解其参数账本。

Mamba-2 与混合序列混合器

Mamba-2 把选择性状态空间模型重写为结构化矩阵乘形式以利用矩阵硬件;它以线性复杂度处理序列、隐状态固定、天然携带位置信息,因此无需位置编码(NoPE)也没有逐 token 增长的 KV cache。混合架构(Jamba、Nemotron-H)在多数 SSM 层间插入少量注意力层保留检索能力。

Logos 的 20 层中有 17 层是 Mamba-2、只有 3 层保留 KV cache,这既是它长上下文显存便宜的原因,也是它 131K 上下文检索仍偏弱的结构根源;论文的长上下文短板分析依赖这一点。

循环/权重共享模型 (looped model)

自 Universal Transformer 以来的思路:让同一组网络块以共享权重反复执行,用“递归深度”替代堆叠参数。优点是参数省、可做自适应计算;难点是 label-shift 与残差缩放不稳定。本文最激进——把整个混合 trunk 对每个 token 完整跑两遍。

"one stack, two passes" 是本文的标题级设计(Think Twice),第 10 节的 10B-token 消融专门检验循环该怎么设计,其结论直接催生了下一代架构 MoRE。

GRPO 与组均值归一化优势

GRPO 对同一 prompt 采样一组回答(本文 16 个),用组内奖励均值 $\bar{r}$ 与标准差 $\sigma$ 把优势归一化为 $A_i=(r_i-\bar{r})/\sigma$,免去价值网络。其隐含假设是“中心化不改变奖励排序语义”,而本文第 7.2 节证明:通过率够高时长正确回答的优势会变号。

论文最重要的理论贡献——长度奖励在组均值归一化下的符号反转(Proposition 1)及其一行修复 $A_i\leftarrow\max(0,A_i)$——完全建立在这个机制上,是所有做 RLVR 的人都该知道的反直觉结论。

Muon/AdaMuon 优化器

Muon 对矩阵参数的动量更新做正交化(Newton–Schulz 迭代逼近),使更新在各奇异方向更均匀,配合 Nesterov 动量;AdaMuon/NorMuon 加入二阶矩归一化,Polar Express 给出更优正交化多项式。近年开源大模型训练常用它取代 AdamW 处理矩阵参数。

第 4.2 节的重大事故正是优化器错配:声明的 AdaMuon 从未被实例化,注意力参数因路由名不匹配而全程用 AdamW 训练;第 10 节还量化了修正后的 Muon 变体相对调优 AdamW 的收益($-0.18190$ nats),这是理解本文“错误复盘”主线的钥匙。

研究动机

过去几年语言模型的能力进步基本是用规模买来的:模型不够强就把它做大,参数量像奖杯一样被公布,仿佛大小本身就是成就。但这笔账越来越付不起——在本报告的比较集中,所有 40 基准平均分超过 Logos 的开放模型里,披露训练数据量最低的也用了 28T token(Nanbeige4.2-3B 为 28T,Qwen3-4B 与 Qwen3-8B 各 36T),LFM2.5-8B-A1B 甚至用了 38T;中位数约为 12T。单人研究者根本无法负担这种预算,而更糟的是“安全选择”往往获胜:在 2023 年的配方上再加一万亿参数,而不是冒险换一个更好的配方。于是真正的问题变成:没有前沿预算的人,能否用更好的数据、训练算法与架构,以极低的成本(本文是 1.02T token)换回接近的能力,并在小规模上系统地验证这条路?

本文的目标是Metis-1.6 的目标从来不是挑战前沿模型,而是在小规模上检验 Metis 项目的核心赌注:更好的数据、更好的训练方法与新架构能买到参数买不到的能力,且成本只是零头。具体目标是:由一名研究者从零构建完整技术栈(数据采集与去污染、分词、训练内核、强化学习环境、评测 harness 全部自研,任何一环随时可换),训练两个模型——Praxis(2.94B 参数,每遍激活 0.46B)与 Logos(10.23B 参数,每遍激活 1.18B)——在总计 1.02T token(804.76B 预训练 + 18B 上下文扩展至 131,072 + 200B 监督微调)上,让模型从“能对话”(1.5)进化到“能真正使用”:解奥赛题、调用工具、遵循指令;同时公开 checkpoint 哈希谱系与全部基准数据,并诚实记录每一个错误,为 Metis-1.7 铺路。

与已有工作不同的是,本文的独特切入有三点。第一,架构上敢赌:不堆更多层,而是把同一个 Mamba-2+注意力混合栈对每个 token 完整跑两遍(“Think Twice”),配合 1,024 维潜空间 MoE 与 4 条残差流——这是大厂配方里看不到的组合。第二,方法论上把“失败”当作一等公民:常规发布报告只描述成功的配方,本报告用一半篇幅审计错误——静默错配整个预训练的优化器、被模型自己学会关闭的 1.77B 参数 n-gram 记忆、五个 SFT 失败模式、以及被证明并一行修复的长度奖励符号反转。第三,证明“单人可以靠快速迭代取胜”:想法出现的当天就试验,采用最新而非最安全的设计,冒大组织不敢冒的险,四代模型(1.3 的 201M 到 1.6 的 10.23B)每一代都做出前一代做不到的事。

核心方法

直觉上,模型对每个 token“看两次”:第一遍快速读上下文并写下笔记,第二遍带着笔记重新处理。技术上,Logos 由 20 层组成(宽 2,560),第 5、10、15 层是分组查询注意力(40 query 头/10 KV 头),其余 17 层是 Mamba-2(展开 2、状态 128、头维 64、宽 4 因果卷积);Praxis 是 12 层版本(宽 2,048,注意力在第 4、8 层)。每层末尾是潜空间 MoE:$z=W_{\text{down}}x$ 把表征压到 1,024 维,1 个常开共享 SwiGLU 专家加 top-4 路由专家(Logos 192 个宽 768,Praxis 128 个宽 512)在此计算后投影回模型宽度。整栈每 token 跑两遍,除注意力上的 rank-8 pass-LoRA 与 pass 嵌入外权重全共享;模型维护 4 条残差流,由感知 pass 的控制器读写,混合矩阵经 8 次 Sinkhorn–Knopp 迭代投影为双随机的 $4\times4$ 矩阵;一个 4 头、256 维、输出门初始化为 $-6$ 的跨 pass 深度记忆让第二遍读到第一遍的笔记。全程无位置编码(NoPE),KV cache 仅存在于 3/20 层,词表为 65,536 字节级 BPE。

核心创新是 "one stack, two passes" 的两遍循环设计:与按 token 路由递归深度的既有工作不同,它把整个混合 trunk(序列混合器+MoE)对每个 token 完整执行固定两次,使每个 token 经历 40 个功能阶段(Logos)却只存储 20 层的参数。与简单堆叠的本质区别在于第二遍并非重复劳动:pass 嵌入让注意力经由 rank-8 低秩更新以不同方式行为,4 条残差流与跨 pass 深度记忆承载第一遍写下的中间结论。效率来自潜空间 MoE 与循环的结合——Logos 存储 10.23B 参数但每 token 只触碰 2.36B。第 10 节的受控消融进一步证明这个方向的天花板:把两遍权重完全解绑、加入逐层动态专家数、跨 pass 门控残差高速路与第一遍余弦衰减辅助损失后,循环模型的验证交叉熵达 2.5529(PPL 12.845),以 $-0.0464\pm0.0008$ nats 击败 Metis-1.6 循环(2.5993),并以 $-0.0245\pm0.0008$ 击败专家参数两倍的非循环 16 层模型(2.5774)——一个循环模型打赢两倍参数的非循环模型,正是 MoRE 架构的立足点。

方法步骤详情

流水线分五步。第一步数据:密封数据集 metis-1.6-data-r2 共 806 个来源哈希分片、804,755,808,835 token,按三阶段课程服务——Phase A 68.9%(554.85B)广泛网络/书籍/科学/代码,Phase B 26.1%(209.67B)加权数学/科学/软件,Phase C 5.0%(40.24B)退火精选;每分片用 13-gram shingle 对评测基准去污染。第二步预训练:4,096-token 序列,Logos 219,280 步、Praxis 191,870 步(实际运行的优化器与配置不符,见 4.2 节)。第三步上下文扩展:18B token 长文档(打包至 163,840),Logos 1,968 步、Praxis 3,456 步,支持 131,072 上下文。第四步 SFT:response-only 交叉熵,metis-chat/v1 模板含 think/direct 双模式;cold-start 用 120.01B token 长推理链,overall 用 80.00B token 全域数据,两模型都发布在 Stage 2 末。第五步(仅 Logos)六阶段 RL 研究:GSPO 桥接 264 步、五个领域专家、on-policy 蒸馏——结论进报告但不进发布模型。最后以统一 harness 评测 40 个基准。

技术新颖性

技术新颖性有四。(1) 架构组合:固定两遍循环+潜空间 MoE+pass 感知超连接是新的——既有工作或按 token 路由递归深度,或在循环迭代间加门控高速路,而“整栈跑两遍、权重共享、深度记忆传笔记、rank-8 LoRA 区分两遍”是本文特有设计,且 10B-token 消融给出了 11 个循环变体的量化排序(极差 0.0595 nats)。(2) 理论诊断:Proposition 1 证明组均值归一化下长度奖励符号反转的条件 $p>\frac{1-\lambda}{1-\lambda(1-q)}$(中位数分割、$\lambda=1/2$ 时即 $p>2/3$),与 Dr. GRPO/DAPO 的偏差分析同源但更具体,并给出一行修复 $A_i\leftarrow\max(0,A_i)$。(3) 负结果的证据强度:Engram 式 n-gram 记忆在两模型中都被学习关闭(门均值 0.0001–0.0031),删除 1.77B 参数表后 bits/byte 仅从 0.733898 变为 0.733894,并给出两个结构性解释。(4) 工程审计文化:全程记录 checkpoint SHA-256(Table 3/6)、从张量形状精确计数参数、用统一 harness 重跑所有比较模型。

Metis-1.6 on one page, drawn for Logos. (a) Twenty layers, three of them attention, each ending in a mixture-of-experts block. (b) Inside a layer. (c) The mixture-of-experts block works in a 1,024-dimensional latent space.
Figure 2: Metis-1.6 on one page, drawn for Logos. (a) Twenty layers, three of them attention, each ending in a mixture-of-experts block. (b) Inside a layer. (c) The mixture-of-experts block works in a 1,024-dimensional latent space.

实验结果

主结果:40 个 post-trained 基准上 Logos 平均 52.93,17 模型中第 7,落后 Qwen3-4B(53.68)0.75 分,被 Gemma 4 12B(64.06)等压过;但只用 1.02T token(比较集披露中位数 12T,超过它的模型至少 28T)、每 token 激活 2.36B 参数(超过它的模型为 3.57–10.94B),在数据与算力两轴都处于帕累托前沿。数学最强:11 基准平均 65.83,30 模型中第 2,仅落后 Gemma 4 12B 2.77 分;HMMT Feb 2025 62.92、HMMT Feb 2026 63.64、IMO-AnswerBench 57.00 均为集合最高。短板:代码工具 17/30(HumanEval+ 62.20 落后 13/15)、指令遵循 18/30(IFEval 71.07)、安全 19/30、长上下文 24/30(think 模式 RULER 从 4K 的 68.33 跌至 128K 的 34.17);Praxis 平均 36.62,18 中第 15。上下文扩展实验:131K 埋针检索 19.53%→68.75%,尾部 NLL 5.689→1.495,RULER 128K 达 60.14%。n-gram 记忆负结果:门均值仅 0.0001–0.0031,删除后损失变化小于 $10^{-3}$ nats。RL 实验:GSPO 第 198 步 IFEval +13.41 分但 AIME 60.00%→40.00%;推理专家第 40 步 GPQA Diamond +8.08 至 43.94%,第 78 步因符号反转(通过率超 2/3)回落至 34.34%。优化器消融:修正的 split-Muon+NorMuon 比调优 AdamW 好 $-0.18190\pm0.00299$ nats,即注意力全程误用 AdamW 的代价;循环消融中最优设计 2.5529 nats 击败两倍专家参数的非循环模型(2.5774)。

Configuration and exact parameter counts of the released (n-gram-free) checkpoints.
Table 1: Configuration and exact parameter counts of the released (n-gram-free) checkpoints.
Context extension. Gate metrics are measured on 384 held-out documents at 131,072 tokens.
Table 2: Context extension. Gate metrics are measured on 384 held-out documents at 131,072 tokens.
Training lineage of Metis-1.6 Logos.
Table 3: Training lineage of Metis-1.6 Logos.
Mean n-gram gate and relative residual update at each injection site (pass 1 / pass 2).
Table 4: Mean n-gram gate and relative residual update at each injection site (pass 1 / pass 2).
Eight-suite gate of the Logos post-training study: pass@1 (%), with median generated tokens in parentheses.
Table 5: Eight-suite gate of the Logos post-training study: pass@1 (%), with median generated tokens in parentheses.
Training lineage of Metis-1.6 Praxis.
Table 6: Training lineage of Metis-1.6 Praxis.
Eight-suite gate of Praxis post-training.
Table 7: Eight-suite gate of Praxis post-training.
Released Metis-1.6 models (think mode) against open post-trained models on selected benchmarks.
Table 8: Released Metis-1.6 models (think mode) against open post-trained models on selected benchmarks.
Loop-design campaign (10B tokens, 256 paired evaluation windows).
Table 9: Loop-design campaign (10B tokens, 256 paired evaluation windows).
Optimizer campaign on f2_smart_untie_layer (10B tokens, 9,535 steps, 256 paired windows).
Table 10: Optimizer campaign on f2_smart_untie_layer (10B tokens, 9,535 steps, 256 paired windows).
Mean score on the 40 post-trained benchmarks against (a) published training tokens and (b) active parameters per token. The blue staircase is the Pareto frontier.
Figure 1: Mean score on the 40 post-trained benchmarks against (a) published training tokens and (b) active parameters per token. The blue staircase is the Pareto frontier.
Where Logos does well and where it does not. Each panel is one domain: the unweighted mean of its benchmarks (think mode) against published training tokens.
Figure 3: Where Logos does well and where it does not. Each panel is one domain: the unweighted mean of its benchmarks (think mode) against published training tokens.
The same eight domains against active parameters per token, the compute every generated token costs.
Figure 4: The same eight domains against active parameters per token, the compute every generated token costs.
查看结构化数据
任务指标本文基线提升
40 个 post-trained 基准平均(统一 harness) 平均分 52.93(Logos) Qwen3-4B 53.68;最优开放模型 Gemma 4 12B 64.06 落后 Qwen3-4B 仅 0.75 分,但训练 token 只有 1/35(1.02T vs 36T),每 token 激活参数只有 1/1.5–1/4.6,处于数据与算力双帕累托前沿
竞赛数学(11 个数学基准平均) 平均分(30 模型排名) 65.83,第 2 名 Gemma 4 12B 68.60(第 1 名) 落后 2.77 分;HMMT Feb 2025/2026 与 IMO-AnswerBench 为集合最高,无一个披露训练数据的比较模型超过它
AIME 2025 pass@1, avg@8 (%) 67.50 Qwen3-4B 53.75;Gemma 4 12B 79.17 集合第 3,超 Qwen3-4B 13.75 分
HMMT February 2026 pass@1, avg@8 (%) 63.64 Gemma 4 12B 53.03;Qwen3 8B 31.82 集合最高,超最强比较模型 10.61 分
GPQA Diamond pass@1 (%) 55.05 Qwen3-4B 52.02;Gemma 4 12B 81.31 集合第 5;首个在披露数据上超过它的模型用了 28T token
HumanEval+(代码,弱项) pass@1 (%) 62.20 Gemma 4 12B 83.54 落后 13/15 比较者,代码域排名 17/30
RULER 128K(长上下文,弱项) 通过率 (%) 60.14(direct 模式;think 模式 34.17) 上下文扩展前远低于此;Qwen3.5 4B 在 32K 达 97.22 长上下文域排名 24/30,为最弱领域
10B-token 循环设计消融 验证交叉熵(nats,256 配对窗口) 2.5529(f2_smart_untie_layer:完全解绑+逐层动态专家+高速路+第一遍辅助损失) Metis-1.6 循环 2.5993;非循环 16 层、2 倍专家参数 2.5774 对 Metis-1.6 循环 $-0.0464\pm0.0008$;对两倍参数非循环模型 $-0.0245\pm0.0008$,均超 50σ
10B-token 优化器消融 验证交叉熵(nats) 2.37104(split Muon+NorMuon+Polar Express+调度权重衰减) 调优 AdamW 2.55293 $-0.18190\pm0.00299$,256/256 个评测窗口全胜

局限与改进

作者承认四点。(1) 去污染清单早于若干竞赛:AIME 2025/2026、HMMT 2025/2026、Brumo 2025、IMO-AnswerBench、Apex 2025 不在清单上,而 Logos 恰恰在这些基准上相对最强,存在训练-测试重叠风险。(2) 所有比较用自研 Lernex harness,比较模型在开发者推荐设置下可能更高。(3) 架构结论被优化器事故污染:注意力全程 AdamW、其余 plain Muon,无法分离架构贡献;10B 消融每臂只跑一次且用更小代理,标准误只反映评测噪声。(4) clamp 修复未在发现符号反转的推理专家上重跑。我的补充:基座非常弱(Logos base 44 基准平均 31.03,仅超 OLMoE 的 29.44;Praxis base 24.66 垫底),几乎全部能力来自 200B SFT 与课程加权,“更好配方”的归因还需基座层面对照;40 基准平均掩盖了“数学第 2、长上下文第 24”的巨大方差;RL 研究结论全部未进发布模型,有效性只有 8 套件 gate 的小规模证据。

独立分析的弱点

独立分析四条弱点。(1) 长上下文是结构性短板:think 模式 RULER 从 68.33(4K)跌至 34.17(128K)、LongBench v2 仅 13.20,每 token 只有 3/20 层注意力可能不足以支撑远距离检索——应提高注意力层比例、引入滑动窗口+局部注意力混合,并在退火阶段加入检索类长文档数据(上下文扩展仅占总预算不到 2%)。(2) 能力过度依赖后训练:基座几乎垫底而 post-trained 数学跃居第 2,说明 200B SFT 课程更像“教应试”而非改变表征,泛化存疑;应在预训练退火阶段注入推理数据并做基座-后训练配对消融。(3) 长度奖励与组归一化的交互是系统性陷阱:clamp $A_i\leftarrow\max(0,A_i)$ 是补丁,随通过率上升任何与长度耦合的奖励都会经组均值失真——更根本的解法是解耦正确性优势与长度整形(DAPO 式动态采样或按通过率分层归一化)。(4) 训练工程脆弱:一个名字匹配错误让注意力静默用错优化器 804.76B token 且损失曲线毫无异常;启动断言优化器组已被采纳,但权重范数漂移监测(当时 RMS 漂移 0.57–1.49 倍无人察觉)也应自动化。

未来方向

作者已规划的:Metis-1.7 采用 MoRE 架构——完全解绑两遍权重、逐层动态专家数+逐 pass 路由器、跨 pass 门控残差高速路、第一遍余弦衰减辅助损失(最优配置 f2_smart_untie_layer 达 2.5529 nats);优化器换成启动断言+分块正交化的 split-Muon+NorMuon($\beta_2=0.95$、$\epsilon=10^{-16}$、5 步 Polar Express+1 步重启 Gram 迭代、权重衰减随学习率调度 $\lambda_t=\lambda_0\eta_t/\eta_{max}$);n-gram 记忆扩到 50B 参数、三元格式(1.85 bit/值)存于加速器外并用量化感知直通估计训练;GSPO 桥接改为域平衡配方(30% 数学、20% 科学/STEM、15% 代码、15% 指令、10% 工具、10% 知识);SFT Stage 2 改用参考锚定采样目标(式 (3))混 10% 交叉熵;蒸馏钉住专家最佳 checkpoint 并跑 40–80 步全学习率。Metis-1.7 规模 12.48B 神经参数(每 pass 激活 0.794B、每 token 1.59B),外加 422M 视觉编码器与 204.8M 投机解码 drafter。可延伸方向:把 Proposition 1 推广到一般组归一化奖励整形并实证 clamp 的长期效应;让两遍循环在推理时可变(利用检查点中已有的 halting 头做自适应 pass 数);系统扫描 Mamba-2/注意力比例以修复长上下文短板。

复现评估

复现条件较好但并非开箱即用。开源情况:四个模型(Logos/Praxis 及其 -Base)在 HuggingFace Lernex 组织发布,仓库附带生成报告表格的基准数据、完整 checkpoint SHA-256 谱系(Table 3/6)与逐项精确参数表(Table 1);BF16 权重 Logos 20.80 GB、Praxis 6.16 GB,均已移除 n-gram 记忆。数据集为 sealed release(806 分片、SHA-256 65ef27f6...),是否公开下载未说明;训练、RL 与评测代码是否开源也未提及,这是复现的最大不确定性。算力:预训练在 AMD Instinct MI300A 上进行(1T token 的 10B 级 MoE 属社区可承担范围);RL 阶段用 224 个数据并行 rank、峰值保留显存 74.60 GiB、64 并发沙箱容器,个人复现门槛较高。综合评级:推理与评测容易(单卡即可跑这两个模型),端到端复现难(需重建数据管线、自研融合内核与统一 harness)。

相似论文推荐