← 全部论文2605.13473v1 / r1
版本与下载

OSDN:在线预条件如何改进 Delta Rule

论文:OSDN: Improving Delta Rule with Provable Online Preconditioning in Linear Attention
作者:Chenyu Zhou、Hongpei Li(共同一作)、Yuerou Liu、Jianghao Lin、Dongdong Ge、Yinyu Ye。
版本:arXiv:2605.13473v1,提交于 2026-05-13;获取与研读日期:2026-09-22。
材料:原文 PDF、作者 TeX 主入口、源码清单。

本次以 main_arxiv.tex → main.tex 实际引用的正文和附录 A–M 为阅读范围,结合 44 页 PDF 核对公式、页码和关键图表。阅读了方法推导、三种骨干的实现伪代码、两组证明及全部实验附录;没有运行作者模型、独立复现实验或逐篇核验参考文献。以下页码均指 PDF 页码,与此版本印刷页码一致。作者代码地址见原文;本次没有审计代码仓库。

论文总览

主线:把 DeltaNet 的每 token 标量写入步长扩展为逐特征的在线缩放,同时保留大矩阵更新的分块计算形式。 新增状态是一条每头 $K$ 维向量 $d_t$。二次回归损失使更新 $d_t$ 的反馈消去记忆残差,因此可以先从 key 与 gate 计算预条件器,再执行记忆写入。

最明确的任务结果是:340M / 约 10B-token 设置下,OSDN 的 JRT 总体 contains accuracy 从 0.150 升至 0.198,即 +4.8 个百分点、相对 +32%。1.3B / 100B-token 设置下,OSDN-APF 的局部残差比几何均值从 0.432 降至 0.265(约 −38.7%),但重复召回准确率从 0.227 降至 0.217。因此,“39%”不能读成大模型召回准确率提高 39%。证据支持改进写入机制,尚不支持普遍提升下游能力。[Table 1, p.7;Table 8, p.35;Table 16, p.40]

结构地图

原文章节内容与作用PDF 定位
1 Introduction标量写入的局限与两阶段方案pp.1–3,Fig.1
2 Related WorkDeltaNet、门控递归与内层优化的关系p.3
3 Preliminary状态维度、Delta rule、OSGM surrogatepp.3–4,Eq.(1)
4 A Preconditioned Delta Rule写 key 等价、超梯度、APF、chunk 实现pp.4–6,Eqs.(2)–(7),Alg.1
5 Theory: Residual Contraction总体目标与 token-local 两种条件保证pp.6–7,Thms.5.1–5.2
6 Experiments主表、机制诊断、1.3B 扩展pp.7–9,Table 1,Fig.2
7 Discussion作者结论及限制pp.9–10
A Method Details and ProofsWY 推导、单步下降、成本pp.15–17,Table 2
B Backbone Extensions: Gated DeltaNet and KDA门控与转置状态约定pp.17–19,Table 3
C Chunkwise Implementation Pseudocode三种骨干的两阶段实现pp.20–23,Listings 1–3
D Theoretical Analysis: Quadratic Memory-Regression Dynamics完整证明、反例、比较器差距pp.23–31,Thms.D.5/D.7,Cor.D.8
E Benchmark Suite and Metrics各评测的定义与用途pp.31–32
F Training and Evaluation Reproduction Details训练、随机种子、packing、扩展配置pp.32–34,Tables 4–5
G Additional 340M Benchmark Breakdowns逐任务和跨骨干结果pp.34–37,Tables 6–12,Fig.3
H PG-19 Length Extrapolation Details20K 外推与位置桶pp.37–38,Fig.4,Table 13
I OSDN Variant Ablations初始预条件器及 APF 消融pp.38–39,Table 14
J Training-Loss Diagnostic匹配 token 预算的 CE 曲线p.39,Fig.5
K 1.3B / 100B Scaling Breakdown大模型逐任务与缺失评测说明pp.39–41,Tables 15–18
L Inference Throughput Protocol and Results推理吞吐、状态内存、内核一致性pp.41–42,Tables 19–20
M Extended Related Work Taxonomy内层优化视角的扩展分类pp.42–44,Table 21

注:附录 C 与 D 在 p.23 交接(D 标题从 p.23 开始);浮动图表可能先于所属小节出现,例如 Table 14 在 p.38、小节 I 在 p.39。

一:动机

