Google 发布 Muir 线性注意力架构:25MB 状态流式处理 1000 万 token

CategoryNormal149 · reddit · 2026-11-07

Google Research 发布线性注意力论文 Muir(Linear Attention with Sensitivity-Allocated Gating and Matched-Filter Readout),核心卖点是只用 25 MB 固定状态流式处理 1000 万 token——同等长度下 1.3B Transformer 的 KV cache 高达 2.1 TB(即使 2.5-bit 量化仍需 0.32 TB)。

论文解决了 gated delta 类模型(如 GDN-2)在 100B token 训练中暴露的三个瓶颈:

结果:在所有规模上困惑度与零样本推理均优于 GDN-2 和 Transformer(对 GDN-2 有 1.2 倍 token 效率),32K 长度训练速度比 Splash Attention 快 2.7 倍;RULER 5-hop 变量追踪根保留率从 28.4% 提到 48.0%,8K 早期深度召回从 1% 提到 31%;1M token passkey 达 90.0% 零样本 / 99.4%(cooldown 后),而 GDN-2 在 4K-8K 后崩溃、Transformer 超过 64K 就 OOM。

所属事件:Google 发布 Muir 线性注意力:千万 token 流式处理仅 25MB 状态(3 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →