Good Papers

DoSA: Dots Sparse Attention

DoSA skips full key scoring via certificates that preserve exact top-k selection, accelerating sparse attention indexing up to 2.74x.

Fanxu Meng, Taosong Fang

Published Sep 26, 2026▲ 6 on Hugging FaceCode ★ 5arXiv ↗

88%
OverallMust read
?
OverallMust readVote to see the score
Readers
–

Only vote on papers you've read. Sign in to vote.

AI panel15/20reviewers recommend it
lenient 3/5
medium 9/10
strict 3/5
AI panel?Vote to see what the 20 AI reviewers said

Abstract

Sparse attention visits few keys, yet finding them can still require scoring every key with every indexer head. We introduce DoSA (Dots Sparse Attention), a lossless method for making this computation conditional while preserving exact token selection. Its central observation is that Top-K selection requires sufficient evidence to reject a candidate, not necessarily its full score. A seed set supplies a lower bound on the winning threshold. Nonnegative head projections and joint residual bounds upper-bound positive contributions, while activation-informed aggregation bounds the magnitude subtracted by negative heads. The certificate directs full scoring toward candidates that remain eligible; exact-score filtering further reduces writeback and selection work. We establish the certificate and its exact-selection conditions, and analyze how geometry and selection margins govern its effectiveness. Our GPU implementation accelerates the complete Indexer by up to 2.74 times over SGLang and 1.69 times over LiteTopK across English, Chinese, and code workloads. End-to-end evaluation on GLM-5.2 and DeepSeek-V3.2 achieves up to 1.94 times and 1.50 times time-to-first-token speedups over the two baselines, respectively.