1.1 DeltaNet 已经会“纠错”,为什么还需要预条件?

普通加性线性注意力将 $v_tk_t^\top$ 累加到记忆矩阵,同向写入容易叠加。DeltaNet 先读出 $S_{t-1}k_t$,再只写入当前值与读出值的差。这使它可以覆盖过时关联,但每个 key 坐标仍共享一个标量步长 $\beta_t$。

作者认为,不同坐标受到的写入频率和局部几何不同,共享步长需要折中。OSDN 用 $d_{t,j}$ 调整每个坐标的有效步长 $\beta_td_{t,j}$。这里的“曲率适应”是一种优化解释;实际算法并未维护或求逆完整协方差矩阵。它优化的是后文的 token-local surrogate,不能直接等同于逐渐学到了总体 Hessian 的逆。[Sec.1, pp.1–3;Sec.4, pp.4–6]

1.2 真正的设计约束

预条件方法必须同时满足三点:新增状态足够小;不依赖正在更新的大记忆矩阵;不破坏已有 chunkwise DeltaNet 的矩阵乘结构。直接对总体目标使用稠密 Newton 预条件不符合后两点。本文的关键选择是“对角约束 + 当前 token 的二次损失”,由此得到可以解耦的闭式反馈。

相关工作的比较角色:DeltaNet 是直接宿主与主要 baseline;Gated DeltaNet(GDN)和 KDA 检验能否与其他状态门控组合;OSGM 提供预条件器的在线学习框架;TTT/MesaNet 是优化视角的背景,本文没有与它们开展同预算主实验。[Sec.2;App.M]

二:方法

2.1 符号、输入与输出

符号含义与维度
$x_t$当前层 token 表征,维度 $d_m$
$q_t,k_t\in\mathbb R^K$query、物理/read key
$v_t\in\mathbb R^V$value
$S_t\in\mathbb R^{V\times K}$DeltaNet 每头快速记忆
$u_t=v_t-S_{t-1}k_t$写入前残差,$V$ 维
$\beta_t\in(0,1)$学习得到的标量写门
$d_t\in\mathcal D=[d_{\min},d_{\max}]^K$在线预条件向量
$D_t=\operatorname{Diag}(d_t)$对角右预条件器
$\tilde k_t=d_t\odot k_t$存储/write key,$K$ 维
$\eta$、$r_{t,h}$预条件器内层步长、每 token 每头保留门
$o_t=S_tq_t$层的记忆读出,$V$ 维

原文还用 $q_t$ 表示残差比值,与 query 向量重名;下文涉及 $q_{\rm geo}$ 或收缩时,$q_t$ 均指标量比值。序列长度与平滑常数在原文也都可能记作 $L$,应按上下文区分。

2.2 右预条件恰好等于只缩放写 key

对 token 的回归目标为

$$ f_t(S)=\frac12\|Sk_t-v_t\|_2^2,\qquad \nabla f_t(S_{t-1})=-u_tk_t^\top. $$

把 DeltaNet 的梯度步右乘 $D_t$:

$$ S_t=S_{t-1}-\beta_t\nabla f_t(S_{t-1})D_t =S_{t-1}+\beta_tu_t(d_t\odot k_t)^\top. $$

所以只需改变写入方向。残差计算仍用原始 $k_t$,query 读出仍用 $q_t$。 同时替换残差里的 key 会得到不同算法。记忆转移变成

$$ S_t=S_{t-1}(I-\beta_tk_t\tilde k_t^\top)+\beta_tv_t\tilde k_t^\top. $$

秩一结构保留,但原来对称的 $k_tk_t^\top$ 变成不对称的 $k_t\tilde k_t^\top$。[Sec.4.1/4.3,Eqs.(2)/(5),pp.4–5]

2.3 为什么更新 $d_t$ 不必读取 $S_t$?

令 $s_t=k_t\odot k_t$、$n_t=\|k_t\|_2^2$,并令 $a_t=\beta_t\langle d_t,s_t\rangle$。一次写入后,在同一个 key 上回读得到

$$ S_tk_t-v_t=-u_t(1-a_t),\qquad f_t(S_t)=\frac12\|u_t\|_2^2(1-a_t)^2. $$

用梯度平方归一化损失变化,残差模长被消去:

$$ h_t(d_t)=\frac{f_t(S_t)-f_t(S_{t-1})}{\|\nabla f_t(S_{t-1})\|_F^2} =\frac{(1-\beta_t\langle d_t,s_t\rangle)^2-1}{2n_t}, $$
$$ \nabla_dh_t(d_t)=-\frac{\beta_t}{n_t}(1-\beta_t\langle d_t,s_t\rangle)s_t. $$

公式只含 key、标量门与当前 $d_t$;不再含 value、记忆或残差。因此第一阶段可以独立运行。这个推导对非零 key 且非零写入前残差直接成立;零残差的定义存在额外细节,见后文边界。[Lemma 4.1/Eq.(3), p.5;App.A.3, pp.15–16]

2.4 在线更新、顺序与单步稳定性

实践中用 $\bar n_t=\max(n_t,\epsilon)$:

$$ d_{t+1}=\Pi_{\mathcal D}\left[d_t+ \eta\beta_t\frac{1-\beta_t\langle d_t,s_t\rangle}{\bar n_t}s_t\right]. $$

先使用 $d_t$ 发出 $\tilde k_t$,再更新到 $d_{t+1}$,不能把两步交换。340M 最终配置给出 $d_1=\mathbf1$、$\eta=0.003$、$\mathcal D=[0.5,2]^K$,代码示例的数值地板为 $10^{-6}$。[Alg.1, p.5;Listing 1, pp.20–22;App.F, p.32]

若 key 单位归一化,则 $\langle d_t,s_t\rangle$ 是坐标缩放的加权平均,落在 $[0.5,2]$。由于 $0<\beta_t<1$,有 $0

$$ q_t=\frac{f_t(S_t)}{f_t(S_{t-1})}=(1-a_t)^2<1. $$

这保证当前 token 写入后更接近自身目标,不保证早先 token 的关联不被破坏。此外它是区间给出的稳定性,不等同于在线预条件一定优于 baseline。[Cor.A.2, p.16]

教学例子(非论文实验):$k=(1,0)$、$\beta=0.5$ 时,$d=(1,1)$ 给出 $q=0.25$;若后续在线学习将该方向提高到 $d_1=1.5$,则 $q=0.0625$。这是相同残差的单步损失比改善,不是准确率提高四倍,也不表示一次 $\eta=0.003$ 更新就能达到 $1.5$。

2.5 两阶段与 chunkwise 公式

阶段 1 顺序读取 $(k_t,\beta_t)$,输出整段 $\tilde K$,每头常驻状态 $O(K)$、每 token 工作量 $O(K)$。未投影更新可写成 $A_td_t+b_t$,其中 $A_t$ 为单位阵减秩一项;无需构造稠密 $K\times K$ 矩阵。加入 clamp 后是分段仿射。不能因“仿射 scan”就断言整个第一阶段已获得常数深度或对数深度并行;Table 2 明列其顺序深度为序列长度。[pp.5–6、17]

阶段 2 使用既有 WY 分块结构。省略 chunk 下标,$K,\tilde K\in\mathbb R^{C\times K}$、$V\in\mathbb R^{C\times V}$、$B=\operatorname{Diag}(\beta)$,则

$$ T=(I+\operatorname{tril}(BK\tilde K^\top,-1))^{-1},\quad W=TBK,\quad U=TBV, $$
$$ S_{\rm next}=S+(U-WS^\top)^\top\tilde K. $$

三角系统、矩阵形状保留;块内输出的得分因子用 $Q\tilde K^\top$。注意 $W$ 中仍为 $K$。这说明算法结构兼容原来的硬件计算布局,不意味着现有对称 key 的 kernel 不作接口改动即可调用。物化整段 $\tilde K$ 的临时存储也不同于常驻的 $K$ 维状态。[App.A.2, p.15;App.C, pp.20–23]

训练时外层参数仍通过语言模型损失反传:$\tilde k_t=d_t\odot k_t$ 向 $d_t$ 与 $k_t$ 分流梯度,再反向遍历预条件器递推。推理时也继续更新 $S_t,d_t$,这是前向计算内部的在线学习,不是额外对外层模型做一次 AdamW。[Sec.4.5, p.6]

2.6 APF:遗忘预条件器校准

APF 为每个 token/head 预测 $r_{t,h}=\sigma(w_{r,h}^\top x_t+b_{r,h})$,并使用

