【LLM】从零开始学大语言模型 | SSM篇 | 11 | SSM、Transformer 与 RNN 的统一视角
SSM、Transformer 与 RNN 的统一视角 前面几篇里,我们分别拆解了 S4、Mamba 以及家族中的其他变体。本篇把视角拔高一层,把 RNN、Transformer 和 SSM 这三大门派放在同一张桌子上,先看它们共同的数学本质,再看具体的指标差异,最后给出实战选型建议。 一、统一视角:三者都在对历史做加权求和 如果剥开 RNN、Transformer 和 SSM 复杂的网络结构(FFN、LayerNorm、激活函数),只看它们最核心的序列混合(Sequence Mixing)机制,也就是“当前输出如何依赖过去的输入”,会发现一个统一的事实。 它们在数学本质上都是在对历史输入做加权求和。假设要计算第 $t$ 个时间步的输出 $y_t$,输入序列是 $x_1, x_2, \ldots, x_t$,三种模型都可以被抽象为同一个公式: $$y_t = \sum_{k=1}^{t} \alpha_{t,k} \cdot x_k$$ 这里的 $\alpha_{t,k}$ 是一个权重(可以是标量,也可以是矩阵),它决定了在计算当前输出 $y_t$ 时,历史第 $k$ 个输入 $x_k$ 到底有多重要。 三大模型的根本分歧因此收敛为一个问题:权重 $\alpha_{t,k}$ 是如何计算出来的? 二、三种计算 $\alpha_{t,k}$ 的思路 1. RNN(LSTM / GRU):黑盒的隐式压缩 RNN 没有显式计算 $\alpha_{t,k}$ 的公式。它把所有的历史信息一股脑塞进一个固定大小的隐藏状态 $h_t$ 中,通过复杂的非线性门控(遗忘门、输入门)进行压缩。 这个方式有两个特点。 隐式且不可控。 很难解释 RNN 为什么记住了第 3 个词而忘了第 5 个词,注意力分布没有显式表达,只能从状态里反推。 天然的衰减与遗忘。 隐藏状态容量有限,每次更新都有信息损耗,$\alpha_{t,k}$ 会随着时间差 $(t-k)$ 的增大而不可避免地趋向于 0。这就是梯度消失现象在权重视角下的对应表现。 ...