知识库导读 · 约 1,766 字 · 预计推送 1 部分

平均之路:从Attention到TTT —— 存档导读 未开始

目录

题记与说明

元信息内容
标题平均之路:从Attention到TTT
作者苏剑林
来源 URLhttps://kexue.fm/archives/11993
发布日期2026-11-01
存档日期2026-11-03
标签attention、线性注意力、TTT、DeltaNet、KV-Cache

这篇讲什么

这是苏剑林继《将Softmax Attention线性化为Gated DeltaNet》(archives/11823)之后的续作。上一篇演示了"从 Softmax Attention 逐步引入近似,最后得到一个线性模型"的参考过程;本文换一条新路线:从分布推广出发,把 V 改写成条件分布的期望,再引入三步平均操作,建立起 Attention 与 TTT(Test-Time Training)之间的近似联系——DeltaNet、MesaNet 在这个大框架下自然地成为特例。出发点始终是那件事:压缩 KV Cache。

核心观点

推导链的每一步都是一次"平均",也各付一次代价:

  1. 第一步(分布平均):不硬存 (k_i, v_i),改用参数模型 μ_W(k_i) 预测 V,把 (k_i, v_i) 当作流式到达的训练数据,用 SGD 最大化似然来更新 W——这正是 TTT 的递归形式,收益是省掉 V Cache,W 本质是个以参数为状态的 RNN。
  2. 第二步(平均场近似):Σa_i·μ_W(k_i) ≈ μ_W(Σa_i·k_i),把 Attention 从输出端挪到输入端;对线性模型 μ_W(k)=Wk,这一步从近似变成精确恒等式。
  3. 第三步(加权平均):Σa_i·k_i ≈ λq,输入彻底换成 q,Attention 变成线性复杂度 RNN。作者直言这是过于理想的简化——实践中难以完全吻合其假设,需要定时补回 Softmax Attention,而这正是当前主流的 Hybrid 架构做法。

线性版本里:μ_W(k)=Wk + SGD 流式更新就得到 DeltaNet;改用解析解(G=Σvkᵀ, H=Σkkᵀ)加 Decay 得到 MesaNet,其状态变量要多 50%。

最值得记住的金句与数据

对我的启发

适合谁读

做 LLM 架构、长上下文与推理优化的工程师和研究者;想理清 Attention/TTT/DeltaNet/MesaNet/线性注意力族谱关系的人。需要线性代数和概率功底,不适合零基础入门;但用一条数学主线把一族模型讲透的功夫,正是科学空间十五年如一日的风格。

← 资料库今日播报 →