← 全部论文2410.17897v5 / r1
版本与下载

Value Residual Learning:中文精读笔记

  • 作者:Zhanchao Zhou、Tianyi Wu、Zhiyun Jiang、Fares Obeid、Zhenzhong Lan。
  • 论文版本:arXiv:2410.17897v5,2025-06-08 修订;初稿提交于 2024-10-23。
  • 获取与研读日期:2026-09-11;本次成品为 v5/r1。
  • 原始材料:本地论文 PDF、作者主 TeX、作者 acl_latex.bbl 与原始 PDF 图件。源码状态为 available,身份、版本与 PDF 一致;主 TeX 无外部正文 input/include。
  • 阅读范围:正文 Sec. 1–5、Limitations,以及附录 A.1–A.3,包括全部实质性结果与训练配置。参考文献用于核对方法身份,未逐篇研读所引论文。源码中的注释掉的旧表格不作为 v5 结果。
  • 定位约定:下文“p.”均指从 1 开始的 PDF 物理页码,共 16 页;该 PDF 没有连续显示的印刷页码。本文是分析笔记,不是全文翻译,也未运行作者训练代码。

论文总览

主张与核心操作:深层 Transformer 即使保留 hidden-state residual,也未必能有效利用最初的 token 信息。ResFormer 把首层 value $\mathbf V_1$ 直接送到后续层,与该层自己的 value 混合,然后一起乘以当前层的 attention 矩阵。这样,深层可以用当前的语义选择关系重新读取最初的 token 内容。进一步,SVFormer 让所有层只使用 $\mathbf V_1$,每层仍独立生成 query 和 key,从而减少 value 缓存。

最有证据支持的结果:在论文的语言模型预训练设置中,ResFormer 的验证损失与下游平均准确率优于 vanilla Transformer;468M、20B tokens 的 Learnable ResFormer 平均准确率从 40.6% 提高到 42.3%,即 +1.7 个百分点。作者根据 scaling 曲线分别报告等损失时约少用 16.11% 参数或 20.3% 训练 tokens。SVFormer 的缓存优势来自结构,但质量收益有明显条件:2,048 长度时损失更差,64,000 长度和报告的训练预算下可更好。(Sec. 4.2、4.6,pp. 4–5、9,Fig. 2、Tables 2、8)

阅读时应保留的区分:ResFormer 的混合残差主要改善建模;SVFormer 的完全共享主要改善 KV 存储。前者不是参数剪枝,后者不是无条件无损压缩。首层内容旁路是否是改善效果的唯一原因,论文没有给出理论证明。

结构地图

原文章节论证作用PDF 位置与证据
1 Introduction局部信息传播问题,提出两个变体pp. 1–2;Fig. 1
2.1 Shortcut Connections对比 hidden residual、DenseFormer、NeuTRENOpp. 2–3
2.2 KV cache compressing对比 head 与 layer 维度的共享p. 3
3.1 Preliminary基本符号、对照方法公式p. 3;Eqs. (1)–(4)
3.2 ResFormervalue 混合、固定/可学习/稀疏/稠密变体p. 3;Eq. (5)
3.3 SVFormer全层共享首层 valuepp. 3–4;Table 1
4.1 Setting数据、架构、优化与评估pp. 4–5
4.2 ResFormer vs. NeuTRENO, DenseFormerscaling、下游与变体比较pp. 4–5;Figs. 2–3、Tables 2–3
4.3 Truly Better or Just Faster?学习率、梯度路径解释的控制实验pp. 5–6;Fig. 4
4.4 Ablation Study of Value Residual从哪里取、送到哪里、为何不直接改 hidden/Q/K/App. 6–8;Figs. 5–7、Tables 4–7
4.5 Post-Analysis of ResFormervalue 相似度、模块移除与 PCApp. 8–9;Figs. 8–9
4.6 SVFormer vs. GQA,CLAKV 压缩比较、长度与学习率条件p. 9;Tables 8–9、Figs. 10–11
5 Conclusion / Limitations结论及参数选择、规模限制p. 9
Ethics Statement / Acknowledgments / References使用声明与文献pp. 10–12
A.1 Attention pattern analysisattention sinks、value drains、稀疏与重缩放pp. 12–14;Figs. 12–15、Tables 10–11
A.2 Pre-train Dataset7 域数据比例pp. 14–15;Table 12
A.3 Training Details长度、批量与模型尺寸pp. 14–16;Tables 13–14

一:动机——残差保留了路径,是否也保留了可用信息?

1.1 从“能到达”到“能被使用”

初始 embedding 主要表示每个 token 自身的信息;经过 attention,hidden states 融合上下文。普通残差累加保证早期信息存在直接传播路径,但并不保证后续层能从不断叠加的表示中有效提取它。作者把这个问题与过度平滑(over-smoothing)联系起来:不同 token 的表示越来越相似,局部细节可能被序列级语义淹没。(Sec. 1,pp. 1–2)

