读书笔记 · 约 22,594 字 · 预计推送 3 部分

Transformer 架构演进史:从 Attention Is All You Need 到 Mamba/线性注意力 — 深度精读笔记 未开始

目录

题记与说明

精读日期:2026-10-29
类型:主题研究(网络检索挑选来源 → 逐一精读)
主题问题:Transformer 八年演进中,每一代架构改进分别要解决什么问题?为什么每解决一个旧问题,就暴露或制造一个新问题?
方法:用 WebSearch 检索筛选,锁定 5 个主来源 + 1 个辅来源,全部抓取 arXiv 原文(ar5iv 全文渲染)做逐篇针对性精读提取(原文引述、复杂度表、实验数字),所有引述均来自本次会话实际抓取的内容。

一、来源清单与选择理由

#来源作者/年份链接角色精读深度
1Attention Is All You NeedVaswani et al., 2017arXiv:1706.03762起点原典:Transformer 为什么诞生全文针对性提取(摘要/引言/复杂度表/实验/位置编码)
2Efficient Transformers: A SurveyTay, Dehghani, Bahri, Metzler, 2020arXiv:2009.06732第一代效率改进的完整谱系 + 事后反思分类法逐类提取 + 局限批判原文
3Transformers are RNNs: Fast Autoregressive Transformers with Linear Attention(辅来源)Katharopoulos et al., 2020arXiv:2006.16236线性注意力的奠基之作:O(n) 注意力→RNN数学思路与实验数字提取
4Mamba: Linear-Time Sequence Modeling with Selective State SpacesGu & Dao, 2023arXiv:2312.00752SSM 路线代表作:选择性机制动机/机制/基准/局限逐段提取
5Transformers are SSMs(Mamba-2 / SSD)Dao & Gu, 2024arXiv:2405.21060统一视角:注意力与 SSM 的对偶理论框架与实现结果提取
6Repeat After Me: Transformers are Better than State Space Models at CopyingJelassi, Brandfonbrener, Kakade, Malach, 2024arXiv:2402.01032批评性对照:SSM 的根本局限定理/实验/结论逐项提取

选择逻辑:这条主题的主线是「每代改进要解决什么」,所以来源必须覆盖问题链的每一环——①起点(为何抛弃 RNN)→ ②第一代改进的谱系与失效(综述视角)→ ③线性注意力的数学本质 → ④SSM/Mamba 的选择性革命 → ⑤两条路线的理论统一 → ⑥统一之后暴露出的根本权衡。跳过的替代方案:各类博客科普(如 Illustrated Transformer)只讲机制不讲问题链;Mamba-360 等综述与 Tay 综述功能重叠;FlashAttention 原论文未列入主来源(见第七节「局限」)。


二、主线叙事:每一代改进要解决什么

第 0 代 · 前夜:RNN 的串行枷锁(2017 之前的背景)

RNN/LSTM 把计算沿着符号位置逐步分解,上一步的隐状态是下一步的输入——串行依赖使得训练样本内部无法并行,且任意两 token 之间的信息交互路径长度是 O(n)。这是 Transformer 要解决的第一个问题。

"This inherently sequential nature precludes parallelization within training examples."(Vaswani et al. 2017,引言)

第 1 代 · 起点:Attention Is All You Need(2017)——用并行换掉串行,代价是 O(n²)

要解决:训练无法并行 + 长距离依赖路径太长。

方案:完全抛弃循环和卷积,只用自注意力。自注意力一层内任意两位置路径长度 O(1),序列操作数 O(1),且整层计算可并行。

代价(为后续所有演进埋下伏笔):每层复杂度 O(n²·d)。Vaswani et al. 论文 Table 1:

层类型每层复杂度串行操作数最大路径长度
Self-AttentionO(n² · d)O(1)O(1)
RecurrentO(n · d²)O(n)O(n)
ConvolutionalO(k · n · d²)O(1)O(log_k(n))
Self-Attention (restricted)O(r · n · d)O(1)O(n/r)

注意:表中最后一行「受限自注意力」正是八年后仍在用的思路——用局部窗口/固定模式把 O(n²) 降为 O(r·n)。起点论文里已经预言了第一代改进的方向。

