arXiv:2610.22985v1 Announce Type: cross Abstract: Long-context decoding in large language models (LLMs) is increasingly bottlenecked by the memory and computation required to attend over an ever-growing Key-Value (KV) cache at each autoregressive step. Although token-level sparse attention reduces this burden by restricting computation to a small subset of relevant past tokens, its practical gains are often offset by repeatedly recomputing layer-wise sparse indices as the context grows. A common remedy is to share sparse indices across layers by computing them only at designated anchor layers, but such static reuse ignores the layer-wise evolution of attention patterns, causing the reused indices to progressively deviate from the layer-specific optima and incur substantial attention loss. By revisiting cross-layer sparsity from the perspective of attention dynamics, we find that inter-layer salient token shift is highly localized: although the exact top-$\kappa$ indices differ across layers, the core high-attention tokens remain stable, with changes concentrated near the expanded boundary of previously selected regions. Motivated by this, we propose Prism Attention, a proposal-refined sparse attention framework. It replaces rigid cross-layer index reuse with a coarse-to-fine strategy, preserving layer-wise adaptability while retaining the efficiency benefits of cross-layer sharing. Extensive experiments show that Prism Attention achieves state-of-the-art performance among mainstream sparse attention methods on both reasoning and long-context benchmarks, while attaining a $3.4\times$ end-to-end decoding speedup over FlashAttention at 128K context length.
Prism Attention: Proposal-Refined Index Sharing Mechanism for Efficient LLMs Inference
Researchers posting arXiv paper 2610.22985v1 introduced Prism Attention, a proposal-refined sparse attention framework that achieves a 3.4x end-to-end decoding speedup over FlashAttention at 128K context length. Prism Attention replaces rigid cross-layer reuse of sparse indices with a coarse-to-fine strategy that preserves layer-wise adaptability, after the authors found that inter-layer salient token shift is highly localized, with core high-attention tokens staying stable and changes concentrated near the expanded boundary of previously selected regions. The authors report state-of-the-art performance among mainstream sparse attention methods on both reasoning and long-context benchmarks.
Run your AI side-project on zahid.host
EU-based hosting, git-push deploys, automatic HTTPS, no cold starts. Free tier with a custom domain — perfect for shipping the agent you just read about.