English

Prism: Spectral-Aware Block-Sparse Attention

Computation and Language 2026-05-26 v2 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Block-sparse attention is promising for accelerating long-context LLM pre-filling, yet identifying relevant blocks efficiently remains a bottleneck. Existing methods typically employ coarse-grained attention as a proxy for block importance estimation, but often resort to expensive token-level searching or scoring, resulting in significant selection overhead. In this work, we trace the inaccuracy of standard coarse-grained attention via mean pooling to a theoretical root cause: the interaction between mean pooling and Rotary Positional Embeddings (RoPE). We prove that mean pooling acts as a low-pass filter that induces destructive interference in high-frequency dimensions, effectively creating a "blind spot" for local positional information (e.g., slash patterns). To address this, we introduce Prism, a training-free spectral-aware approach that decomposes block selection into high-frequency and low-frequency branches. By applying energy-based temperature calibration, Prism restores the attenuated positional signals directly from pooled representations, enabling block importance estimation using purely block-level operations, thereby improving efficiency. Extensive evaluations confirm that Prism maintains accuracy parity with full attention while delivering up to 5.1×\mathbf{5.1\times} speedup.

Keywords

Cite

@article{arxiv.2602.08426,
  title  = {Prism: Spectral-Aware Block-Sparse Attention},
  author = {Xinghao Wang and Pengyu Wang and Xiaoran Liu and Fangxu Liu and Jason Chu and Kai Song and Xipeng Qiu},
  journal= {arXiv preprint arXiv:2602.08426},
  year   = {2026}
}

Comments

ICML 2026

R2 v1 2026-07-01T10:27:33.100Z