Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

You Only Index Once: Cross-Layer Sparse Attention with Shared Routing

About

Long-context inference in modern LLMs is increasingly constrained by decoding efficiency, especially in reasoning-heavy settings where models generate long intermediate chains of thought. Existing sparse attention methods often face a practical efficiency-quality trade-off. Structured block sparse methods typically provide stronger acceleration but incur noticeable quality loss, while token sparse methods are usually more accurate yet deliver limited end-to-end speedup because top-k routing over the full cache remains expensive. In this work, we propose cross-layer sparse attention (CLSA), which is built on top of KV-sharing architectures such as YOCO. The core idea is to share not only the KV cache across cross-decoder layers, but also the routing index. A single indexer computes token-level top-k selection once and reuses the resulting index across layers, thereby preserving the fine-grained selectivity of token sparse attention while amortizing the routing overhead. The resulting architecture improves all major inference bottlenecks jointly, including pre-filling, KV-cache storage, and long-context decoding. Experiments across short-context and long-context benchmarks show that CLSA is both accurate and efficient, achieving up to 7.6x decoding speedup and 17.1x overall throughput improvement at 128K context. These results suggest a more complete architectural solution for long-context LLMs that jointly advances model quality and inference efficiency.

Yutao Sun, Yanqi Zhang, Li Dong, Jianyong Wang, Furu Wei• 2026

Related benchmarks

TaskDatasetResultRank
ReasoningBBH
Accuracy41.8
770
Commonsense Question AnsweringWinoGrande
Accuracy63
89
Grade School Math Word ProblemsGSM8K
Accuracy0.47
76
ReasoningARC Challenge
Accuracy (ARC)0.465
56
ReasoningDROP--
42
Commonsense QAHellaSwag
Accuracy67.6
19
Decode Throughput GenerationNVIDIA B200 GPU
Throughput (tokens/s)6.74e+3
15
End-to-end Text GenerationNVIDIA B200 GPU Overall Generation
Throughput (tokens/s)4.92e+3
15
Prefill ThroughputSynthetic
Prefill Throughput (tokens/s)2.09e+4
15
Long-context retrieval and reasoningRULER 16k context
S1 Accuracy100
10
Showing 10 of 13 rows

Other info

Follow for update