arXiv:2610.42147v1 Announce Type: new
Abstract: Agentic workflows are increasingly shifting towards long-horizon tasks, such as open-ended scientific work, where models benefit from long context windows. However, at ultra-long context (above 1M tokens), full attention is prohibitively expensive. A promising approach is sparse attention, in which a dedicated indexer selects a subset of keys for the attention. Yet current methods of indexing either still have quadratic cost in context size or lose accuracy at ultra-long context scale. We address this with Latent Sparse Attention (LSA), an efficient attention method that maintains accuracy at ultra-long context. In LSA, keys are hierarchically clustered in a learned latent space, and queries score the hierarchy of clusters instead of scoring every key, making prefill, decode and training faster at long context. We propose two variants, Distilled-LSA (D-LSA), which can be applied to pre-trained models via distillation, and Native-LSA (N-LSA), which is trained natively on the next-token loss alone. On GLM-5.3-FLASH (QWEN-3.8-27B) at 10M tokens, D-LSA speeds up one attention layer by 22.7x (72.4x) in prefill and 4.6x (42.5x) in decode over its original sparse attention (full attention), while keeping similar accuracy across a variety of tasks. N-LSA matches the accuracy of full attention with a training speedup of 1.6x per layer at 128k tokens. LSA offers practical and efficient attention for models to benefit from ultra-long context.
arxiv
Latent Sparse Attention: Subquadratic Attention for Ultra-Long Contexts
Latent Sparse Attention (LSA) is a new attention method that clusters keys hierarchically in a learned latent space, allowing queries to score clusters rather than individual keys. This reduces computational cost from quadratic to subquadratic, enabling efficient prefill, decode, and training at contexts above 1M tokens. Two variants are proposed: Distilled-LSA for adapting pre-trained models and Native-LSA trained from scratch. On GLM-5.3-FLASH at 10M tokens, D-LSA achieves up to 22.7x prefill and 4.6x decode speedups over sparse attention while maintaining accuracy.
This paper introduces a practical solution to the growing need for efficient ultra-long context processing in agentic AI, with significant speedups and accuracy retention.
Comments & Questions (0)
No comments yet
Be the first to comment!