HISA: Efficient Hierarchical Indexing for Fine-Grained Sparse Attention
Token 级稀疏注意力(如 DeepSeek Sparse Attention)通过轻量级索引器为每个 query 对所有历史 key 评分并选择子集来实现细粒度的 key 筛选。但索引器本身仍需扫描全部历史 key,成为性能瓶颈。HISA 提出层次化索引结构,将 key 组织为多层级索引,使索引器的计算复杂度从 O(n) 降低到 O(√n) 级别,在长序列场景下显著加速稀疏注意力的 key 选择过程。
INSIGHTHISA 将信息检索中的层次化索引思想引入注意力机制,与搜索系统中的倒排索引分层检索高度类似。对搜索推荐系统而言,这种层次化稀疏注意力可直接用于长文档理解和大规模候选集的高效注意力计算。该方法解决了 DeepSeek Sparse Attention 等方案中索引器随序列长度线性增长的瓶颈,对需要处理长上下文的搜索场景具有直接的工程参考价值。