作者的进一步假设是,直接往深层 hidden states 加回初始 embedding,可能同时改变下一层的 Q、K、V,干扰已经形成的抽象语义 attention 分布。若只向 value 端补充原始内容,则可以沿用当前层的 attention 来决定读取哪些 token。这里“较少干扰”的论点是架构直觉,后文消融提供支持,但并非 Q/K 在整个网络训练中完全不受影响。

1.2 三种连接位置的区别

原图 1:五种结构的连接方式

来源:原 Fig. 1,PDF p. 2,准确裁切,保留五个面板标签。箭头从下向上表示层次推进;蓝线为额外连接;图中省略 MLP、归一化等,不能当成完整网络实现。

方法补充信息进入哪里当前层如何使用首层信息
DenseFormer层输出 hidden state学习组合历史 hidden,影响后续全部 Q/K/V
NeuTRENOattention 输出加上 $\lambda_n(\mathbf V_1-\mathbf V_n)$,该差值未经过当前 attention 聚合
ResFormerattention 输入的 value混合 $\mathbf V_1$ 与当前 value,再使用当前 $\mathbf A_n$
SVFormerattention 输入的 value只保留 $\mathbf V_1$;各层 $\mathbf A_n$ 仍不同

对应 Sec. 3,p. 3,Eqs. (3)–(5)。区别的重点是连接位置和映射方式,而非仅仅增加一条残差线。

二:方法——用当前层的选择关系读取首层内容

2.1 符号与计算流程

以下沿用原符号,并额外用 $\widetilde{\mathbf V}_n$ 表示混合后的 value,避免原 Eq. (5) 对 $\mathbf V_n$ 的覆写造成混淆。

符号意义论文简化维度
$l,d,N$序列长度、hidden 宽度、层数正整数
$\mathbf H_0,\mathbf H_n$token embedding、第 $n$ 层输出$l\times d$
$\mathbf Q_n,\mathbf K_n,\mathbf V_n$当前层投影得到的 query/key/value$l\times d$
$\mathbf W_n^{\mathbf Q},\mathbf W_n^{\mathbf K},\mathbf W_n^{\mathbf V},\mathbf W_n^{\mathbf O}$四个通道投影$d\times d$
$\mathbf A_n,\mathbf U_n$attention 权重、投影前 attention 输出$l\times l$、$l\times d$
$\lambda_{n,1},\lambda_{n,2}$首层 value 与当前 value 的系数每层标量

原文在 Eq. (1)–(2) 中简写为

$$ \mathbf U_n=\operatorname{Attn}(\mathbf Q_n,\mathbf K_n,\mathbf V_n), \qquad \mathbf H_n=\operatorname{Mlp}(\mathbf U_n\mathbf W_n^{\mathbf O}). $$

这组表达式省略了实际 Llama-like 模型的 pre-normalization、hidden residual、多头划分和 RoPE。完整理解仍须结合 Sec. 4.1;不能据此认为作者删除了普通残差。

为解释维度,按单头写出常见 attention 形式(教学展开,非论文新增编号公式):

$$ \mathbf A_n=\operatorname{softmax}_{\rm row} \left(\frac{\mathbf Q_n\mathbf K_n^\top}{\sqrt{d_h}}+\mathbf M\right), \qquad \mathbf U_n=\mathbf A_n\widetilde{\mathbf V}_n. $$

其中 $d_h$ 是 head dimension,$\mathbf M$ 为因果 mask;多头时分别计算并拼接。$\mathbf A_n$ 负责“读取谁”,value 负责“读出什么”。

2.2 ResFormer 的关键公式

首层取标准 value $\mathbf V_1=\mathbf H_0\mathbf W_1^{\mathbf V}$。对 $n\ge2$,原 Eq. (5) 等价写成:

$$ \widetilde{\mathbf V}_n =\lambda_{n,1}\mathbf V_1 +\lambda_{n,2}\underbrace{\mathbf H_{n-1}\mathbf W_n^{\mathbf V}}_{\mathbf V_n}, \qquad \mathbf U_n=\mathbf A_n\widetilde{\mathbf V}_n. $$

由于矩阵乘法对 value 线性,

$$ \mathbf U_n=\lambda_{n,1}\mathbf A_n\mathbf V_1+ \lambda_{n,2}\mathbf A_n\mathbf V_n. $$

这说明首层内容也被当前层的关注关系聚合,并不是原封不动加到 attention 输出上。实现可先混合 V 再调用一次 attention;上式只是解释,不要求计算两次完整 attention。(Sec. 3.2,p. 3;Table 7,p. 7)

“首层 value 是原始信息”的准确意思是:在论文的简化记号中,它只有通道线性投影,没有经过跨 token attention 混合;并非它等于原始 embedding,也并非输入前没有归一化。它由当前输入动态生成,不是跨样本共享的一张固定记忆表。

教学例子:当前 token 的 attention 权重为 $[0.2,0.8]$,首层标量 values 为 $[1,3]$,当前层为 $[10,20]$。固定 $\lambda_1=\lambda_2=0.5$ 后,混合 value 为 $[5.5,11.5]$,输出为 $0.2\times5.5+0.8\times11.5=10.3$。等价地,$0.5\times(0.2\times1+0.8\times3)+0.5\times(0.2\times10+0.8\times20)=10.3$。例中数据仅用于教学,与论文实验无关。

