棱镜:一种基于频谱感知的块稀疏注意力机制 / Prism: Spectral-Aware Block-Sparse Attention
1️⃣ 一句话总结
这篇论文提出了一种名为Prism的新方法,它通过分析注意力信号中的频率成分,巧妙地解决了现有块稀疏注意力机制在识别重要信息块时效率低、精度差的问题,从而在不损失模型准确性的前提下,大幅提升了长文本处理的速度。
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 $\mathbf{5.1\times}$ speedup.
棱镜:一种基于频谱感知的块稀疏注意力机制 / Prism: Spectral-Aware Block-Sparse Attention
这篇论文提出了一种名为Prism的新方法,它通过分析注意力信号中的频率成分,巧妙地解决了现有块稀疏注意力机制在识别重要信息块时效率低、精度差的问题,从而在不损失模型准确性的前提下,大幅提升了长文本处理的速度。
源自 arXiv: 2602.08426