战果:WMT 2014 英德翻译 28.4 BLEU(当时最优),英法 41.8 BLEU,"a new single-model state-of-the-art BLEU score of 41.8 after training for 3.5 days on eight GPUs"(8× NVIDIA P100,大模型训练 30 万步,2.3×10¹⁹ FLOPs)。

同场加映的两个伏笔:①正弦位置编码 vs 学习式位置编码"produced nearly identical results",选正弦是因它"may allow the model to extrapolate to sequence lengths longer than the ones encountered during training"——外推性从此成为架构争论的持久议题;②作者已注意到自注意力的可解释性红利:"As side benefit, self-attention could yield more interpretable models."

(背景补充,未逐字核验原文:2018–2020 年 BERT/GPT 系列证明「Transformer + 大数据 + 大参数」的规模化红利,长上下文需求随之而来,O(n²) 从理论问题变成工程痛点。)

第 1.5 代 · 百家争鸣:近似注意力运动(2019–2021)——解决 O(n²),但多数没真正解决问题

要解决:n 大时注意力矩阵的 n×n 计算/显存爆炸。

方案谱系(依 Tay et al. 2020 综述的分类法,这是本代改进最权威的地图):

类别思路代表模型
固定模式(FP)限制感受野到预定义模式:分块(复杂度 N²→B²)、strided/膨胀窗口、压缩池化Sparse Transformer, Longformer, Image Transformer
组合模式(CP)多种访问模式拼 CoverageSparse Transformer(local+strided 头分工)、Axial
可学习模式(LP)端到端学习谁注意谁:LSH 哈希、在线 k-means 聚类、可学习分桶Reformer, Routing Transformer, Sinkhorn
神经记忆(M)全局 memory/inducing 点从全序列汇聚信息Set Transformer, ETC, Longformer, BigBird
低秩(LR)假设注意力矩阵低秩,沿长度维投影降维(N→k)Linformer
核方法(KR)核特征映射重写注意力,不物化 N×N 矩阵Performer(FAVOR+随机特征)、Linear Transformer
循环(RC)不稀疏化注意力,而用段间循环连接Transformer-XL, Compressive Transformer
下采样(DS)降序列分辨率Funnel, Perceiver, Swin
(附带)MoE不动注意力,稀疏激活 FFNSwitch Transformer, GShard

这一代的整体教训(Tay et al. 的批判性结论,几乎全部命中后来历史):

"We would like to emphasize that complexity does not always translate to real world throughput or latency."
"It is still a mystery to which fundamental efficient Transformer block one should consider using."
"It is commonly misunderstood that efficient attention models always imply that the Transformer is fast. The truth is that many of these efficient attention models, owing to their innovation constraints, may make the model much slower."

具体批判:①各家评测碎片化,"there is not an easy way to compare these models side by side",才有了后来 LRA 基准;②"efficient"一词被严重过载;③不少模型需要定制 CUDA kernel、不能因果 mask、自回归训练时反而慢;④FFN 占了约一半算力,只优化注意力收益有限;⑤综述回顾部分直言:真实应用中原始二次方 Transformer 仍然占统治地位,没有一个变体成为事实标准,"local attention, when done right, can be a really tough baseline to beat"。

历史裁决(背景补充,未逐字核验):这一代在学术上热闹、在工业上几乎全军覆没——真正被大规模采用的是工程路线 FlashAttention(2022,不改注意力的数学,改显存读写,exact attention)和 KV-cache 优化(MQA/GQA)。而学术遗产由两条线继承:核方法 → 线性注意力家族;固定模式 → 后来的滑动窗口(Mistral 等)。

第 2 代 · 线性注意力:把注意力真正变成 RNN(2020)——解决推理 O(n²),代价是「记住一切」变成「忘记大部分」

要解决:自回归解码时每步都要重算注意力(softmax 的 max 归一化无法增量更新),推理代价 O(n²)。

方案(Katharopoulos et al. 2020):把 softmax 换成任意非负相似度的核特征映射 φ,利用矩阵乘法结合律改写:

softmax 注意力:  (Q Kᵀ) V          → 需物化 N×N 矩阵
线性注意力:      φ(Q) (φ(K)ᵀ V)    → φ(K)ᵀV 只算一次,O(N)

φ(x) = elu(x) + 1(有限维正特征映射)。

关键洞察(这一步直接通向 Mamba):因果 mask 下,运行和 Sᵢ = Σ_{j≤i} φ(Kⱼ)Vⱼᵀ 可以每步常数时间更新——

"any transformer layer with causal masking can be written as a model that, given an input, modifies an internal state and then predicts an output, namely a Recurrent Neural Network (RNN)."(Katharopoulos et al. 2020)

数据:CIFAR-10 图像生成吞吐 4,462× 于 softmax 注意力(17.85 vs 0.004 images/s);摘要称自回归推理"up to 4000x faster";线性注意力可在 11GB GPU 上拟合 16,384 长度(softmax 只有 4,096);WSJ 语音识别 PER 8.08 胜 Reformer 9.33。

代价:固定的 φ 决定了「能记什么」由核函数而非内容决定——状态是所有历史 token 的(有损)求和,无法按内容选择性地记住/遗忘。这正是 Mamba 论文诊断的 "inability to perform content-based reasoning"。

(背景补充,未逐字核验:此路线演化出 RWKV、RetNet、Gated Linear Attention、DeltaNet 等,与 SSM 在「无 softmax 的循环形式」上逐渐合流。)

第 3 代 · SSM 与 Mamba:选择性解决「内容无关」的死穴(2021–2023)——解决线性模型的死记硬背,代价是固定小状态

前史(背景补充,未逐字核验):S4(2021)用线性时不变(LTI)状态空间模型 + HiPPO 理论 + 卷积并行训练,在长序列基准(如 LRA)上称王,但 LTI 意味着参数与输入无关。

Mamba 要解决的两个问题:

  1. 注意力:O(n²) + 窗口外的世界不可见;
  2. LTI SSM(及线性注意力):对内容不敏感——
"a key weakness of such models is their inability to perform content-based reasoning"(Gu & Dao 2023)

对注意力的诊断则一句话直指要害:

"an inability to model anything outside of a finite window, and quadratic scaling with respect to the window length"(Gu & Dao 2023)

方案:让参数依赖输入(选择机制)——Δ、B、C 由输入生成(s_B(x)=Linear_N(x),s_C(x)=Linear_N(x),s_Δ(x)=Broadcast_D(Linear_1(x)),τ_Δ=softplus)。论文给出一条漂亮的理论通道:选择机制就是门控——取 N=1, A=−1, B=1 时递推退化为经典 RNN 门 h_t = (1−g_t)h_{t−1} + g_t·x_t;"Δ in SSMs can be seen to play a generalized role of the RNN gating mechanism"(大 Δ 重置状态专注当前输入,小 Δ 保持状态忽略输入)。

根本张力(Mamba 的理论内核,我认为是整条演进史最深刻的一句):

"efficient models must have a small state, while effective models must have a state that contains all necessary information."(Gu & Dao 2023)

选择机制 = 让小状态学会「装什么、扔什么」。

工程:硬件感知选择性扫描——kernel fusion + parallel scan + recomputation;扩展状态 (B,L,D,N) 从不物化到 HBM,参数在 SRAM 中做离散化和递推,显存占用与 FlashAttention 优化的 Transformer 相当;扫描比朴素 PyTorch 实现快 20–40×。

数据:

作者自认的局限(难得的诚实):选择机制对 LTI 擅长的数据(如音频)有损害——"No Free Lunch";评测"limited to small model sizes, below the threshold of most strong open source LLMs",7B+ 是否仍优"remains to assess";微调、上下文学习、RLHF、量化等下游行为未验证。

注意:Mamba 论文并没有承认复制/联想召回缺陷(它反而展示 Mamba-2L 解决归纳头到 1M 长度、selective copying ~99.8%)。这个批评来自下一份文献。

第 4 代 · 统一与合流:SSD 对偶(2024)——证明两条路线是同一个数学对象

要解决:注意力派与 SSM 派相互对立、各造生态的局面;同时 Mamba-1 的 scan 难以用张量并行等成熟系统优化。

主张(Dao & Gu 2024):SSM 与注意力变体是同一类结构化矩阵的不同分解方式:

"We show an equivalence between state space models and a well-studied family of structured matrices called semiseparable matrices."
"A central message of this paper is that different methods of computing state space models can be reframed as various matrix multiplication algorithms on structured matrices."

对偶的两面:线性形式 = SSM 递推(每 token O(1));二次形式 = 注意力式计算 Y = Mx(全序列一次算)。因果 mask 本身就是 1-semiseparable 矩阵;反过来(Theorem 5.2):任何有快速递推形式的核注意力,其 mask 必然是 semiseparable 矩阵——想绕开 SSM 做线性注意力,数学上无路可走。

实践收益:

第 5 幕 · 根本性批评:复制与联想召回暴露状态容量下界(2024)——「便宜」的极限在哪

要解决:Mamba/SSM 派的评测(缩放律、perplexity)掩盖了什么?有没有注意力能做而 SSM 原理上做不了的事?

理论结果(Jelassi et al. 2024):

实验证据:

结论:perplexity 掩盖了架构差异——"results suggest a fundamental gap between transformers and GSSMs on tasks of practical interest";两条路线不是替代关系:

"we believe that future work should focus on building hybrid architectures that endow state space models with an attention-like mechanism."(Jelassi et al. 2024)

(背景补充,未逐字核验:这一预言当年即被兑现——Jamba/Zamba/Samba/Griffin/RecurrentGemma 等混合架构成为 2024–2025 长上下文模型的主流配方:少数层放全注意力负责检索,多数层放线性/SSM 负责吞吐。)


三、逐篇精读卡片(七维度)

卡片 1 · Attention Is All You Need(Vaswani et al., 2017)

卡片 2 · Efficient Transformers: A Survey(Tay et al., 2020)

卡片 3 · Transformers are RNNs(Katharopoulos et al., 2020,辅来源)

卡片 4 · Mamba(Gu & Dao, 2023)

卡片 5 · Transformers are SSMs / Mamba-2(Dao & Gu, 2024)

卡片 6 · Repeat After Me(Jelassi et al., 2024)


四、七维度综合提取(主题级)

维度内容
核心问题序列建模的核心矛盾:表达信息(全连接、内容寻址)与计算效率(线性、常数状态)不可兼得。RNN 串行(不能并行)→ Transformer 二次方(长度不可扩展)→ 近似注意力(丢精度/丢因果性/丢速度)→ 线性注意力与 SSM(丢内容寻址)→ Mamba 选择性(小状态学会取舍,但容量下界仍在)→ SSD 统一(承认同一谱系内的取舍)→ 混合架构(工程上承认取舍、不再强求单一架构通吃)。
主要主张八年演进不是「谁取代谁」,而是围绕一张三元权衡表(每 token 算力 × 状态容量 × 内容寻址能力)的连续移动:softmax 注意力占据「容量+寻址」角,付出二次方算力与 KV 显存;Mamba/线性注意力占据「效率」角,付出检索上限(Jelassi 下界证明这是原理性的,不是工程的);最终合流点在 SSD 的数学统一与混合架构的工程妥协。
方法各代的武器依次是:换计算图(并行注意力)→ 近似/稀疏化数学对象 → 换参数化(核映射/SSM)→ 换参数化+硬件协同设计(选择机制+SRAM/HBM 感知扫描)→ 数学对偶(semiseparable 矩阵)→ 下界证明+机制探针。
数据与证据每一代都有标志性数字:28.4/41.8 BLEU@8GPU-3.5 天(2017);线性注意力 4,462× 生成吞吐(2020);Mamba 吞吐 5×、3B≈6B Transformer、1M token 外推(2023);SSD 比 Mamba-1 scan 快 2–8×、16K 序列比 FlashAttention-2 快 6×、2.7B 胜 Pythia-6.9B(2024);复制任务 Transformer 样本效率 100×、410M Transformer 胜 2.8B Mamba 于电话簿查询(2024)。证据链完整度:起点/SSD/批评三篇相互独立且一致;综述的批判被后续历史(FlashAttention 胜出、近似变体式微)印证。
结论①注意力从「被抛弃的对象」变成「被理解的对象」——SSD 证明它是 SSM 家族的二次形式特例;②「效率」之争的真正裁判是 wall-clock 与下游任务而非常记号复杂度(Tay 批判 + FlashAttention 现象);③架构选择对检索类能力有原理性影响,perplexity 掩盖不了;④当前理性终点是混合:少数注意力层管检索,多数线性/SSM 层管吞吐。
局限①所有缩放律证据止于 ~3B/300B tokens 量级(Mamba 论文自认),大模型时代结论外推存疑;②Mamba-2 的统一以表达力为代价(A 退化为标量×单位阵);③Jelassi 等的下界针对精确检索,真实任务的灰区有多大未量化;④Tay 综述的基准问题(LRA 不含自回归任务)至今没有完全等价物;⑤本笔记的 FlashAttention/GPT/BERT/混合架构现状部分为背景知识补充,未逐字核验原文。
对我的启发①「每一代改进要解决什么」的最优组织法是矛盾链:串行 vs 并行 → 全连接 vs 二次方 → 近似 vs 保真 → 小状态 vs 内容寻址 → 统一 vs 混合——每个架构都是这条链上的一个点,写综述/做分享都可以按「矛盾 → 机制 → 代价 → 下一个矛盾」四段循环展开;②方法论上,这个领域最好的论文都同时做了三件事:给出问题的最小形式化(如「efficient 小状态 vs effective 全信息」)、给出可证明的性质(下界/对偶)、给出 wall-clock 数字——三者缺一容易被历史证伪(第一代近似注意力就是只有复杂度记号的反例);③判断任何「新架构」时先问三个问题:它在三元权衡表的哪个点?它的评测里有没有检索类任务(复制/联想召回/长文档 QA)?它的速度是记号复杂度还是实测 wall-clock?

五、关键原文引述汇集(均来自本次会话对原文的抓取)

Vaswani et al. 2017(arXiv:1706.03762)

  1. "We propose a new simple network architecture, the Transformer, based solely on attention mechanisms, dispensing with recurrence… while being more parallelizable and requiring significantly less time to train."
  2. "This inherently sequential nature precludes parallelization within training examples… The fundamental constraint of sequential computation, however, remains."
  3. "As side benefit, self-attention could yield more interpretable models."
  4. 正弦位置编码 "may allow the model to extrapolate to sequence lengths longer than the ones encountered during training."

Tay et al. 2020(arXiv:2009.06732)

  1. "We would like to emphasize that complexity does not always translate to real world throughput or latency."
  2. "It is still a mystery to which fundamental efficient Transformer block one should consider using."
  3. "While the field is bustling with new Transformer models, there is not an easy way to compare these models side by side."
  4. "It is commonly misunderstood that efficient attention models always imply that the Transformer is fast. The truth is that many of these efficient attention models, owing to their innovation constraints, may make the model much slower."

Katharopoulos et al. 2020(arXiv:2006.16236)

  1. "any transformer layer with causal masking can be written as a model that, given an input, modifies an internal state and then predicts an output, namely a Recurrent Neural Network (RNN)."

Gu & Dao 2023(arXiv:2312.00752)

  1. "an inability to model anything outside of a finite window, and quadratic scaling with respect to the window length"(论注意力)
  2. "a key weakness of such models is their inability to perform content-based reasoning"(论既往亚二次方模型)
  3. "efficient models must have a small state, while effective models must have a state that contains all necessary information."
  4. "Δ in SSMs can be seen to play a generalized role of the RNN gating mechanism."

Dao & Gu 2024(arXiv:2405.21060)

  1. "We show an equivalence between state space models and a well-studied family of structured matrices called semiseparable matrices."
  2. "A central message of this paper is that different methods of computing state space models can be reframed as various matrix multiplication algorithms on structured matrices."

Jelassi et al. 2024(arXiv:2402.01032)

  1. "the transformers need 100x less samples than the best GSSMs to learn the copy task."
  2. "Even the smallest transformer model dramatically outperforms the largest GSSM."(预训练模型复制 C4 时)
  3. "we believe that future work should focus on building hybrid architectures that endow state space models with an attention-like mechanism."

六、时间线总表