2.3 变体与初始化

变体连接/参数方式应注意的条件
Constant-ResFormer两个系数固定不要求和为 1;例如 $2\mathbf V_1+\mathbf V_n$
Identity-ResFormer两个系数相等,默认均为 0.5本文默认的简洁固定方案,不是将 $\mathbf V_1$ 经恒等映射加到输出
Learnable-ResFormer每层两个可学习标量,均初始化 0.5普通版本未给出非负或归一化约束
Sparse-ResFormer部分层不接首层 value在小模型实验中人工搜索位置与系数
Learnable ResFormer Plus跨层耦合首层 value 的系数初始化/参数化细节不够完整,见下文
Dense-ResFormer组合所有历史层的 value增加更多连接;实测不自动优于只接首层

Plus 在 Sec. 3.2 写为:$\lambda_{n,2}$ 初始化为 0.5;首层分支系数通过

$$ \lambda_{n,1}=\lambda_{\mathrm{scale}} \frac{\exp(\lambda'_{n,1})}{\sum_{j=1}^{N}\exp(\lambda'_{j,1})}, \qquad \lambda_{\mathrm{scale}}\text{ 初始化为 }N $$

得到。按公式求和可知首层分支系数之和为 $\lambda_{\mathrm{scale}}$。如果各 logits 初值相同,则每层首层分支初值为 1,而非 0.5;但论文没有明确给出 logits 初值,也没有充分区分“仅用于初始化”与“训练全程保持此参数化”。不能把“让后层更需要首层信息”的动机写成已规定的单调深度初始化。

Dense-ResFormer 的原文形式为

$$ \mathbf V_n=\lambda_{n,n}\mathbf H_{n-1}\mathbf W_n^{\mathbf V} +\sum_{i=1}^{n-1}\lambda_{n,i}\mathbf V_i,\quad n\ge2. $$

默认当前层系数为 1。源码重用 $\mathbf V_i$ 记号;若做精确复现,需要确认历史 V 缓存取混合前还是混合后,不能仅依据简写推定递归实现。

2.4 SVFormer:把混合退化为完全共享

SVFormer 第一层采用标准 attention,后续层为

$$ \mathbf U_n=\mathbf A_n\mathbf V_1,\quad n\ge2. $$

它可理解为首层分支系数 1、当前层分支系数 0 的极端方案,但论文单独命名。各层 Q、K、attention 分布、输出投影和 MLP 仍可变化,因此“所有层共享 value”不等于“所有层输出相同”。高层仍能利用上下文相关 Q/K,从首层内容中选取不同组合。(Sec. 3.3,pp. 3–4)

训练/推理区别:训练时让梯度从所有相关层共同流向首层 value 投影;自回归推理时,为新增 token 计算一次首层 value,追加到共享缓存,各层只保留各自 K。ResFormer 仍需各层自己的混合 V 缓存,不能直接领取 SVFormer 的缓存节省。两者的改动需要在训练中学习,论文并未证明把现有 checkpoint 直接替换后仍保留性能。

2.5 “近半 KV cache”的精确含义

以下是根据架构推导的元素数量,不是论文实测显存/延迟。设批大小 $B$、缓存长度 $l$、每层 KV 宽度 $d_{kv}$、层数 $N$,忽略额外元数据,且 K/V 宽度相同:

$$ C_{\rm Transformer}=2NBl d_{kv},\qquad C_{\rm SVFormer}=(N+1)Bl d_{kv}. $$

因此剩余比例是 $\frac{N+1}{2N}$,节省比例是 $\frac{N-1}{2N}$。8 层时剩余 56.25%、节省 43.75%;24 层时剩余 52.08%、节省 47.92%。只有深度增大时才趋近 50%。缓存字节数再乘每个元素的字节数。

若在等深、等宽模型上再用 GQA 将 KV heads 减为原来的 $1/g$,SVFormer 相对原 MHA 的缓存比例为 $\frac{N+1}{2Ng}$。例如 $N=8,g=4$ 时是 14.06%,接近但并非恰好 12.5%。Table 8 用“nearly 1/2”和“nearly 1/8”,应保留其近似性。

ResFormer 的额外混合是每层 $O(Bld_{kv})$ 元素操作,attention 仍是原有复杂度;SVFormer 少了多数层的 V 投影,却仍需各层 QK 与 AV 运算。缓存容量、参数数量、FLOPs、训练速度和解码延迟是不同指标。论文未提供完整端到端吞吐或显存测量表,不能由缓存公式直接声称推理加速一倍。

三:实验——在哪些条件下检验了方法?

3.1 主实验协议

