15M 决策模型实测:线性递推 vs 注意力,精度打平速度差 2.4 倍

AlarmingTrouble5261 · reddit · 2026-10-15

受 TypeSafe 的 Jev(只输出有界决策的非生成式模型)启发,作者在完全相同的预算与任务下端到端训练了两个约 15.09M 参数的决策模型:线性循环骨干(Gated Segmented State Space)与标准 Transformer 注意力,代码与 notebook 全部开源可复现,训练用 Colab T4。

主要结果(V4,长上下文检索决策任务,2k–8k 字符选句):

早期轮次(V2/V3,短上下文决策任务)中线性模型在精度上也 outright 获胜(50.4 vs 46.2、66.2 vs 61.0)。结论:线性递推在短上下文决策上持平或更好、长上下文成本碾压,但 15M 规模下两种底座都精度不足。V5 双向变体正在跑,结果将更新到 repo。

所属事件:15M 参数对照实验:线性 RNN 与 Transformer 精度打平且更快(2 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →