读完这篇论文,我终于理解了为什么 AI 圈有个永恒的吵架话题:
"Mamba 比 Transformer 快那么多,为什么大家还在用 Transformer?"答案是:因为快和记得牢之间,有一道数学砌成的墙。而且这道墙不是工程师能拆掉的——它是信息论的基本定律。
🎯 1. 三个看似合理的愿望
假设你要设计一个处理超长文本的模型。你的需求清单大概长这样:
| 愿望 | 符号 | 通俗解释 |
|---|---|---|
| ⚡ 效率 | E | 每处理一个新 token,计算量不能随文本变长而增加 |
| 💾 紧凑 | C | 模型占用的内存不能随文本变长而增加 |
| 🧠 记得牢 | R | 能准确回忆的历史信息量,要和文本长度成正比 |
这三个要求听起来都不过分,对吧?
但 Yan Zhou 的论文用一页证明告诉你:这三个愿望,最多只能实现两个。
不可能三角(impossibility triangle):一个系统中三个 desirable 属性无法同时满足,最多只能满足其中两个的结构性限制。经典例子包括 CAP 定理(一致性、可用性、分区容错)和投资学中的"不可能三角"(汇率稳定、资本自由流动、货币政策独立)。
📐 2. 把"序列模型"抽象成一台机器
为了把这个直觉变成定理,作者定义了一个叫 OSP(Online Sequence Processor)的抽象:
OSP(在线序列处理器):一个七元组,统一描述所有因果序列模型。它维护一个状态 \(s_t\),每读入一个 token \(x_t\) 就更新状态,并能从当前状态回答关于历史的查询。
各部分的含义:
- \(S\):状态空间(模型脑子里存的东西)
- \(\mathcal{X}\):输入字母表,大小 \(V\)(词汇表大小)
- \(Q, A\):查询空间和答案空间
- \(\delta\):状态转移函数(读入新 token 后怎么更新记忆)
- \(\rho\):读出函数(从当前状态怎么回答问题)
- \(s_0\):初始状态
在这个框架下,三个愿望被精确定义为:
E(效率):
状态转移的计算成本被一个关于模型维度 \(d\) 的多项式限制,与序列长度 \(T\) 无关。换句话说,处理第 100 个 token 和第 100 万个 token 的计算量差不多。
C(紧凑性):
状态的比特数被多项式限制,与 \(T\) 无关。模型占用的内存是固定的,不会随着你喂给它的文本变长而膨胀。
R(强召回):
存在某个常数 \(\gamma > 0\),使得模型能以至少 \(1-\varepsilon\) 的准确率,回忆序列中 \(\gamma T\) 个不同的 key-value 对。也就是说,能记住的历史事实数量要和文本长度成正比。
🔒 3. 定理:不可能三角
Theorem 1(不可能三角) 🔺
设 \(\mathcal{P}\) 是一个满足基本公理的 OSP,词汇表大小 \(V \geq 2\)。不存在任何 \(\mathcal{P}\) 同时满足 E、C、R 三者。
更狠的是,作者还给出了一个定量上界:
任何同时满足 E 和 C 的模型,最多只能回忆:
\[n^* \leq \frac{q(d)}{(1-\varepsilon)\log V - 1}\]
个 key-value 对(精度 \(1-\varepsilon\))。
由于 \(q(d)\) 与 \(T\) 无关,这意味着 \(n^* = O(\text{poly}(d)/\log V) = o(T)\)。当序列无限长时,能回忆的信息量相对于序列长度趋于零。
定量上界的含义:即使你把模型做得很大(\(d\) 增加),能记住的 key-value 对最多也只是 \(d\) 的多项式级别,而文本长度 \(T\) 可以无限增长。所以当文本足够长时,模型注定会"遗忘"绝大部分内容。
🧮 4. 证明的核心:信息论的两把刀
证明只用了两个经典工具:
4.1 数据处理不等式(DPI)
DPI:信息在传递过程中只能减少,不能增加。如果 \(X\) 通过 \(Y\) 影响 \(Z\),那么 \(Z\) 能告诉你的关于 \(X\) 的信息,不可能比 \(Y\) 能告诉你的更多。
应用到 OSP 上:输入序列 \(x_{1:T}\) 先被压缩成状态 \(s_t\),然后从 \(s_t\) 中读出答案。DPI 告诉我们:状态 \(s_t\) 能携带的关于输入的信息,有一个上限——就是状态本身的熵。
4.2 Fano 不等式
Fano 不等式把"回忆准确率"和"信息量"联系起来:要准确回忆 \(n\) 个 key-value 对,状态必须携带足够的信息量。但 C 限制了状态的大小,DPI 限制了状态能携带的信息量。
两个不等式一夹,就得到了 \(n^*\) 的上界。
Fano 不等式:在通信和估计理论中,它给出了在存在噪声时,正确解码概率的下界。通俗说:如果你的"信道容量"不够大,就不可能可靠地传输太多信息。
📊 5. 52 个架构的"体检报告"
作者把 2026 年 3 月前发表的 52 个架构 全部丢进了这个三角形,做了系统性分类。
| 架构族 | 代表模型 | E | C | R | 在三角中的位置 |
|---|---|---|---|---|---|
| 🔵 Transformer + KV-cache | GPT, LLaMA | ❌ | ❌ | ✅ | R 顶点 |
| 🟢 SSM | Mamba, S4 | ✅ | ✅ | ❌ | EC 边 |
| 🟢 线性 RNN | RWKV, HGRN2 | ✅ | ✅ | ❌ | EC 边 |
| 🟢 核化注意力 | Performer, cosFormer | ✅ | ✅ | ❌ | EC 边 |
| 🟡 混合架构 | Zamba, Samba | ~ | ~ | ~ | 三角形内部 |
R 顶点(如 Transformer):牺牲效率和紧凑性,换取完美召回。KV-cache 内存随 \(T\) 线性增长,attention 计算量也随 \(T\) 增长(至少是线性读取)。
EC 边(如 Mamba、RWKV):保持固定大小的状态,每步计算恒定,但召回能力被限制在 \(O(\text{poly}(d)/\log V)\)。长文本中的细节注定会丢失。
三角形内部(混合架构):通过调整 attention 层和 recurrent 层的比例,在三个维度之间做连续 trade-off。论文证明这些混合架构形成连续的轨迹,但永远无法触及三个顶点同时相交的区域——因为那区域不存在。
一个特别有趣的发现是:混合架构的 recall 能力随 attention 层比例单调变化。你可以把它想象成一个滑动条:往左滑(更多 recurrent),模型更快更省内存但更容易忘;往右滑(更多 attention),模型记得更牢但更慢更耗内存。
🧪 6. 实验:理论是对的
作者在合成的 associative recall 任务上测试了 5 个代表性架构:
- Transformer(R 顶点)
- Mamba(EC 边)
- Zamba(混合,偏 EC)
- Samba(混合,偏 R)
- 线性 Transformer(EC 边)
结果?
✅ Transformer 几乎完美回忆所有 key-value 对(但内存爆炸)
✅ Mamba 在短序列上表现尚可,但随着序列变长,回忆能力迅速饱和——饱和点正好落在理论预测的 \(O(\text{poly}(d)/\log V)\) 附近
✅ 混合架构的表现介于两者之间,与 attention 层比例一致
✅ 没有任何架构突破理论上限
Associative recall(关联回忆):一种合成的长上下文测试任务。模型被喂入一系列 "key → value" 对(如 "Alice → 工程师"),然后在序列末尾被问到 "Alice 的职业是什么?"。这是测试模型能否在长文本中准确定位和提取特定信息的标准方法。
💡 7. 对实践的启示
7.1 为什么 RAG 不会死
这篇论文从数学上证明了:任何固定内存的模型,在长文本上的精确回忆能力都有硬性上限。
这意味着:
- 如果你需要模型从 100 万 token 的文档中准确找到第 50 万 token 处的一个特定事实
- 而且要求处理速度恒定、内存占用恒定
- 数学上不可能
RAG(检索增强生成)之所以有效,正是因为它把"回忆"任务外包给了外部检索系统,绕过了模型本身的记忆限制。
RAG(Retrieval-Augmented Generation):一种架构,让 LLM 在生成回答前先从一个外部知识库中检索相关文档。这样模型不需要把所有知识都"记"在参数里或上下文里,而是按需查询。
7.2 Mamba 能替代 Transformer 吗?
答案是:取决于你的任务
- 如果你主要做语言建模(下一个 token 预测),Mamba 很快很省内存,很好 ✅
- 如果你需要从长文档中精确提取分散的事实,Mamba 注定会遗漏 ❌
- 如果你做两者兼顾的混合任务,混合架构(如 Zamba)可能是折中选择 ⚖️
7.3 硬件优化的极限
这篇论文的定理是无条件的——它不依赖于任何特定的实现细节或硬件限制。即使你发明了量子计算机,也不可能突破这个界限。
因为这是信息论的基本限制,不是工程问题。
📚 论文详细信息
| 属性 | 内容 |
|---|---|
| 标题 | The Impossibility Triangle of Long-Context Modeling |
| 作者 | Yan Zhou |
| 机构 | School of Mathematics and Statistics, Changsha University of Science and Technology (CSUST), Changsha, Hunan 410114, China |
| arXiv ID | 2605.05066 |
| 发表日期 | 2026-05-06 |
| 分类 | cs.CL, cs.AI, cs.LG |
| 核心贡献 | 证明长上下文模型的不可能三角:效率(E)、紧凑性(C)、召回(R)三者不可兼得;使用数据处理不等式和 Fano 不等式证明任何满足 E 和 C 的模型最多回忆 \(O(\text{poly}(d)/\log V)\) 个 key-value 对;系统分类 52 个架构到三角形的不同区域;实验验证理论界限 |
| 定理 | Theorem 1:不存在同时满足 E、C、R 的 OSP |
| 定量界限 | \(n^* \leq q(d) / [(1-\varepsilon)\log V - 1]\) |
| 证明工具 | 数据处理不等式(DPI)+ Fano 不等式 |
| 分类规模 | 52 个架构(截至 2026 年 3 月) |
| 实验验证 | 5 个代表性架构在合成 associative recall 任务上,实证容量严格低于理论上限 |
#CrushAI #FeynmanLearning #LongContext #InformationTheory #Transformer #Mamba #智柴系统实验室
讨论回复
加载中...正在加载回复...
推荐
智谱 GLM-5 已上线
我正在智谱大模型开放平台 BigModel.cn 上打造 AI 应用,智谱新一代旗舰模型 GLM-5 已上线,在推理、代码、智能体综合能力达到开源模型 SOTA 水平。