项目原文报告来源
模型Llama-like decoder;pre-norm、SwiGLU、RoPE、无 dropoutSec. 4.1,p. 4
训练数据SlimPajama 抽样约 20B tokens,7 个域pp. 4、14–15
tokenizerRedPajama-INCITE-7B-Base,词表 50,277;输入输出 embedding 不绑定p. 4、Table 14,p. 16
尺度82M、180M、320M、468MSec. 4.2,p. 5
默认长度/批量/步数2,048;约 2M tokens/step;10,000 stepsSec. 4.1,p. 4
优化AdamW,$\beta=(0.9,0.95)$,weight decay 0.1,grad clip 1.0p. 4
学习率初始 $10^{-7}$、峰值 $6\times10^{-4}$,cosine 至峰值 10%p. 4
warmup正文 1,200 steps;Table 14 写 120,未能消解pp. 4、16
设备/计算8×A100 80GB,FP16,DeepSpeed ZeRO-2,FlashAttentionp. 4
主指标完整 SlimPajama validation split 上的七域平均 validation loss,越低越好Sec. 4.1,p. 4
可视化样本随机选定的固定 1,000 条序列p. 4
下游zero-shot accuracy 与 WikiText/LAMBADA perplexityTables 2,p. 4;说明 pp. 4–5

七域 loss 的具体域权重、token 加权实现,以及下游评测 harness 版本、提示细节、随机种子与重复实验标准差未明确报告。不要把“七域平均”擅自补成某一种加权方式,也不要给微小差异添加统计显著性。

训练数据比例(Table 12,p. 15):CommonCrawl 50% / 10B,C4 20% / 4B,GitHub 10% / 2B,Books、arXiv、Wikipedia、StackExchange 各 5% / 1B。

Table 14(p. 16)给出的代表配置:

参数规模层数attention headshidden 宽度FFN 宽度
82M885121,792
180M12127682,688
468M24161,0243,584

表中另有 2M 配置,但没有正文 scaling 所用 320M 的完整配置。不能通过插值代填。

3.2 大规模验证单独理解

1.6B 实验使用约 200B 内部预训练 tokens、长度 8,192、学习率 $1.5\times10^{-4}$、批量约 2M tokens,每次运行约 3,584 H800 GPU 小时。这不是“同一份 20B SlimPajama 实验简单放大”,也不使用主实验的默认峰值学习率。Fig. 2 右每 10B tokens 评估一次,支持该设置下 loss 持续改善;不能从源码里已注释掉的 1.6B 下游表补出最终 PDF 没有报告的结果。(Sec. 4.1–4.2,pp. 4–5)

3.3 baseline 与指标的角色

  • vanilla Transformer:同一 Llama-like 基础模型,衡量结构改动本身。
  • NeuTRENO / DenseFormer:改善深层信息传播的不同连接方案;前者默认 $\lambda=0.4$,后者当前层初值 1、历史层初值 0。
  • CLA2:每两层共享 K/V;GQA2/4/8:按论文记号分别将 KV heads 减少到 $1/2,1/4,1/8$。两者用于同量级缓存预算下比较 SVFormer。
  • validation loss / PPL 越低越好;PPL 衡量序列预测的不确定性,不能与 accuracy 混合平均。Table 2 的 Avg. ACC 对应八个 accuracy 列,含 BoolQ 和 LAMBADA accuracy;同一 LAMBADA 的 PPL 列是另一个指标。

四:结果——从有效性到机制与边界

4.1 Sec. 4.2:等 loss 效率与下游表现

原 Fig. 2 左:参数 scaling

原 Fig. 2 中:训练数据 scaling

来源:Fig. 2 左、中,PDF p. 4。第一张横轴为参数量(M),第二张为训练 tokens(B),纵轴均为平均 validation loss,越低越好;不同线条表示结构。应在相同纵轴 loss 水平上读取横向距离,才能理解资源节省,不能把它读成 loss 降低 16.11% 或 20.3%。

作者报告的参数节省来自 82M–468M、20B tokens 的 scaling 曲线;数据节省来自 468M 模型每 2B tokens 的评估曲线。它们是两个比较维度上的等损失估计,没有证明同时减少这两个比例后仍达到同样 loss,也不是单次训练直接实测的总成本折扣。正文未提供足以独立重建这两个精确百分比的拟合公式和置信区间。

原 Fig. 2 右:1.6B 大模型训练

来源:Fig. 2 右,p. 4。横轴为训练 tokens,纵轴为 validation loss;Learnable 与 Plus 曲线在此 1.6B / 200B 内部数据协议下保持优势。它补充了较大尺度证据,但并不证明数十亿至更大尺寸、公开数据或全部下游任务上保持相同收益。

Table 2(p. 4)关键结果,468M / 20B tokens:

方法WikiText PPL ↓LAMBADA PPL ↓八任务 Avg. ACC ↑
Transformer24.833.440.6%
NeuTRENO24.336.341.4%
DenseFormer24.028.040.8%
Identity ResFormer23.832.941.3%
Learnable ResFormer23.732.542.3%
Learnable ResFormer Plus23.231.442.0%

Learnable 比 Transformer 的平均 accuracy 高 1.7 个百分点,而非 1.7% 相对提升。Plus 的 WikiText PPL 最好,但平均 accuracy 略低于普通 Learnable。DenseFormer 的 LAMBADA PPL 更好,因此不能说 ResFormer 在每个指标上全面胜出。