$$ d_{t+1}=\Pi_{\mathcal D}\left[r_{t,h}d_t+ \eta\beta_t\frac{1-\beta_t\langle d_t,s_t\rangle}{\bar n_t}s_t\right]. $$

保留门只衰减 $d_t$,不直接衰减 $S_t$。公式也不是重置回 $\mathbf1$;当衰减占主导时会向下移动,受投影下界约束。作者希望它应对话题变化造成的过时校准。它每层增加 $H(d_m+1)$ 个参数;340M 配置作者称不超过 0.3%。无 APF 的 vanilla OSDN 不增加可训练参数,但仍增加运行时状态。[Sec.4.4, p.6;App.A.5, p.16]

2.7 GDN/KDA 与 1.3B 配置差异

GDN 先得 $\bar S_{t-1}=\alpha_tS_{t-1}$,再用 $u_t=v_t-\bar S_{t-1}k_t$ 写入。这一“门控后残差”保持相同超梯度结构;用门控前残差会引入额外项。KDA 采用 $K\times V$ 状态,因此存储 key 出现在左侧:

$$ S_t=(I-\beta_t\tilde k_tk_t^\top)\operatorname{Diag}(\alpha_t)S_{t-1} +\beta_t\tilde k_tv_t^\top. $$

向量记忆门 $\alpha_t$ 和预条件器保留门 $r_t$ 是不同机制。[App.B, pp.17–19]

1.3B 并非逐字沿用主文递推。 App.B/C 指出它使用不含 $\beta_t$ 的 phase-1 更新,在归一化 key 下为

$$ d_{t+1}=\rho_t^dd_t+\eta(1-\langle d_t,k_t^2\rangle)k_t^2, $$

实际记忆写入仍含 $\beta_t$;还采用零初始化的可学习初始 scale 和约 0.999 初始化的数据依赖保留门。应把这些看作作者报告的实现配置;不能假定它直接满足主文 beta-aware surrogate 的理论。[App.B, p.17;App.C, p.20]

三:理论分析与实验设计

3.1 总体目标定理:理想 Newton 比较器

定义 $f(S)=\tfrac12\mathbb E\|Sk-v\|^2$、$\Sigma_k=\mathbb E[kk^\top]$、$L=\lambda_{\max}(\Sigma_k)$。Hessian 是常数 $\Sigma_k\otimes I_V$,所以 Hessian-Lipschitz 常数为零。右 Newton 比较器 $D_\star=\Sigma_k^\dagger$ 可以在可辨识子空间内一步到最优。

在总体全梯度更新、损失单调下降、对稠密 $D_\star$ 的累计 regret 不超过 $\mathcal R_T$ 的假设下:

$$ f(S_{T+1})-f^\star\le[f(S_1)-f^\star] \left(\frac{2L\mathcal R_T}{T}\right)^T. $$

证明路线是:用 Newton 一步最优性表达最优反馈;用 Rayleigh 商得到 $|h_t(D_\star)|^{-1}\le2L$;regret 控制各步误差比例之和;AM–GM 将平均比例转为乘积。$\mathcal R_T=o(T)$ 时括号趋零,得到超几何收敛。[Thm.5.1, p.6;Thm.D.5, pp.26–27]

条件不能省略:实际算法只搜索有界对角类,理论比较器一般稠密;实际训练流提供单 token 梯度而非总体梯度。App.D.3 把差距分解为在线学习 regret、box 约束差距、对角近似差距;第一项小并不自动使三项总和次线性。

3.2 token-local 定理:与实现更接近,但对象更窄

对单位 key,$q_t=1+2h_t(d_t)$。定义固定对角比较器误差

$$ \varepsilon_T(d)=\frac1{2T}\sum_{t=1}^T(1-\beta_t\langle d,s_t\rangle)^2, \qquad\varepsilon_T^{\rm diag}=\min_{d\in\mathcal D}\varepsilon_T(d). $$

若在此 surrogate 上有对固定比较器的 regret 界 $R_T$,则

$$ \prod_{t=1}^Tq_t\le\left(2\varepsilon_T^{\rm diag}+\frac{2R_T}{T}\right)^T. $$

推导只需 $\prod q_t\le(\frac1T\sum q_t)^T$、$q_t=1+2h_t$ 和 regret 定义。真正的超几何特例还要求同一个可行 $d^\star$ 对所有 token 满足 $\beta_t\langle d^\star,s_t\rangle=1$,使比较器误差为零,并且 $R_T=o(T)$;$R_T=O(\sqrt T)$ 时得到 $(O(1)/\sqrt T)^T$。比较器误差不为零时,不能无条件宣称超几何。[Thm.5.2, p.7;Thm.D.7, pp.28–29]

本次分析:可行性条件相当强。 对 box 上界 2,若某步 $\beta_t<0.5$,即使单位 key、所有坐标取 2,也有 $\beta_t\langle d,s_t\rangle<1$,该步不可能满足精确条件。固定 $\eta=0.003$ 的实际运行也没有获得专门的次线性 regret 证明;通常 OGD 的 $O(\sqrt T)$ 结果需要相应的步长选择。作者在 Discussion 中承认这一限制。APF 还需动态 regret 分析,现有定理不能直接覆盖。[p.9、pp.30–31]

3.3 为什么这不是记忆全局收敛定理

相邻 token 的 $f_t$ 不同,乘积不发生望远镜消去。原文反例是 $K=V=1$、$k_t=1$、$v_t=(-1)^t$:即使每次完全写对,$S_t$ 仍在正负 1 间振荡,不收敛到总体最优 0。[App.D.4, p.29]

Cor.D.8 在更强的“类别值固定、key 正交且支撑块互不重叠、预条件器尊重块结构”下,才把 token 比值乘积连接到各类别最终损失比的乘积。自然语言重复上下文并未被证明满足这些条件。它是机制直觉的支持,不是对 JRT 数据结构的已验证描述。[pp.29–30]

零残差还需谨慎:App.D 在 $u_t=0$ 时定义 $h_t=0,q_t=1$,但此时闭式 $(1-a_t)^2$ 一般不等于 1,且不依赖 $u_t$ 的极限也不一般为零。因此“连续扩展”这一表述与后续对所有 token 直接代入闭式的证明存在衔接问题。本笔记的比值推导限定在非退化 token;严格处理零残差应统一约定或分离这些步,不把 $0/0$ 当作普通观测。

3.4 实验设置与指标

实验协议与目的指标
340M 匹配训练FineWeb-Edu;20,480 步;每步 524,288 tokens,实际约 10.74B控制骨干内架构差异
JRT cloze2K context;FDA/SWDE/SQuAD 及各自 twicecontains accuracy,越高越好;不是严格 EM
机制 replay每个 twice 任务 16 prompts;fp32 回放每层每头$q_{\rm geo}=\exp(\operatorname{mean}\log q_t)$,越低越好
LMWikiText/LAMBADA两者 PPL 几何均值,越低越好
Common.PIQA/HellaSwag/WinoGrande/ARC-E/ARC-C/SIQA/BoolQ/LAMBADA8 任务零样本准确率平均
LongBench英文 14 任务原文平均分,越高越好;逐任务值未报告
PG-1920K blocks;每 2K 位置桶Final corpus PPL 与后段漂移分开
检索 LM-evalDROP、NQ-Open、TriviaQADROP F1;另两者空白归一化 EM
推理效率单 H100,batch=1,2048 prefill+128 greedy decode;预热2次、测5次中位数合计 tokens/s 与 decode ms

340M DeltaNet:24 层、width 1024、8 heads;KDA:23/1024/8;GDN:21/1024/6。因此最干净的是各骨干内部比较,不应把三者视作同一架构。AdamW 外层学习率 $10^{-3}$,cosine + warmup,裁剪 1.0,seed 42,bf16/fp32 reductions;4 GPU×累积2或8 GPU×累积1匹配每步 token 数。warmup 长度及部分其他优化器超参数在此文中未给出。[App.F, pp.32–34]

训练最大连续递归上下文依 App.F 为 4096;单 GPU 的 65,536 长度是 packing 后的张量,各文档边界重置状态,不能解释为 64K 长上下文训练。App.E 却写“训练 2K”,与 App.F 和 Fig.4 图注冲突;本笔记采用更具体的 App.F,并保留差异。

1.3B:24 层、width 2048、16 heads、100B tokens,只比较 DeltaNet 与 OSDN-APF。具体学习率、步数、batch、硬件数量、墙钟成本被指向训练日志,论文没有给出这些数值,不能套用 340M 配置。FW-Edu 使用训练集切片作 pseudo-validation,不能当成独立域外泛化证据。[App.E/F/K]

四:结果与证据解释

