← 返回时间线

paper

First Passage Attention

arXiv ↗
ID
2610.22714
分类
—
首次捕获
2026-10-21
状态
unread
作者
Yuxuan Liu
信号
🔥 0

信号历史

  • 2026-10-21HF Daily Papers · 🔥0
暂无信号数据

摘要

Augmenting softmax attention with forget gates typically relies on summing decay contributions along the key--query path. Because such accumulation is monotonic, these mechanisms cannot selectively erase intermediate context while preserving older, structurally relevant tokens. We propose First Passage Attention (FPA), which shifts from path summation to path extrema by lowering a key's logit proportionally to the deepest dip of a learned scalar field below that key's level over the intervening span. FPA encompasses the Forgetting Transformer (FoX) as a special case, and when the field and levels follow a shared learned walk, the alive keys with unpenalised logits naturally form a stack. We show that a constant-dimension FPA head realises any subset of keys left alive by subsequent dips, whereas softmax heads with positional biases or token-level forget gates require a head dimension linear in the number of keys. In synthetic experiments, one-layer FPA heads converge to these target sets while comparable softmax and FoX baselines fail on every seed. Scaling to a 110M-parameter language model, FPA matches FoX in perplexity and uniquely achieves perfect passkey retrieval at 8times the training length, even with its key--value cache compressed to about 25%.

我的笔记

还没有笔记。

在 GitHub 上写笔记 ↗(新建 content/notes/2610.22714.md,PR 合并后本页自动更新)