八项 accuracy 原表值(单位 %),用于查看收益分布:

任务TransformerLearnable ResFormer差值(百分点)
ARC-Challenge18.821.2+2.4
ARC-Easy49.650.3+0.7
BoolQ57.560.9+3.4
HellaSwag31.132.3+1.2
LAMBADA34.136.3+2.2
OpenBookQA17.618.8+1.2
WinoGrande49.851.2+1.4
PIQA66.167.0+0.9

4.2 Sec. 4.2:系数与变体有影响

Table 3(p. 5),8 层 / 82M 模型的平均 validation loss:

方法/关键配置loss ↓相对 2.739 的绝对变化
Transformer2.7390
DenseFormer,可学习2.722−0.017
NeuTRENO,固定 0.42.720−0.019
Identity,固定 0.5/0.52.712−0.027
Dense-ResFormer,可学习2.709−0.030
Learnable,0.5/0.5 初始化2.705−0.034
Constant,$2\mathbf V_1+\mathbf V_n$2.700−0.039
Sparse,第 6–8 层只用 $\mathbf V_1$2.696−0.043
Sparse,第 6–8 层用 $5\mathbf V_1+\mathbf V_n$2.687−0.052
Learnable ResFormer Plus2.681−0.058

固定 0.5/0.5 已经有效,因此收益不依赖大量额外参数。不过,手调 Constant/Sparse 优于普通 Learnable,表明学习系数并没有自动找到最优连接方案。Sparse 仅用首层 value 的后 3 层还能省去对应 $\mathbf W_n^{\mathbf V}$,但它不是所有层都共享的 SVFormer。作者明确说 Constant/Sparse 系数经多次实验搜索,表格不是所有方法相同调参成本下的比较。

Fig. 3(p. 5)横轴是固定 $\lambda$,纵轴是 loss;Constant 采用 $\lambda\mathbf V_1+\mathbf V_n$,不能误读成 $\lambda\mathbf V_1+(1-\lambda)\mathbf V_n$。在报告区间内其最优点约为 2,对系数比 NeuTRENO 更稳健;不是任意大的系数都保证更好。

4.3 Sec. 4.3:只是学习更快吗?

原 Fig. 4 右:学习率控制

来源:Fig. 4 右,p. 5;解释见 Sec. 4.3,pp. 5–6。横轴学习率,纵轴平均 loss,比较 Transformer 与 Identity ResFormer,并标出两种首层局部提高学习率的对照。

作者检查了五种学习率,两种模型在约 0.003 附近最优,Identity 在各点均更好。这是专门的学习率消融,不能与默认 $6\times10^{-4}$ 混用。梯度分析(Fig. 4 左)发现,相较 Transformer,Identity 的首层 $\mathbf W_1^{\mathbf V}$ 的输出梯度范数约为两倍,$\mathbf W_1^{\mathbf O}$ 约为一半;作者认为部分梯度通过 value shortcut 绕过原 hidden 路径。

随后仅将 Transformer 首层、或仅 $\mathbf W_1^{\mathbf V}$ 的学习率加倍,都未重现提升。合理结论是收益不能由这些简单的学习率/梯度幅值改动解释;正文“与梯度变化无关”的说法更强,有限控制实验尚不能排除所有优化动力学解释,也没有证明训练无限延长后的全局最优差异。

4.4 Sec. 4.4:从哪里取,送到哪里?

原 Fig. 5 左:value 来源层

原 Fig. 5 右:接入目的层

来源:Fig. 5,p. 6。左图横轴表示来源层,纵轴 loss;右图对比仅接一层和接一组层的 Sparse 配置,应结合图例区分“来源”与“接收层”。在 8 层模型中,首层 $\mathbf V_1$ 最有效,$\mathbf V_2$ 基本无增益;仅接单层时第 7 层最好,多层时第 6–8 层最好,再扩大到较浅层会削弱收益。

这组结果支持“深层更需要初始信息”,而非“任意跨层连接都有效”。相邻连接会让首层信息在多次混合中衰减;稠密连接可能稀释首层分支。后两句是作者对结果的解释,不是测得的独立因果量。

原 Fig. 6 左:学习到的系数比值

来源:Fig. 6 左,p. 6。横轴层号,纵轴 $\lambda_1/\lambda_2$,两条曲线对应 82M 和 468M;较后层整体更依赖首层分支,但曲线并非严格单调。右图 dense 系数热图也突出首层来源,不过原文对其模型尺寸存在矛盾,见后文核验表。

已有 hidden residual 为什么还不够? Table 4(p. 7)比较额外连接到初始 hidden 与首层 value,系数都是可学习的:

额外连接系数初始化loss ↓
无—2.7389
value0.5/0.52.705
hidden0.5/0.52.781
value0/12.730
hidden0/12.722

强行一开始大量注入初始 hidden 会恶化效果;0/1 初始化的 hidden 方案却优于同样初始化的 value 方案。因此“value 在一切公平匹配初始化下都更好”不成立。原文支持的是精心选择的 value 配置效果最好,并且直接 hidden 混合有初始化敏感性。

