MINIBLOG

Blog Note Tags Links About
Home Search
May 17, 2026
miniyuan

大模型 1:Attention 与 Transformer


Image Captioning

Image Captioning Defination

Image Captioning(图像描述生成)要求输入一张图像,输出一句自然语言描述。它是早期典型的多模态生成任务,通常采用:

  • CNN Encoder:从图像中提取视觉特征;
  • RNN / LSTM Decoder:基于视觉特征逐词生成 caption。

若使用 word-level 语言模型,每个 token 对应 vocabulary 中的一个词。实际现代系统常使用 sub-word 分词,例如 BPE,以降低词表规模并提升未知词处理能力。

CNN + RNN Architecture

早期 Image Captioning 的基本架构是将图像编码成向量,再作为 RNN 的初始条件或额外输入。

Pipeline:

  1. 使用在 ImageNet 上预训练的 CNN 提取图像特征。
  2. 选择 CNN 的高层特征,并通过线性层映射到 RNN hidden state 的维度。
  3. 将 <START> token 输入 RNN,开始生成第一个词。
  4. 每一步根据当前 hidden state 输出 vocabulary 上的 softmax 概率。
  5. 从概率分布中采样或选择下一个词,并作为下一步输入。
  6. 当模型输出 <END> token 时终止生成。

RNN 的基本更新可写为:

ht=tanh⁡(Whhht−1+Whxxt+bh)h_t = \tanh(W_{hh}h_{t-1} + W_{hx}x_t + b_h)ht​=tanh(Whh​ht−1​+Whx​xt​+bh​)

其中 xtx_txt​ 是当前输入 token 的 embedding,hth_tht​ 是当前 hidden state。图像特征通常在初始步作为额外偏置或初始状态注入 decoder。

CNN Feature Selection

对于预训练分类 CNN,选择哪一层作为图像表示非常关键。

特征层特点适用性
FC1000 / logits贴近 ImageNet 1000 类分类标签信息过窄,容易只保留类别而丢失细节
FC4096 / penultimate feature更通用的高层语义特征更适合 caption,需要保留物体、属性和场景信息

例如,logits 层可能只强烈表达 person,但 caption 需要 straw hat, surfboard 等更细粒度信息。因此图像描述通常使用倒数第二层或更通用的视觉特征,而不是最终分类 logits。

Special Tokens

Caption decoder 通常需要两个特殊 token:

  • <START>:生成开始标志,作为 decoder 的第一个输入。
  • <END>:生成结束标志,采样到它时停止生成。

注:<END> 不等价于句号。训练数据中一个 caption 可能包含多个句子,句号只是普通标点,而 <END> 表示整个序列终止。

Training Strategy

早期 Image Captioning 常固定 CNN 参数,只训练 RNN decoder 或少量投影层。原因是 CNN 参数量大且已在大规模图像数据上预训练,若在小规模图文对数据集上端到端微调,成本高且容易过拟合。

Caption 训练使用从 <START> 到 <END> 的完整 Backpropagation Through Time(BPTT)。与长文本语言模型不同,这里通常不适合截断 BPTT,因为后续词的生成强依赖初始图像特征。

Problems

早期 CNN + RNN captioning 在小数据集上可能出现一种退化行为:模型表面生成流畅句子,实际更接近 nearest neighbor retrieval。

其机制是:

  1. CNN 将图像映射到 feature space。
  2. 模型在训练集中找到相似图像的视觉表示 feature。
  3. RNN 直接输出训练集中相似样本的 caption。

这说明架构理论上能生成语言和模型实际学会组合式语言能力是两回事。神经网络会优先选择降低 loss 的简单路径,因此需要通过错误案例分析模型到底在做生成、检索还是记忆。

Visual Question Answering

VQA Defination

Visual Question Answering(VQA)输入一张图像和一个自然语言问题,输出答案。例如给定图像和问题 “How many horses are in this image?”,模型需要回答数量。

在早期设置中,VQA 常被建模为分类问题:将最常见的答案集合离散化,例如限定为 1000 个候选答案,再用 softmax 输出类别。

VQA Architecture

典型 VQA 架构如下:

  1. 用 CNN 提取图像特征。
  2. 用 LSTM 编码问题文本,得到问题向量。
  3. 将图像特征与问题特征做融合,例如 concatenation 或 point-wise multiplication。
  4. 融合特征经过全连接层和 softmax,预测答案类别。