4.1 主结果:340M 的收益集中在受控召回

以下数值来自 Table 1(p.7);LM PPL 为 WikiText/LAMBADA GeoMean,JRT 均是比例值。

340M 模型JRT 总体 ↑单次 ↑重复 ↑LM PPL ↓LongBench ↑
DeltaNet0.1500.1550.14532.000.072
OSDN0.1980.1790.21831.670.087
OSDN-APF0.1760.1520.19930.990.073
GDN0.1540.1690.13930.010.073
OSGDN0.1820.1690.19529.780.073
OSGDN-APF0.2030.1850.22129.500.080
KDA0.1680.1870.15026.750.088
OSKDA0.1750.2180.13327.930.090
OSKDA-APF0.1850.1910.17928.020.098

OSDN 总体 +0.048 是 +4.8 个百分点、相对 +32%;重复 +0.073 是 +7.3 个百分点、相对约 +50.3%。APF 总体 +0.026,相对约 +17.3%,但单次 0.152 低于 baseline 0.155;所以 APF 没有在所有召回设置中优于 vanilla。

逐任务发现更有解释力:OSDN 的 SWDE-twice 从 0.158 升至 0.227,SQuAD-twice 从 0.231 升至 0.383,FDA-twice 却为 0.045→0.044。平均收益并非三个任务普遍提高。[Table 7, p.34]

扩展到 GDN/KDA 的结果是混合的:OSGDN-APF 重复召回相对 GDN 提高约 59%;OSKDA 无 APF 时单次改善而重复下降,两个 OSKDA 变体的 LM PPL 都比 KDA 差。这支持“可组合的机制”,不足以支持“无代价的通用改进”。

4.2 机制诊断:比值下降确实存在

模型$q_{\rm geo}$ ↓$q_{\rm arith}$ ↓相对本尺度 baseline 的 $q_{\rm geo}$ 下降
DeltaNet 340M0.5370.636—
OSDN 340M0.4330.56619.4%
OSDN-APF 340M0.4250.57320.9%
DeltaNet 1.3B0.4320.546—
OSDN-APF 1.3B0.2650.48438.7%

来源:Table 8, p.35。340M 覆盖 48 prompts、24 层×8 头,约 $7.85\times10^6$ token-layer-head 测量;这些测量共享 prompt 和 checkpoint,不能当成数百万个独立实验样本。1.3B 头数变为 16,不能沿用 340M 测量总数。

原图2:残差比的相对位置曲线与总体均值

原图 Fig.2, p.9,来自作者矢量图。左图横轴为 1–63 相对位置桶,纵轴为每桶残差比几何均值;竖虚线分隔第一次和重复上下文。右图为总体几何均值。OSDN/APF 曲线低于 DeltaNet,且在重复段仍有明显优势。不能将左图解释成 OSDN 自己的重复段比第一次段整体更低:重复边界后曲线也有上升。图支持相对 baseline 的优势,不证明每个位置或每个 token 都改善。

APF 的几何均值略优于 vanilla,但算术均值更差(0.573 vs 0.566),说明聚合口径会影响排序。$q_t$ 衡量当步自回读的局部改善;它不直接测量远处 query 最后能否答对。

4.3 1.3B:更强的机制信号,接近持平的任务结果

指标DeltaNetOSDN-APF解读
WikiText PPL ↓17.2318.42变差
LAMBADA PPL ↓11.8410.98改善
两者 GeoMean ↓14.2814.22数值接近
FW-Edu pseudo-val PPL ↓8.719.10变差
FDA/SWDE 单次 ↑0.2930.315+2.2 个百分点
FDA/SWDE 重复 ↑0.2270.217−1.0 个百分点
四项总体 ↑0.2600.266+0.6 个百分点
Common. ↑0.5600.566小幅差异
LongBench ↑0.1150.116小幅差异

来源:Tables 15–18, p.40。重复任务分项中 FDA-twice 0.061→0.073,SWDE-twice 0.392→0.360。SQuAD/SQuAD-twice 的 contains accuracy 因 harness 失效被排除;但内部残差诊断仍报告了 SQuAD-twice 的 0.473→0.102。这是两种测量,不能拿后者补齐缺失的答题准确率。