为什么是 $\mathbf V_1$,而不是 $\mathbf V_2$? 作者通过删除早期 hidden residual 改变信息可达性(Table 5,p. 7):

hidden residual 从何处开始无额外 value加入来源加入后的 loss
$\mathbf H_0$,默认2.739$\mathbf V_1$2.712
$\mathbf H_0$,默认2.739$\mathbf V_2$2.738
$\mathbf H_1$2.780$\mathbf V_2$2.780
$\mathbf H_2$2.820$\mathbf V_2$2.787
$\mathbf H_2$2.820$\mathbf V_3$2.833
$\mathbf H_3$3.057$\mathbf V_3$2.883

一旦较早表示缺少普通残差通路,对应 value shortcut 才产生明显补偿效果。这支持“旁路补充通常难以获取的信息”的解释。它不表示 $\mathbf V_2$ 在所有架构下无用;恰恰相反,其作用依赖 hidden 路径设计。Fig. 7(p. 7)的 hidden 层间相似度显示 $\mathbf H_0$ 与深层差异较大;去掉跨层 hidden residual 后,dense value 比只接首层更好,说明两类路径有互补关系。

能否改接 Q/K/A,或不经过当前 attention? Tables 6–7(p. 7)分别回答:

残差位置(固定 0.5/0.5)loss ↓首层 value 的映射方式loss ↓
无2.739无2.739
Query2.742恒等映射直接加到输出3.137
Key2.746使用 $Q_n$ 与拼接的 $K_n,K_1$ 重算 attention2.729
Attention matrix2.757共用当前层 attention2.712
Value2.712——

重算 cross-layer attention 不增加投影参数,但扩大 attention 计算;直接加到输出损失很高。最有辨识度的结果是:在当前 attention 下混合 value,而非笼统地“首层信息加回来就好”。

4.5 Sec. 4.5:表示究竟发生了什么变化?

原 Fig. 8:首层与其他层 value 相似度(左面板)

来源:Fig. 8 左,p. 8。横轴层号,纵轴余弦相似度;图例区分 vanilla、ResFormer 混合前和混合后。混合后 value 与首层更相似,其中一部分是加入 $\mathbf V_1$ 的直接代数结果;不能单靠这一现象证明泛化提升的原因。对首层依赖较大的层,混合前 value 常与 $\mathbf V_1$ 更不同,作者据此解释为当前层学习补充残差。

Fig. 8 右(原图面板)测的是不同 token 之间的 value 相似度,不是“层与首层相似度”。在 ResFormer 中,深层混合前 value 的 token 间相似度升高,作者假设它只需学习较小的 $\Delta\mathbf V$。但余弦相似度不测范数,也不能单独证明残差幅度下降。

原 Fig. 9 左:逐步移除模块

来源:Fig. 9 左,p. 8。横轴从后向前移除的层/模块数,始终保留首层;纵轴 test loss;图例分别表示移除 attention 或 MLP。ResFormer 移除 attention 的损害更接近移除 MLP 的损害,说明训练后的模型更依赖 attention 分支。它是扰动敏感性实验,不是可直接迁移的剪枝配方。

原 Fig. 9 右:PCA 的 99% 方差维数

来源:Fig. 9 右,p. 8。横轴层号,纵轴解释 99% 方差所需的最少主成分数。从第 2 层起 ResFormer 通常需要更多主成分。作者称之为更高的信息密度;更保守的解释是这些样本上的表示方差分布在更多线性方向上,并不能将 PCA 维数直接等同于任务有效信息量。

4.6 Sec. 4.6:SVFormer 的缓存优势与质量边界

先看为什么只共享 V。Table 1(p. 3)在每两层共享的比较中,无共享 / 只共享 V / 只共享 K / 共享 K+V 的 loss 分别为 2.739 / 2.743 / 2.753 / 2.776。它支持“V 的跨层共享代价较小”,但只共享每两层 V 的结果不等于所有层使用首层 V 的结果。

Table 8 左(p. 9),近半 KV cache 的方法比较;Transformer 行为完整 KV 基线:

方法训练长度 2,048:loss ↓训练长度 64,000:loss ↓
Transformer,全 KV2.7392.753
CLA22.7762.793
GQA22.7482.773
SVFormer2.7742.7485

在短长度,SVFormer 比 Transformer 高 0.035,也比 GQA2 高 0.026;在 64,000 长度,SVFormer 比 Transformer 低 0.0045,比 GQA2 低 0.0245。没有误差条,所以不能把 0.0045 宣称为已证明的显著优势。这里是改变训练序列长度后的模型比较,不是一个短训练模型直接外推到 64k,也不是检索基准上的长上下文能力证据。

Table 8 右(p. 9),64,000 训练长度,近 1/8 KV cache 的方案:

方法loss ↓
Transformer,全 KV2.753
GQA82.807
CLA2 + GQA42.815
SVFormer + GQA42.741

