arXiv Machine Learning

Beyond Deduplication: Sharp Retrieval Limits for Attention with Correlated Distractors

The paper studies how redundant context influences attention mechanisms, distinguishing between normalization effects from repeated items and score variability within correlated groups. Using a planted retrieval model with exponentially many Gaussian distractors organized into large clusters, the authors derive exact retrieval thresholds via a two‑level generalized random energy model and Gaussian interpolation. They show that subtracting the log cluster size from pre‑softmax scores removes multiplicity but fails to eliminate extreme within‑cluster scores, establishing a sharp limit on signal strength that cannot be overcome by temperature scaling alone; averaging scores before exponentiation mitigates this effect but only when the relevant item is a singleton.

arXiv AI
Oct 21

Stratified Sparse Attention: Reducing the Sparse-to-Dense Gap at Extreme Sparsity

The paper introduces Stratified Sparse Attention (SSA), a framework that blends deterministic top‑k retrieval with score‑guided sampling to reduce the gap between sparse and dense attention in large language models. SSA attends to high‑scoring tokens deterministically while sampling the remaining attention mass using proxy scores, thereby recovering more of the attention distribution. Experiments on benchmarks such as RULER, LongBench, LOFT, and AIME show that SSA outperforms existing sparse methods across models ranging from 3B to 27B parameters, achieving significant accuracy gains and up to 2.96× decode speedup with optimized CUDA kernels.

By Sahil Joshi, Prithvi Dixit, Aryan Katiyar, Agniva Chowdhury, Joseph E. Gonzalez, Anshumali Shrivastava, Aditya Desai, Kumar Krishna Agrawal
arXiv Machine Learning
Oct 8

Attention-Mass Condensation for Sparse Decoding

arXiv:2602.06317v3 Announce Type: replace Abstract: Attention-mass concentration creates an opportunity for sparse decoding, but retained mass alone does not guarantee a stable greedy decision: retri...

By Jorge L. Ruiz Williams
arXiv AI
Jul 14

Remembering Distinct Items, Not Tokens: A Learnable Dirichlet-Process Cache Between State-Space Models and Attention

arXiv:2607. 09889v1 Announce Type: cross Abstract: Fixed-state sequence models compress an unbounded past into a bounded state, which caps their associative recall at roughly the state dimension; attention escapes the cap by keeping a key-value entry for every token, at quadratic compute and a cache that grows with the sequence.

By Siddharth Pal, Viktoria Rojkova
arXiv AI
Sep 12

Relevance Is Not Permission: Localizing and Controlling Metric-Facing Attention Contributions

The paper introduces Warrant, a method that locates and controls the contributions of attention mechanisms to model metrics. Warrant exposes the item‑wise contribution path to the reported metric and applies query‑conditioned permission on that path. Experiments on multiple datasets show that Warrant improves primary metrics in most comparisons, reveals a weak correlation between attention and prediction utility, and demonstrates that learned permission can recover evidence ranking while suppressing distractors.

By Minwoo Yu, Young-guk Ha