340M 总体召回平均 6 项,1.3B 平均 4 项,不能直接把二者的总体平均差解释成纯 scale 效应。1.3B 也没有完成匹配的 PG-19 20K sweep。作者称重复召回变化“在噪声内”,但没有提供足以检验这一说法的多 seed 置信区间;本笔记仅报告点估计和近似持平的描述,不声称统计等效。

4.4 PG-19:绝对 PPL 与漂移量必须分开

原图4:PG-19位置桶困惑度

Fig.4/Table 13, p.38。横轴为 20K 测试块中的 2K 位置桶,纵轴为 PPL;颜色对应骨干和变体,红色背景标记后段。Final 是全语料评估输出,不是桶 PPL 的简单平均。漂移量定义为末桶 PPL 减去 2–4K 桶 PPL。

DeltaNet 家族Final PPL ↓2–4K18–20K后段漂移 ↓
DeltaNet20.7820.1021.79+1.69
OSDN20.0219.5720.76+1.19
OSDN-APF19.8519.1021.23+2.13

APF 的绝对 Final 最好,但晚段漂移大于 vanilla 和 baseline。不能概括为“APF 在所有长期稳定性指标上最佳”。GDN→OSGDN 的 Final 为 20.11→19.70,OSGDN-APF 为 20.21;KDA→OSKDA 为 18.73→18.53,OSKDA-APF 为 19.00。这些分组也显示 APF 存在权衡。

4.5 初值消融与训练 CE

Table 14(p.38)比较固定初值 OSDN、学习 $D_0$、放松初值投影的 $D_0^\star$ 及 APF。学习 $D_0$ 后,召回为 0.172、PG-19 Final 为 21.57;$D_0^\star$ 为 0.164/21.42;均不如 vanilla 的 0.198/20.02。它们的 Common. 点估计略高,但不足以抵消召回和长度漂移上的代价。原文未报告每种超参数、多 seed 的完整网格,因此不把消融视为所有初始化策略的结论。

原图5:同预算训练交叉熵

Fig.5, p.39。横轴为训练 tokens;全程面板使用对数纵轴,其余三个面板放大最后 25% 训练、使用线性纵轴。曲线是 257 步居中滚动均值,最终 CE 取最后 256 步平均;全部完成配置在 2.421–2.479 nats 内,相差 0.058。DeltaNet/OSDN/APF 分别为 2.475/2.479/2.459。此图不能证明“在线预条件提升训练优化速度”,也不能把语言模型 CE 与定理中的内层回归损失混同。

4.6 通用任务与开放域检索

340M 的 Common. 多数仅在小范围变化;特别是 Table 9 的图注说明,vanilla OSDN 采用各任务已完成 no-APF cell 的最佳值后再平均,不宜把这行当成严格单一 checkpoint 的整套评测。[p.35]

开放域检索在这些未指令微调的 340M 模型上分数很低。Table 12(p.37)中 DeltaNet/OSDN/APF 的 DROP、NQ、TriviaQA 三项平均均四舍五入为 0.016;KDA 是 0.018,两个 OSKDA 变体是 0.014。受控 cloze 的优势没有普遍转化成开放域 QA 收益。FW-Edu screen(Table 11)提供域内拟合对照:OSDN 12.32 vs DeltaNet 12.43,KDA 11.38;该切片来自训练 split,不能把它当成独立测试集。

4.7 吞吐与内存:低开销成立,百分比有算术问题

模型合计 tokens/s ↑相对 baseline常驻状态 MiB(bf16计数)
DeltaNet 340M784.5—6.000
OSDN 340M782.7−0.2%6.047
OSDN-APF 340M767.2−2.2%6.047
DeltaNet 1.3B794.0—12.000
OSDN-APF 1.3B739.9−6.8%12.094

Tables 19–20, p.41。指标按 $(2048+128)/(t_{prefill}+t_{decode})$ 计算,不能称为纯 decode tokens/s。作者报告 340M OS-* 相对宿主约 −2.2% 至 +5.5%,但没有给出计时误差条,也未覆盖长 prompt、大 batch、训练吞吐等全部场景。

本次复核发现:原文多处“新增状态 ≤0.05%”与表格不符。 340M DeltaNet 的 $(6.047-6.000)/6.000\approx0.783\%$;1.3B 同样约 0.783%;GDN 的 0.031/7.875 约 0.394%。公式 $VK\to VK+K$ 给出相对增量 $1/V$,也支持约 0.4%–0.8% 的量级。作者似乎混淆了很小的 MiB 增量与百分比;不能直接采用原文的 ≤0.05%。这不推翻“增量很小”,但应修正定量表述。此计数排除权重、激活、allocator、短卷积 cache,并按 bf16 元素计量,不等同于整卡实测显存。[App.A.6/L, pp.17、41–42]