SVFormer 能与 GQA 叠加,并在这个固定协议下保持较低 loss;其中 −0.012 是相对 Transformer 的绝对 loss 差,不是百分比。各压缩方法的有限层数缓存不完全相同,模型总参数也会因投影共享/移除改变,原文未给 Table 8 每一行的完整参数与实际缓存字节数。

原 Fig. 10 左:SVFormer 与基线的训练 loss 差

原 Fig. 10 右:交叉步数随长度变化

来源:Fig. 10,p. 9。左图横轴训练步数,纵轴 SVFormer 相对 vanilla 的 loss 差(负值更好);图例为训练长度 512、2,048、8,192、32,000、64,000。固定每步约 2M tokens,SVFormer 在早期更有优势,随训练推进差值逐渐上升。交叉点明显延后主要出现在 32k 和 64k;512、2,048、8,192 三个较短长度的交叉步数接近,并非严格单调。

右图横轴训练长度,纵轴交叉步数。64,000 的交叉点由最后 1,000 个数据点线性外推得到,不是已经训练观察到的交叉。正文用“exponential trend”描述长度关系,但图中趋势及少量点不足以构成可靠普遍 scaling law;“vanilla 最终一定超越”的全称表述也超出已观察范围。

作者将这种依赖解释为长序列下首层 value 更难充分优化。需要澄清:随长度增长的是 value 激活矩阵的 token 维度,不是 $\mathbf W_1^{\mathbf V}$ 的参数尺寸;论文没有分离上下文分布、每 batch 序列数、参数共享和学习率积分的全部混杂因素。

Table 9(p. 9),报告的是 $\Delta L=L_{\rm SVFormer}-L_{\rm Transformer}$,不是原始 loss:

架构学习率warmup steps$\Delta L$ ↓
Llama-like$10^{-4}$120−0.033
Llama-like$3\times10^{-4}$120+0.021
Llama-like$6\times10^{-4}$120+0.035
Llama-like$6\times10^{-4}$1,200+0.036
GPT2$6\times10^{-4}$120+0.029

更小的学习率影响比这两种 warmup 差异大,作者联系到累积学习率 $\sum_t\eta_t$。这是经验解释,而非严格推导;仅有一个 GPT2 对照也不能证明完全不受架构影响。Fig. 11(p. 9,原图)报告在 82M–468M scaling 比较中,SVFormer 达到同样 loss 需约 多 12.2% 参数。这个参数补偿数字与“等深等宽下近半缓存”的结构推导应分开陈述。

4.7 Appendix A.1:attention sink、value drain 与潜在副作用

测量对象:attention sink 是少数 token(特别是首 token)吸收大量注意力;value drain 指对应 value 范数偏小;hidden/residual peak 指 hidden 范数异常大。它们和 over-smoothing 不是同一指标。Fig. 12(p. 12)显示随机初始化模型可以有 hidden 相似度随深度升高,却不呈现相同注意力集中;Llama 8B 和 Mistral 7B 的曲线只用于已有模型的模式分析,不是 ResFormer 已在这些尺度上训练验证。

原 Fig. 13(a):token importance entropy

原 Fig. 13(b):首 token 的 importance

来源:Fig. 13,p. 12。横轴为层号。ResFormer/SVFormer 的 token importance entropy 整体更高、首 token importance 更低;其余面板显示 value drain 与 hidden peak 缓解。Fig. 14(p. 13)按模型行、importance/value norm/hidden norm 列展示热图;各图色条范围不同,不能只凭同一颜色跨模型比较绝对大小。Fig. 15(p. 14)的 Top-1/2/3/10/1% importance 曲线支持注意力分配较分散的观察,但不保证长上下文检索一定更好。

熵的关键校正(本次分析):附录定义 $e=-\sum_i a'_i\log a'_i$ 后,却写“熵越高越集中”。按该公式,均匀分布的熵最大,集中分布的熵更低。因此图和文字必须以“更高熵 = 更分散”理解,不能照抄方向。

另外,原式 $a_i=\frac1l\sum_j A_{ij}$ 的下标也有歧义。若按常规 $A_{ij}$ 表示 query $i$ 对 key $j$ 的权重,逐行和恒为 1,原式将使 $a_i$ 全部相等;统计 token 被多少 query 关注通常应累计列:$a_i=\frac1l\sum_j A_{ji}$。这是对符号一致性的检查,不是声称作者代码一定实现错了;实际实现需要单独核对。

作者的机制假设:首层没有同样的 value drain,将首层 value 注入后,模型不能再通过“很小的 V + 很大的 attention weight”维持同样输出,于是降低对低语义首 token 的过度关注。这与观测相容,但论文未通过独立干预充分证明唯一因果链。

稀疏/范数副作用:更分散的 attention 可能影响原先的稀疏性。Table 10(p. 14)固定后层组标为 5,6,7,$\lambda=1,2,5$ 时 loss 为 2.698、2.688、2.682;浅层 2,3,4 则为 2.724、2.728、2.732。较大首层权重有利于这一后层组,却损害浅层组;正文叙述把后层写成 6–8,与表格不一致,不能擅自统一。

