Latent Sparse Attention paper: indexing eats 98% of prefill at 10M tokens, LSA speeds one layer 22.7x

burny_tech · x · 2026-11-08

Prefrontal AI (with Torsten Hoefler and James Whittington among the authors) published Latent Sparse Attention (LSA), targeting attention cost at ultra-long context (1M+ tokens).

Problem: sparse attention saves compute, but finding the relevant tokens itself becomes extremely expensive at ultra-long context — sometimes 98% of attention's prefill time. Existing indexing is either still quadratic or loses accuracy at that scale.

Method: LSA hierarchically clusters keys in a learned latent space so queries score a hierarchy of clusters instead of every key, speeding up prefill, decode and training. Two variants:

Results: on GLM-5.3-FLASH (QWEN-3.8-27B) at 10M tokens, D-LSA speeds one attention layer 22.7x in prefill (72.4x vs full attention) and 4.6x in decode (42.5x vs full) with similar accuracy; N-LSA matches full-attention accuracy with a 1.6x per-layer training speedup at 128k tokens.

Related event: Open-Sourced Latent Sparse Attention Speeds Up Long Context 22.7x(2 posts)→

Original post →

More from Infra

Infra channel →