跳到正文
/阅读与研究

外观

快速抵达

↑ ↓ 选择 · Enter 打开 · Esc 关闭MyHOT
Hugging Face · 每日论文· 媒体 / 作者来源·精选

First Passage Attention:用路径极值替代累积衰减,实现选择性遗忘与长度外推

原文标题

First Passage Attention

AI 导读

论文提出 First Passage Attention(FPA),将遗忘门从沿键-查询路径累加衰减改为路径极值:根据学习到的标量场在区间内的最深下探,按比例降低键的 logit,从而选择性擦除中间上下文而保留更早的相关 token。FPA 将 Forgetting Transformer(FoX)作为特例;在合成实验中,单层 FPA 头能收敛到目标键集合,而 softmax 和 FoX 基线在每个随机种子上都失败;在 1.1 亿参数语言模型上,FPA 困惑度与 FoX 持平,并在 8 倍训练长度下唯一实现完美 passkey 检索,键值缓存压缩至约 25% 时仍有效。

来源:Hugging Face · 每日论文 · arxiv.org

导读与选稿说明

提出用路径极值替代路径求和的注意力遗忘机制,在合成任务和 1.1 亿参数语言模型上验证了选择性遗忘与训练长度 8 倍的 passkey 检索能力,适用于长上下文研究。

导读由模型辅助整理,非独立核验。完整信息见原文。