Table 11(p. 14)使用 value 混合后范数重缩放:

$$ \widetilde{\mathbf V}_n= \frac{\|\mathbf V_n\|}{\|0.5\mathbf V_1+0.5\mathbf V_n\|} (0.5\mathbf V_1+0.5\mathbf V_n). $$

论文没有明确范数沿 token、head 还是整个张量计算,复现时需要确认。全层 loss 2.712→2.701;浅层 2–4 2.734→2.727;深层 6–8 2.702→2.701。这说明缩放对浅层/全层设置更有帮助;不等于已经全面排除了 sparsity、速度或表达能力的副作用。

五:结论、复现边界与值得继续追问的问题

5.1 可以带走的三点

  1. 信息内容与读取关系可以分开设计。当前层的 Q/K 形成语义选择关系,首层 V 提供未经过跨 token 融合的内容旁路;value 连接的位置比“加更多残差”更关键。
  2. 初始信息对后层更有帮助,但最优强度依赖层与初始化。8 层消融、可学习系数及 hidden 路径删除实验相互支持这一结论,尚不能推广为所有架构都应固定接最后三层。
  3. 全层共享 V 是有条件的资源折中。缓存减少由结构保证,质量受训练长度、预算、学习率与模型参数补偿影响;长序列 loss 结果值得关注,但不是通用无损 KV 压缩证明。

5.2 原文报告中的不一致与本次处理

问题可核验位置本笔记处理
warmup 1,200 vs 120Sec. 4.1 p. 4;Table 14 p. 16同时记录,不猜哪一个用于全部训练
batch 算术不一致Table 13 p. 152,048 长度列的 per-GPU 32 × 8 GPUs × accumulation 32 = 8,192 条,非表列总批量 1,024;差 8 倍,实际配置需核对
hidden skip 的 $H_0/H_1$ 混写Table 4 与相邻正文 p. 7表与实验目标明确写 $H_0$;解释按表,保留正文公式疑似笔误说明
Dense learnable 尺寸Fig. 6 右图注 p. 6 写 468M;正文说 8-layer;热图亦仅显示 8 层不将该热图当已确认的 24 层/468M 定量证据
后层 Sparse 的层号Table 10 p. 14 为 5–7;正文及 Table 11 为 6–8各结果沿用其原表层号,不合并数值
entropy 方向与下标A.1 p. 12明确指出数学方向错误与求和下标歧义
Plus / Dense 实现细节欠明确Sec. 3.2 p. 3不臆造 logits 初始化、全程约束或缓存定义
规模配置不完整Sec. 4.2 p. 5 含 320M;Table 14 p. 16 未列不补造 320M 架构;1.6B 按单独协议记录

这些问题影响精确复现与机制解读,但不自动推翻 Tables 2、3、8 报告的数值。论文也没有报告完整随机种子、多次运行误差、所有比较的调参成本、实际缓存字节数及推理吞吐。本文对“更好”的使用限于报告结果,不加上未测得的显著性或部署收益。

5.3 作者限制与本次分析

作者在 Limitations(p. 9)承认可学习系数仍停在相对较优而非最优设置,受算力限制未做更大规模验证。v5 已有 1.6B 实验,所以应理解为更大规模覆盖仍有限,不能说完全没有较大模型实验。主要机制消融仍集中在 82M,主下游结果在 468M;1.6B 使用内部数据,独立复现受数据可得性限制。

本次分析认为,最稳固的是结构对照与多组 loss 改善,较弱的是“准确的单一机制”与普遍效率比例。梯度范数、value 相似度、PCA 和 attention sink 提供互相呼应的线索,但尚未证明原始信息保留、sink 缓解与下游提升之间的完整因果链。

5.4 组会讨论或后续复现的具体问题

  • 在一致的总参数、token、训练 FLOPs 和调参预算下,普通 Learnable、Plus、后层 Sparse 的优势是否稳定?
  • 为何 Table 4 的 0/1 初始化下 hidden 优于 value,而 0.5/0.5 下顺序相反?这能否帮助区分优化与表示因素?
  • 长训练后 SVFormer 的交叉点是否真实出现?在相同累积学习率下改变长度,现象是否仍在?
  • 使用相同任务与样本,能否独立操纵首 token 的 value norm,检验 sink 缓解是否造成质量提升?
  • 真实推理服务中,共享 V 的读取布局、kernel 支持、cache 带宽和并行策略是否把容量收益转化为延迟收益?

这些是根据本文证据提出的研究问题,不是已经开展或验证的实验。

5.5 使用这些成品

英文 Beamer PDF 与 可编辑 TeX 按约 25 分钟主讲组织,附备份页。图件为作者原图或原 PDF 裁切,来源与裁切坐标记录在 figures/manifest.json。重编译命令:

python3 build_slides.py slides.en.tex

需要已安装的 pdflatex 与标准 Beamer;也可用 --pdflatex /absolute/path/to/pdflatex 指定路径。源码阅读、图表核对和幻灯片构建不等于复现论文训练结果。交付核验记录见 quality-checks.md,论文库导航见 总索引。