对于选择题形式,还可以将每个候选答案也编码为向量,再计算图像-问题联合特征与各答案向量的相似度,选择得分最高的答案。

注:早期 VQA 的一个核心问题是语言偏置。模型可能不真正看图,而是根据问题模式猜答案,例如 What color is the sky? 常预测 blue。因此多模态任务必须分析模型是否真的使用了视觉信息。

Sequence-to-Sequence

Encoder-Decoder Architecture

Sequence-to-Sequence(Seq2Seq)是机器翻译等任务的基础框架。输入为源语言序列 x1,…,xTx_1,\dots,x_Tx1​,…,xT​,输出为目标语言序列 y1,…,yT′y_1,\dots,y_{T'}y1​,…,yT′​,二者长度不必相同。

Encoder RNN 逐步读取源序列:

ht=fW(xt,ht−1)h_t = f_W(x_t,h_{t-1})ht​=fW​(xt​,ht−1​)

Decoder RNN 以 encoder 的最终状态作为初始条件,逐步生成目标序列:

st=gU(yt−1,st−1,c)s_t = g_U(y_{t-1},s_{t-1},c)st​=gU​(yt−1​,st−1​,c)

其中 ccc 是 context vector,常取 encoder 的最后一个 hidden state hTh_ThT​。

Teacher Forcing

Seq2Seq 训练时常使用 Teacher Forcing:即使 decoder 上一步预测错误,下一步输入仍使用 ground truth token,而不是模型自己预测的 token。

训练和测试的差异是:

阶段下一步输入
训练ground truth token
测试上一步模型生成的 token

Teacher Forcing 可以稳定训练,但也会带来 exposure bias:测试时模型必须面对自己生成的历史错误,而训练时没有完全见过这种分布。

Fixed-Size Context Vector Bottleneck

传统 Seq2Seq 将整个输入序列压缩到一个固定长度的 context vector ccc。当输入很短时该方法可行,但当 TTT 很大时会出现瓶颈:

  • 单个向量容量有限,难以保存长句的全部细节;
  • 长距离信息需要通过 RNN hidden state 连续传递,仍受梯度消失影响;
  • decoder 每一步都只能间接访问源序列,无法按需回看输入中的局部片段。

Attention 的核心思想正是让 decoder 在每个输出步都能访问 encoder 的所有 hidden states,而不是只依赖一个固定向量。

Attention

Additive Attention

在带 attention 的 Seq2Seq 中,encoder 仍产生一组 hidden states:

h1,h2,…,hTh_1,h_2,\dots,h_Th1​,h2​,…,hT​

decoder 在第 ttt 步根据上一时刻状态 st−1s_{t-1}st−1​ 与每个 encoder hidden state hih_ihi​ 计算 alignment score:

eti=v⊤tanh⁡(Wsst−1+Whhi)e_{ti} = v^\top \tanh(W_s s_{t-1}+W_h h_i)eti​=v⊤tanh(Ws​st−1​+Wh​hi​)

再用 softmax 得到 attention weight:

αti=exp⁡(eti)∑jexp⁡(etj)\alpha_{ti} = \frac{\exp(e_{ti})}{\sum_j \exp(e_{tj})}αti​=∑j​exp(etj​)exp(eti​)​

最后对 encoder states 加权求和,得到当前步 context vector:

ct=∑iαtihic_t = \sum_i \alpha_{ti}h_ict​=i∑​αti​hi​

其中 αti\alpha_{ti}αti​ 表示 decoder 第 ttt 步对输入第 iii 个 token 的关注程度。该对齐关系无需人工标注,可通过端到端训练自动学得。

Attention Pipeline

Pipeline:

  1. Encoder 读取完整输入序列,得到每个位置的 hidden state。
  2. Decoder 在当前步拿自己的状态作为查询。
  3. 将该状态与每个 encoder state 计算相关性分数。
  4. 对所有分数做 softmax,得到权重分布。
  5. 用权重对 encoder states 加权求和,得到当前 context。
  6. Decoder 使用该 context 生成当前输出 token。

Attention 可以看作一种可学习的软寻址机制:decoder 不再被迫从单一向量中读取全部信息,而是每一步按需要从输入序列中取信息。

Scaled Dot-Product Attention

现代 Transformer 使用更简洁的 Scaled Dot-Product Attention:

Attention⁡(Q,K,V)=softmax⁡(QK⊤dk)V  ∈R(n, dv)\operatorname{Attention}(Q,K,V) = \operatorname{softmax}\left(\frac{QK^\top}{\sqrt{d_k}}\right)V \;\in\mathbb{R}^{(n,\,d_v)}Attention(Q,K,V)=softmax(dk​​QK⊤​)V∈R(n,dv​)

其中 QQQ 是 Query(R(n, dk)\mathbb{R}^{(n,\,d_k)}R(n,dk​)),KKK 是 Key(R(n, dk)\mathbb{R}^{(n,\,d_k)}R(n,dk​)),VVV 是 Value(R(n, dv)\mathbb{R}^{(n,\,d_v)}R(n,dv​))。 dkd_kdk​ 是 key/query 的维度,nnn 为序列长度。

QK⊤QK^\topQK⊤ 的形状为 (n, n)(n,\,n)(n,n),softmax 逐行归一化后每行和为 1,再与 VVV 相乘得到输出 (n, dv)(n,\,d_v)(n,dv​)。

缩放因子 dk\sqrt{d_k}dk​​ 的作用是避免点积值随维度增大而过大。若 QQQ 和 KKK 的各维近似独立且方差为 1,则点积 Q⊤KQ^\top KQ⊤K 的方差约为 dkd_kdk​。不缩放时 softmax 容易饱和,导致梯度变小。

QKV 形式可以理解为检索过程:

向量作用维度
Query QQQ当前 token 想要查询什么信息(n, dk)(n,\,d_k)(n,dk​)
Key KKK每个候选 token 的索引或匹配特征(n, dk)(n,\,d_k)(n,dk​)
Value VVV被加权汇聚的实际内容(n, dv)(n,\,d_v)(n,dv​)
相似度 QK⊤/dkQK^\top/\sqrt{d_k}QK⊤/dk​​未归一化的注意力分数矩阵(n, n)(n,\,n)(n,n)
权重 softmax⁡(⋯ )\operatorname{softmax}(\cdots)softmax(⋯)逐行归一化的注意力分布(n, n)(n,\,n)(n,n)
输出加权聚合后的表示(n, dv)(n,\,d_v)(n,dv​)

Attention 并不是直接用原始输入做加权平均,而是先通过可学习矩阵映射:

Q=XWQ,K=XWK,V=XWVQ = XW_Q,\quad K = XW_K,\quad V = XW_VQ=XWQ​,K=XWK​,V=XWV​

其中输入 X∈R(n, dmodel)X\in\mathbb{R}^{(n,\,d_{\text{model}})}X∈R(n,dmodel​),投影矩阵 WQ, WK∈R(dmodel, dk)W_Q,\,W_K\in\mathbb{R}^{(d_{\text{model}},\,d_k)}WQ​,WK​∈R(dmodel​,dk​),WV∈R(dmodel, dv)W_V\in\mathbb{R}^{(d_{\text{model}},\,d_v)}WV​∈R(dmodel​,dv​)。这样模型可以分别学习如何匹配和匹配后读取什么内容。

多头(Multi-Head)扩展:

设头数为 hhh,通常 dk=dv=dmodel/hd_k=d_v=d_{\text{model}}/hdk​=dv​=dmodel​/h。 每个头独立计算上述 Attention,输出形状为 (n, dmodel/h)(n,\,d_{\text{model}}/h)(n,dmodel​/h)。 hhh 个头拼接后通过 WO∈R(dmodel, dmodel)W_O\in\mathbb{R}^{(d_{\text{model}},\,d_{\text{model}})}WO​∈R(dmodel​,dmodel​) 投影。 最终输出恢复为 (n, dmodel)(n,\,d_{\text{model}})(n,dmodel​),可与输入 XXX 直接残差相加。

Masked Self-Attention

语言模型训练目标是 next token prediction。若第 iii 个 token 在训练时可以看到后续 token,就会泄漏答案,与测试时自回归生成不一致。

Masked Self-Attention 使用 causal mask,将当前位置之后的 attention score 置为 −∞-\infty−∞:

Eij={Eij,j≤i−∞,j>iE_{ij} = \begin{cases} E_{ij}, & j \le i \\ -\infty, & j > i \end{cases}Eij​={Eij​,−∞,​j≤ij>i​

softmax 后,未来 token 的权重为 0。因此每个 token 只能 attend 到自己及其之前的 tokens。

Multi-Head Attention

Multi-Head Attention 将 Q,K,VQ,K,VQ,K,V 投影到多组子空间中,并行执行多次 attention:

head⁡i=Attention⁡(QWiQ,KWiK,VWiV)\operatorname{head}_i = \operatorname{Attention}(QW_i^Q,KW_i^K,VW_i^V)headi​=Attention(QWiQ​,KWiK​,VWiV​)

再拼接各 head 输出并通过输出投影融合:

MultiHead⁡(Q,K,V)=Concat⁡(head⁡1,…,head⁡h)WO\operatorname{MultiHead}(Q,K,V) = \operatorname{Concat}(\operatorname{head}_1,\dots,\operatorname{head}_h)W_OMultiHead(Q,K,V)=Concat(head1​,…,headh​)WO​

不同 head 可以关注不同关系,例如局部邻近、长距离依赖、语法结构或视觉区域之间的关联。

Cross-Attention vs Self-Attention

类型Q 来源K, V 来源相似度矩阵典型用途
Self-Attention同一序列 XXX同一序列 XXX(n, n)(n,\, n)(n,n) 方阵Encoder、双向编码
Masked Self-Attention同一序列 ZZZ同一序列 ZZZ(n, n)(n,\, n)(n,n) 方阵(带掩码)Decoder 第一层(自回归生成)
Cross-Attention目标序列 ZZZ源序列 HHH(nt, ns)(n_t,\, n_s)(nt​,ns​) 矩形Decoder 中间层(读取 Encoder 信息)
  1. Self-Attention 计算过程

    输入 X∈R(n, dmodel)X \in \mathbb{R}^{(n,\, d_{\text{model}})}X∈R(n,dmodel​)。先通过线性投影得到 Query、Key、Value:

    Q=XWQ,K=XWK,V=XWVQ = XW_Q,\quad K = XW_K,\quad V = XW_VQ=XWQ​,K=XWK​,V=XWV​
    张量维度说明
    WQ, WKW_Q,\, W_KWQ​,WK​(dmodel, dk)(d_{\text{model}},\, d_k)(dmodel​,dk​)可学习投影矩阵
    WVW_VWV​(dmodel, dv)(d_{\text{model}},\, d_v)(dmodel​,dv​)通常 dv=dkd_v = d_kdv​=dk​
    Q, KQ,\, KQ,K(n, dk)(n,\, d_k)(n,dk​)同一套 token 的查询与索引
    VVV(n, dv)(n,\, d_v)(n,dv​)同一套 token 的内容

    计算相似度矩阵并缩放:

    E=QK⊤dk∈R(n, n)E = \frac{QK^\top}{\sqrt{d_k}} \in \mathbb{R}^{(n,\, n)}E=dk​​QK⊤​∈R(n,n)

    对 EEE 逐行做 Softmax 归一化,得到注意力权重矩阵:

    A=softmax⁡(E)∈R(n, n)A = \operatorname{softmax}(E) \in \mathbb{R}^{(n,\, n)}A=softmax(E)∈R(n,n)

    其中 AijA_{ij}Aij​ 表示 token iii 对 token jjj 的注意力权重。最后加权聚合 Value:

    Y=AV∈R(n, dv),Yi=∑j=1nAijVjY = AV \in \mathbb{R}^{(n,\, d_v)},\quad Y_i = \sum_{j=1}^{n} A_{ij} V_jY=AV∈R(n,dv​),Yi​=j=1∑n​Aij​Vj​

    每个输出 token 都是全体输入 token 的加权组合,一步实现全局交互。

  2. Cross-Attention 计算过程

    输入为 Decoder 当前状态 Z∈R(nt, dmodel)Z \in \mathbb{R}^{(n_t,\, d_{\text{model}})}Z∈R(nt​,dmodel​) 与 Encoder 输出 H∈R(ns, dmodel)H \in \mathbb{R}^{(n_s,\, d_{\text{model}})}H∈R(ns​,dmodel​)。 投影时 QQQ 来自目标序列,K,VK,VK,V 来自源序列:

    Q=ZWQ,K=HWK,V=HWVQ = ZW_Q,\quad K = HW_K,\quad V = HW_VQ=ZWQ​,K=HWK​,V=HWV​
    张量维度来源作用
    QQQ(nt, dk)(n_t,\, d_k)(nt​,dk​)Decoder ZZZ目标序列”想问什么”
    KKK(ns, dk)(n_s,\, d_k)(ns​,dk​)Encoder HHH源序列的”索引标签”
    VVV(ns, dv)(n_s,\, d_v)(ns​,dv​)Encoder HHH源序列的”实际内容”

    计算跨序列相似度矩阵:

    E=QK⊤dk∈R(nt, ns)E = \frac{QK^\top}{\sqrt{d_k}} \in \mathbb{R}^{(n_t,\, n_s)}E=dk​​QK⊤​∈R(nt​,ns​)

    第 (i,j)(i,j)(i,j) 项表示目标序列第 iii 个 token 对源序列第 jjj 个 token 的匹配分数。逐行 Softmax 归一化:

    A=softmax⁡(E)∈R(nt, ns)A = \operatorname{softmax}(E) \in \mathbb{R}^{(n_t,\, n_s)}A=softmax(E)∈R(nt​,ns​)

    每行和为 1,每个目标 token 在整个源序列上获得注意力分布。最后加权读取源信息:

    Y=AV∈R(nt, dv),Yi=∑j=1nsAijVjY = AV \in \mathbb{R}^{(n_t,\, d_v)},\quad Y_i = \sum_{j=1}^{n_s} A_{ij} V_jY=AV∈R(nt​,dv​),Yi​=j=1∑ns​​Aij​Vj​

    每个目标 token 的输出是源序列所有 token 的 Value 按相关性加权,实现”目标查询源”的信息读取。

  3. Masked Self-Attention(Decoder 专用)

    计算流程与 Self-Attention 完全相同,仅在 Softmax 前加入因果掩码:

    Mij={0,i≥j(now and history)−∞,i<j(future)M_{ij} = \begin{cases} 0, & i \ge j \quad (\text{now and history}) \\ -\infty, & i < j \quad (\text{future}) \end{cases}Mij​={0,−∞,​i≥j(now and history)i<j(future)​ A=softmax⁡(QK⊤dk+M)A = \operatorname{softmax}\left(\frac{QK^\top}{\sqrt{d_k}} + M\right)A=softmax(dk​​QK⊤​+M)

    由于 e−∞=0e^{-\infty} = 0e−∞=0,未来位置权重被强制置零。输出为:

    Yi=∑j=1iAijVjY_i = \sum_{j=1}^{i} A_{ij} V_jYi​=j=1∑i​Aij​Vj​

    第 iii 个 token 只能看到第 111 到第 iii 个 token,保证自回归生成的因果性。

Position Encoding

纯 self-attention 对输入顺序是 permutation equivariant(置换等变)的:

F(σ(X))=σ(F(X))F(\sigma(X)) = \sigma(F(X))F(σ(X))=σ(F(X))

也即,如果输入 token 顺序被打乱,输出也只是按相同方式打乱,具体数值不变。这说明 self-attention 本身不知道 token 的顺序。

因此 Transformer 必须显式加入 Position Encoding(位置编码)。常见方法包括:

  • Absolute Position Embedding:为每个绝对位置学习或定义一个位置向量,并与 token embedding 相加。
  • 2D Position Embedding:在视觉任务中编码 patch 的二维位置。
  • RoPE(Rotary Position Embedding):通过旋转 query 和 key,使点积自然携带相对位置信息。

RoPE 的核心性质可写为:

(R(θi)qi)⊤(R(ϕj)kj)=qi⊤R(ϕj−θi)kj(R(\theta_i)q_i)^\top(R(\phi_j)k_j) = q_i^\top R(\phi_j-\theta_i)k_j(R(θi​)qi​)⊤(R(ϕj​)kj​)=qi⊤​R(ϕj​−θi​)kj​

该式说明,旋转后 query-key 点积依赖相对位置差,而不仅是绝对位置。

Computational Complexity

Self-attention 的主要成本来自 QK⊤QK^\topQK⊤:

QK⊤∈RN×NQK^\top \in \mathbb{R}^{N\times N}QK⊤∈RN×N

因此计算复杂度和 attention matrix 显存都随序列长度呈二次增长,是 O(N2)O(N^2)O(N2) 的。

当 NNN 很大时,例如十万级或百万级 token,直接保存完整 attention matrix 会非常昂贵。

Flash Attention 通过 IO-aware tiling 和重计算优化显存访问,在不近似 attention 结果的前提下降低显存占用。 它计算的是 exact attention,区别于将 attention 近似为线性复杂度的 Linear Attention。

Transformer

Comparison with RNN and CNN

Transformer 用 self-attention 作为 token 之间的信息交互机制。相比 RNN 和 1D CNN,它在并行性和长距离建模上具有明显优势。

特性RNN1D CNNSelf-Attention
并行性顺序计算,难并行高度并行高度并行
感受野理论全局,实际受梯度影响局部,随层数扩大单层即可全局交互
位置信息天然顺序局部结构隐含顺序需显式位置编码
长距离依赖容易遗忘需堆叠多层可直接建立连接
主要代价时间方向顺序执行卷积层数与核大小O(N2)O(N^2)O(N2) attention

RNN 的优势是线性时间步处理,但无法充分并行;CNN 可并行但局部感受野增长慢;self-attention 以更高的空间和计算开销换取全局交互。

Transformer Block

原始 Transformer block 主要由两部分组成:

  1. Multi-Head Self-Attention:负责 token mixing,也即 token 之间的信息交互。
  2. Feed-Forward Network(FFN):对每个 token 独立执行 MLP,负责 channel mixing 和非线性变换。

典型 block 流程:

  1. 输入 token 序列 XXX。
  2. 经过 multi-head self-attention 得到全局上下文表示。
  3. 加 residual connection,并进行 layer normalization。
  4. 经过两层 MLP,即 FFN。
  5. 再次加 residual connection 和 layer normalization。

FFN 常写为:

FFN⁡(X)=σ(XW1+b1)W2+b2\operatorname{FFN}(X)=\sigma(XW_1+b_1)W_2+b_2FFN(X)=σ(XW1​+b1​)W2​+b2​

其中 W1W_1W1​ 通常将维度从 DDD 扩展到 4D4D4D,W2W_2W2​ 再投影回 DDD。

Layer Normalization

Transformer 中通常使用 Layer Normalization。它对每个 token 的 channel 维度独立归一化,而不是跨 batch 或序列长度归一化。

对于 token 表示 xxx,LayerNorm 可写为:

LayerNorm⁡(x)=γx−μσ2+ϵ+β\operatorname{LayerNorm}(x) = \gamma\frac{x-\mu}{\sqrt{\sigma^2+\epsilon}}+\betaLayerNorm(x)=γσ2+ϵ​x−μ​+β

LayerNorm 适合 NLP 和多模态场景,因为 batch 中不同样本、不同 token 的统计分布差异很大,跨 batch 归一化并不稳定。

Transformer for LM

在语言模型中,Transformer (Decoder-only) 的输入输出流程如下:

  1. 将 token id 通过 embedding matrix 映射为 DDD 维向量。
  2. 加入位置编码。
  3. 堆叠多个 masked Transformer blocks。
  4. 将每个位置的输出向量通过 projection matrix 映射到 vocabulary 大小。
  5. 使用 softmax + cross-entropy 训练 next token prediction。

若词表大小为 VVV,模型维度为 DDD,则:

  • embedding matrix 形状为 V×DV\times DV×D;
  • 输出 projection matrix 形状为 D×VD\times VD×V。

注:decoder-only 语言模型必须使用 causal mask;若不 mask,模型训练时会直接看到未来 token,next token prediction 任务会失效。

Modern Improvements to Transformers

Pre-Norm Transformer

原始 Transformer 常采用 Post-Norm,即先 residual addition,再做 LayerNorm。现代大模型更常使用 Pre-Norm:

X′=X+Attention⁡(LN⁡(X))X' = X + \operatorname{Attention}(\operatorname{LN}(X))X′=X+Attention(LN(X)) Y=X′+FFN⁡(LN⁡(X′))Y = X' + \operatorname{FFN}(\operatorname{LN}(X'))Y=X′+FFN(LN(X′))

Pre-Norm 将归一化放到 attention 和 FFN 之前,使 residual path 更接近直接恒等映射,深层网络训练更稳定。

QK-Norm and RMSNorm

QK-Norm 在计算 attention similarity 前对 queries 和 keys 做归一化:

Q=normalize⁡(XWQ),K=normalize⁡(XWK)Q=\operatorname{normalize}(XW_Q),\quad K=\operatorname{normalize}(XW_K)Q=normalize(XWQ​),K=normalize(XWK​)

这样可以控制 QK⊤QK^\topQK⊤ 的尺度,减少梯度尖峰,使训练更稳定。

现代模型也常使用 RMSNorm 替代 LayerNorm。RMSNorm 不减均值,只按均方根缩放:

RMSNorm⁡(x)i=xiRMS⁡(x)γi\operatorname{RMSNorm}(x)_i = \frac{x_i}{\operatorname{RMS}(x)}\gamma_iRMSNorm(x)i​=RMS(x)xi​​γi​

其中:

RMS⁡(x)=ϵ+1d∑j=1dxj2\operatorname{RMS}(x)= \sqrt{\epsilon+\frac{1}{d}\sum_{j=1}^{d}x_j^2}RMS(x)=ϵ+d1​j=1∑d​xj2​​

RMSNorm 计算更简单,实践中常用于大语言模型。

SwiGLU MLP

经典 FFN 为:

Y=σ(XW1)W2Y=\sigma(XW_1)W_2Y=σ(XW1​)W2​

其中 W1∈RD×4DW_1\in\mathbb{R}^{D\times 4D}W1​∈RD×4D,W2∈R4D×DW_2\in\mathbb{R}^{4D\times D}W2​∈R4D×D。

SwiGLU 使用门控结构增强 FFN 表达能力:

Y=(SiLU⁡(XW1)⊙XW2)W3Y = \left(\operatorname{SiLU}(XW_1)\odot XW_2\right)W_3Y=(SiLU(XW1​)⊙XW2​)W3​

其中 W1,W2∈RD×HW_1,W_2\in\mathbb{R}^{D\times H}W1​,W2​∈RD×H,W3∈RH×DW_3\in\mathbb{R}^{H\times D}W3​∈RH×D。门控项使模型可以动态选择哪些 channel 信息通过。

Mixture of Experts

Mixture of Experts(MoE)将每个 Transformer block 中的 FFN 替换为多个 expert MLP。每个 token 只被路由到少数几个 expert:

  • 共有 EEE 个 expert;
  • 每个 token 激活 A<EA<EA<E 个 expert;
  • 总参数量约增加 EEE 倍;
  • 单 token 计算量只约增加 AAA 倍。

Pipeline:

  1. Router 根据 token 表示计算各 expert 的得分。
  2. 选择 top-AAA 个 expert。
  3. token 分别经过被选中的 expert MLP。
  4. 按 router 权重聚合 expert 输出。
  5. 与 residual path 结合进入下一层。

MoE 的优势是扩大总参数量而不等比例增加每次 forward 的计算量。难点在于负载均衡:若 router 总把 token 分给少数 expert,会导致部分 expert 过载、部分 expert 几乎不训练。

Vision Transformer

Patch Embedding

Vision Transformer(ViT)将图像切分为固定大小 patch,并把每个 patch 当作一个视觉 token。

以 224×224×3224\times224\times3224×224×3 图像和 16×1616\times1616×16 patch 为例:

  1. 图像被切分为 14×14=19614\times14=19614×14=196 个 patch。
  2. 每个 patch 大小为 16×16×3=76816\times16\times3=76816×16×3=768。
  3. 将每个 patch flatten 成 768 维向量。
  4. 通过线性层投影到模型维度 DDD。
  5. 加入 2D position embedding。
  6. 输入 Transformer。

分类时,可对所有 patch 输出做 average pooling,再接线性分类层;也可以加入额外的 [CLS] token,用其输出表示整张图像。

ViT Pipeline

Pipeline:

  1. 输入图像 H×W×CH\times W\times CH×W×C。
  2. 按 P×PP\times PP×P 划分为 N=HWP2N=\dfrac{HW}{P^2}N=P2HW​ 个 patch。
  3. 每个 patch flatten 后线性投影为 token embedding。
  4. 加入二维位置编码。
  5. 输入若干层 Transformer encoder。
  6. 汇聚 token 表示,输出图像级分类结果。

与语言模型不同,ViT 分类任务不需要 causal mask。每个 image patch 都可以 attend 到其他所有 patch。

Inductive Bias of ViT / CNN

Inductive Bias(归纳偏置)指模型结构中预设的先验假设。CNN 与 ViT 的核心区别在于归纳偏置强弱不同。

模型归纳偏置特点
CNN局部性、平移等变、层级感受野数据效率高,适合中小规模视觉数据
ViT弱局部先验,依赖位置编码更灵活,依赖大规模数据学习空间关系

CNN 的卷积核天然关注局部区域,并通过堆叠形成 edge、texture、part、object 的层级特征。

ViT 则从第一层开始允许全局 token 交互,没有内置的局部层级结构,因此在小数据集上往往不如 CNN 稳定。

所以 ViT 的性能高度依赖训练数据规模:

数据规模典型现象
ImageNet-1K 级别CNN 往往优于原始 ViT
ImageNet-21K 级别ViT 与 CNN 差距缩小
JFT-300M 级别ViT 可显著超过 CNN

这说明 ViT 并不是简单“比 CNN 更强”,而是在大数据条件下更能发挥弱归纳偏置和全局建模的优势。

ViT Advantages

ViT 的核心价值在于 tokenization:图像 patch 被转化为与文本 token 同质的向量序列。这样视觉 token 和语言 token 可以进入同一个 Transformer,通过 self-attention 直接交互。

在 Vision-Language Model(VLM)中,视觉 token 可以 attend 到语言 token,语言 token 也可以 attend 到图像区域。 这为图文理解、视觉问答、多模态对话和图像生成提供了统一架构基础。

Summary

Architecture, Data, Behaviour

深度学习研究中的三层问题:

  1. 能不能:架构在理论上是否具备完成任务的表达能力。
  2. 数据支不支持:数据量、多样性和标注质量是否足以支撑模型学习真正规律。
  3. 训出来什么样:模型实际学到的是泛化规律,还是记忆、检索或利用数据偏置。

Image Captioning 和 VQA 都说明,神经网络会优先选择最容易降低 loss 的路径。若数据或训练目标允许捷径,模型可能不会学习研究者期望的语义能力。

From RNN to Transformer

RNN、LSTM 和 GRU 通过 hidden state 递推建模序列,但长距离依赖受梯度和顺序计算限制。Attention 让模型在每一步直接访问相关 token,Transformer 则将 attention 作为基础构件,彻底改变了序列建模方式。

Transformer 的关键贡献可以概括为:

  • 用 self-attention 实现全局 token 交互;
  • 用位置编码补足顺序信息;
  • 用 multi-head 提升关系建模能力;
  • 用 FFN 提供 token 内部的非线性变换;
  • 用 residual 与 normalization 支撑深层训练;
  • 通过 tokenization 将语言、视觉和其他模态纳入统一框架。

注: Transformer 已成为现代大模型的核心骨架,但它并不是无代价的结构。 其主要瓶颈仍是长序列下的 O(N2)O(N^2)O(N2) attention 计算,因此 Flash Attention、稀疏 attention、linear attention 和 MoE 等技术都围绕效率与规模化展开。

目录
  • Image Captioning
    • Image Captioning Defination
    • CNN + RNN Architecture
    • CNN Feature Selection
    • Special Tokens
    • Training Strategy
    • Problems
  • Visual Question Answering
    • VQA Defination
    • VQA Architecture
  • Sequence-to-Sequence
    • Encoder-Decoder Architecture
    • Teacher Forcing
    • Fixed-Size Context Vector Bottleneck
  • Attention
    • Additive Attention
    • Attention Pipeline
    • Scaled Dot-Product Attention
    • Masked Self-Attention
    • Multi-Head Attention
    • Cross-Attention vs Self-Attention
    • Position Encoding
    • Computational Complexity
  • Transformer
    • Comparison with RNN and CNN
    • Transformer Block
    • Layer Normalization
    • Transformer for LM
  • Modern Improvements to Transformers
    • Pre-Norm Transformer
    • QK-Norm and RMSNorm
    • SwiGLU MLP
    • Mixture of Experts
  • Vision Transformer
    • Patch Embedding
    • ViT Pipeline
    • Inductive Bias of ViT / CNN
    • ViT Advantages
  • Summary
    • Architecture, Data, Behaviour
    • From RNN to Transformer
© 2026 miniyuan. All rights reserved.
Go to miniyuan's GitHub repo