本文提出一种从旋转位置编码(RoPE)导数出发、精确计算注意力干预局部响应的方法。作者以 RoPE 导数 ∂p z(p)=Az(p) 为起点,通过积分得到有限位置位移,并在不线性化旋转与归一化的前提下贯穿 softmax 推导,从而仅凭缓存的基线状态和一次反向传播即可对候选编辑打分。在 768 个留出提示集、92,160 次位置编辑上,该方法符号准确率达 95.36%–96.52%,相比位置雅可比矩阵将答案边际 MAE 降低 73.6%–82.5%,相比零基线降低 36.2%–50.9%。针对键值同时编辑,作者用同一套差商运算分离出交互项,保留该项后下游边际 MAE 在 5,120 次干预扫描中降低逾九倍。该工作为注意力编辑的快速评分与缓存决策提供了精确且可验证的理论工具。
| RoPE | 旋转位置编码,一种通过旋转向量来编码序列位置信息的方法,常用于 Transformer 模型。 |
| softmax | 归一化指数函数,将向量转换为概率分布,在注意力机制中用于计算注意力权重。 |
| MAE | 平均绝对误差,衡量预测值与真实值之间平均绝对差异的指标。 |
| KL certificate | KL 证书,基于 Kullback-Leibler 散度量化局部近似误差的保证或界限。 |
| query-conditioned gradient-step | 查询条件梯度步,一种依赖于查询向量的梯度更新表示,用于分析注意力机制中的方向。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