paper.dinq.meAI 与前沿科学顶会 · 学术新星雷达
登录 DINQ
返回新星榜
DY

di yin

ICLR 2026×3COLM 2026推理智能体与工具多模态效率与系统可解释性
新星分贡献者
56
一作贡献0
核心/参与20
方向广度13
合作网络12
0 一作 · 4 篇 · ICLR 2026 / COLM 2026

获取完整研究档案

首次获取该候选人的完整档案及 StartLux 评估共 20 积分;每位用户单独解锁,之后查看免费。主动刷新 20 积分。

联系方式

实时查找电子邮箱,费用不超过20积分;未找到邮箱不扣费。

2026 顶会论文

4
作者位次 #12/14COLM 2026Poster· Poster Session 1· Imperial Ballroom· #22

HISA: Efficient Hierarchical Indexing for Fine-Grained Sparse Attention

Yufei Xu · Fanxu Meng · Fan Jiang · Yuxuan Wang · Ruijie Zhou · Zhaohui Wang · Jiexi Wu · Zhixin Pan · Xiaojuan Tang · Wenjie Pei · Tongxuan Liu · di yin · Xing Sun · Muhan Zhang

This work replaces the DSA indexer in DeepSeek-v3.2 and GLM-5 with the more efficient HISA indexer, which preserves fine-grained indexing, delivers a 3.75× speedup, requires no training, and maintains downstream task performance.

摘要

Token-level sparse attention mechanisms, exemplified by DeepSeek Sparse Attention (DSA), achieve fine-grained key selection by scoring every historical key for each query through a lightweight indexer, then computing attention only on the selected subset. While the downstream sparse attention itself scales favorably, the indexer must still scan the entire prefix for every query, introducing an $\mathcal{O}(L^2)$ per-layer bottleneck that grows prohibitively with context length. We propose HISA (Hierarchical Indexed Sparse Attention), a plug-and-play replacement for the indexer that rewrites the search path from a flat token scan into a two-stage hierarchical procedure: (1) a block-level coarse filtering stage that scores pooled block representations to discard irrelevant regions, followed by (2) a token-level refinement stage that applies the original indexer exclusively within the retained candidate blocks. HISA preserves the token-level top-$k$ index interface consumed by the downstream Sparse MLA operator and requires no additional training. In end-to-end serving with DeepSeek-V3.2, HISA improves throughput by $1.65\times$ and time to first token by $1.70\times$ over the official DSA implementation, despite the HISA-specific operators being implemented in TileLang/Triton while the official DSA baseline uses highly optimized hand-tuned CUDA kernels. With matched TileLang implementations of both indexers, controlled microbenchmarks show up to $3.75\times$ indexer speedup at 64K context. On Needle-in-a-Haystack and LongBench, we directly replace the indexer in DeepSeek-V3.2 and GLM-5 with our HISA indexer, without any finetuning. HISA closely matches the original DSA in quality, while substantially outperforming block-sparse baselines.

效率与系统可解释性推理LLMAttentionSparse AttentionKV CacheOpenReview

常见合作者