标注:★=本次精读的主/辅来源;○=来自精读文献内部的提及;△=背景知识补充(未逐字核验原文)
时间事件解决的问题引入的代价
2017.06★ Attention Is All You NeedRNN 串行不可并行、长依赖路径 O(n)注意力 O(n²d);KV 随长度线性增长
2018–2020△ BERT/GPT/GPT-2/3证明 Transformer 规模化红利长上下文需求爆发,O(n²) 成痛点
2019–2021○/△ 稀疏化家族:Sparse Transformer、Longformer、BigBird、Reformer、Routing Transformer、Linformer、Performer、Transformer-XL把 n×n 变稀疏/低秩/可学习精度、因果性、实现性各有牺牲;wall-clock 常不降反升(Tay 批判)
2020.06★ Transformers are RNNs(线性注意力)自回归推理 O(n²)→O(n),注意力变 RNN状态与内容无关,无法选择性记忆
2020.09★ Efficient Transformers: A Survey;△2021 LRA 基准统一谱系与评测(揭示)领域评测碎片化、复杂度≠速度
2022.05△ FlashAttention / FlashAttention-2exact attention 的显存 IO 优化;工程路线胜出(不改数学)KV 显存仍随长度线性
2021.10△ S4(LTI SSM + HiPPO)长序列建模、卷积并行训练参数与输入无关(无内容感知)
2023.12★ Mamba(选择性 SSM + 硬件感知扫描)LTI 的内容不敏感 + 注意力的二次方固定小状态 → 检索/复制有原理性上限
2024.02★ Repeat After Me量化 SSM 的能力边界(揭示)perplexity 与检索能力脱节
2024.05★ Mamba-2 / SSD 对偶两派统一;借用注意力系统生态A 退化为标量×单位阵(表达力换速度)
2024–2026△ 混合架构成为长上下文主流(Jamba/Zamba/Samba/Griffin 等);线性注意力家族(RWKV/RetNet/GLA/DeltaNet)持续演化兼顾检索与吞吐系统复杂度、设计空间爆炸

七、局限与不确定性(本笔记自身的)

  1. FlashAttention 未精读:第 1.5 代的工程路线是按背景知识补写的(△ 标注),未抓取其原文验证;但「Mamba 显存占用与 FlashAttention 优化的 Transformer 相当」「SSD 与 FlashAttention-2 的 2K 交叉/16K 6×」等数字来自本次精读的 Mamba 与 Mamba-2 原文,可靠性较好。
  2. 各代之间的背景事件(BERT/GPT、S4、混合架构现状)为背景知识补充,未逐字核验,均已用 △ 标注。
  3. 引述精确度:所有带引号的句子来自本次会话对 ar5iv 全文渲染的抓取(由模型提取并压缩到 ≤125 字符级别),个别可能有标点级出入;复杂度表与数字以原文 Table 为准。
  4. 时效性:2024 年后的发展(混合架构、线性注意力新变体)截至本笔记日期没有精读原始文献,只有背景性概述;如需继续深读,建议优先:Jamba(arXiv:2403.19844)、Gated Linear Attention(arXiv:2312.06635)、Griffin/RecurrentGemma(arXiv:2402.19427)。
  5. 中文转述:叙事部分的因果判断(如「历史裁决」)是本人基于精读文献的推断,已尽量与 Tay et al. 的原文批判对齐。

八、这次学到什么

  1. 架构史的组织骨架是矛盾链,不是论文列表:串行 vs 并行 → 全连接 vs 二次方 → 近似 vs 保真 → 小状态 vs 内容寻址 → 统一 vs 混合。任何「某架构已死」的断言都应还原为「它在三元权衡表(算力×状态×寻址)上移动了哪个坐标」。
  2. 复杂度记号不是速度,速度不是能力,能力不是 perplexity:Tay 综述的批判(复杂度≠wall-clock)、FlashAttention 现象(exact 也能快)、Jelassi 实验(perplexity 更低 ≠ 检索更强)三层递进,构成评测任何新架构的三重过滤。
  3. 统一视角的杠杆最大:SSD 证明注意力与 SSM 是同一矩阵的两种算法后,双方的系统优化与理论工具互通,创新从「站队」变为「连续谱上选点」。研究上,能解释对手的框架胜过击败对手的框架。
  4. 最好的论文有共同的三件套:最小形式化的问题陈述(Mamba 的效率-有效性张力)、可证明的性质(Jelassi 的状态容量下界、SSD 的 Theorem 5.2)、实测 wall-clock 数字(Mamba 5×、SSD 2–8×)。只有复杂度记号的第一代近似注意力,就是缺后两件的反面教材。
← 资料库今日播报 →