内核一致性也是作者报告,非本次复测:OSGDN 在合成输入上与 chunk reference 的输出相对误差 ≤$6.5\times10^{-3}$、最终状态 ≤$6.7\times10^{-3}$、最终 $d$ ≤$2.0\times10^{-3}$;128-token prefill+1-token decode 对照误差 ≤$5.8\times10^{-3}$。[p.42]

五:结论、限制与后续问题

5.1 可以可靠带走什么

本文最有价值的是一个具体的代数与系统结合点:对角右预条件可写成 write-key 缩放,而归一化 token-local 超梯度可消去高维记忆残差。 因而用很小的在线状态就能改变 DeltaNet 写入几何,同时沿用大部分 chunkwise 矩阵结构。340M 的受控召回与多尺度局部残差实验为这一机制提供了正面证据。

理论价值在于区分了理想总体优化和实际 token-local 优化,但其超几何结论有严格前提。实践价值也有明确边界:APF、不同宿主和更大规模有收益与损失并存,没有证明全面超过原骨干,更没有证明最终语言模型交叉熵超几何收敛。

5.2 阅读中需保留的原文差异

差异或限制原文位置本笔记采用的处理
新增状态“≤0.05%”与 MiB 表格不符pp.6/9/41–42采用表格数值及 $1/V$ 推算,约0.4%–0.8%
训练2K与实际至多4KApp.E p.32 vs App.F p.32、Fig.4 p.38采用详细 packing/reset 协议的4K上限
1.3B phase-1 非 beta-awareApp.B p.17、App.C p.20与主文理论实现分开说明
OSKDA 固定保留0.999与最终 no-retentionApp.C p.23 vs App.F p.32以明确标注 refreshed final 的 App.F 描述实验,复现需核对配置
OSDN-APF Common. 0.454/0.456Table 14 p.38 vs Tables 1/9 pp.7/35主结果采用0.456,消融差异不静默合并
Fig.1写 key使用 $d_{t+1}$,算法先输出 $d_tk_t$Fig.1 p.2 vs Alg.1 p.5、Listing 1 p.21实现顺序依据公式与算法;不复制该示意图作为正确伪代码
零残差扩展与闭式/比较器证明不完全衔接App.D pp.28–29非退化情形推导明确成立;零残差另行定义
“噪声内/持平”缺少显著性支持Sec.6.3、App.K仅作为点估计描述,不宣称统计等效

匹配 seed 与数据顺序有利于减少混杂,但单 seed 不消除随机性;已有结果没有 seed bootstrap 置信区间。主文“相同 protocol”也不足以补齐未给出的 1.3B 训练超参数。本笔记指出这些问题是为了读准证据,并未据此声称整篇论文结论无效。

5.3 适合组会讨论或后续复现的问题

  1. 在完全相同的 1.3B 配置中,分别比较 beta-aware / non-beta-aware、vanilla / APF,能否把 scale 效应与变体变化分开?
  2. 更低的 $q_{\rm geo}$ 为何没有改善重复召回?应同时测早先关联的保留、跨 key 干扰、最终 query 读出和校准分布。
  3. 与固定逐头/逐通道 scale、仅调 $\beta_t$ 的 baseline 比较后,在线更新本身贡献多少?现有初值消融不足以完全回答这一问题。
  4. 给 APF 建立动态比较器 regret,并在实际固定步长、投影、零残差约定下补齐保证;再测试对角比较器误差的实际大小。
  5. 在多随机种子、更多重复 prompts、长上下文和更大 batch 下重复召回与效率测试,报告误差范围及训练/推理各阶段成本。

成品与复核记录

  • 英文组会幻灯片 PDF / 可编辑 Beamer TeX,按约25分钟主讲组织,附备份推导。
  • 图件直接复用作者源码中的 Fig.2/4/5,PNG 用于笔记,PDF 用于幻灯片,未用生成式图片替代实验结果。
  • 构建工具(本地核验附件) 使用本地 pdflatex;编译及逐页视觉检查结果见 quality-check.md。
  • 研读为原文分析,不是逐段翻译或独立实验复现。