mlsanityData to 14 Oct 2026

Guess-Verify-Refine: Data-Aware Top-K for Sparse-Attention Decoding on Blackwell via Temporal Correlation

arXiv 2604.22312 ↗First posted 24 Apr 2026cs.DC cs.AR cs.PFcited by 3

On the map: Efficient LLM inference and serving → KV cache optimization

Abstract (arXiv)

Sparse-attention decoders rely on exact Top-K selection to choose the most important key-value entries for each query token. In long-context LLM serving, this Top-K stage runs once per decode query and becomes a meaningful latency bottleneck even when the indexer and attention kernels are already highly optimized. We present Guess-Verify-Refine (GVR), a data-aware exact Top-K algorithm for sparse-attention decoding on NVIDIA Blackwell. GVR exploits temporal correlation across consecutive decode steps: it uses the previous step's Top-K as a prediction signal, computes pre-indexed statistics, narrows to a valid threshold by secant-style counting in 1-2 global passes, verifies candidates with a ballot-free collector, and finishes exact selection in shared memory. We connect this behavior to the Toeplitz / RoPE structure of DeepSeek Sparse Attention (DSA) indexer scores and validate the design on real DeepSeek-V3.2 workloads integrated into TensorRT-LLM. GVR achieves an average 1.88x single-operator speedup over the production radix-select kernel, with up to 2.42x per layer per step, while preserving bit-exact Top-K outputs. In controlled TEP8 min-latency deployment, it improves end-to-end TPOT by up to 7.52% at 100K context, with larger gains at longer contexts and smaller but still positive gains under speculative decoding. While implemented and validated in the current TensorRT-LLM DSA stack on Blackwell, the same principle may extend to sparse-attention decoders whose decode-phase Top-K exhibits temporal stability.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in KV cache optimization, one of 12 directions in Efficient LLM inference and serving. Builds on 13 of its 21 references that are in the map. First posted on arXiv 24 Apr 2026.

Builds on

13 of 21 references.

Closest ancestors · 2

Same direction, newest first

  1. 2025RocketKV: Accelerating Long-Context LLM Inference via Two-Stage KV Cache Compression19 Feb
  2. 2024Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference16 Jun

Other references · 11

  1. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  2. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  3. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  4. 2023SWE-bench: Can Language Models Resolve Real-World GitHub Issues?cited by 3,373
  5. 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
  6. 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,417
  7. 2023YaRN: Efficient Context Window Extension of Large Language Modelscited by 601
  8. 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 407
  9. 2026HISA: Efficient Hierarchical Indexing for Fine-Grained Sparse Attentioncited by 12
  10. 2025RadiK: Scalable and Optimized GPU-Parallel Radix Top-K Selectioncited by 6
  11. 2024Approximate Top-k for Increased Parallelismcited by 5

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026HiSparse: Scaling Sparse-Attention Decoding with Hierarchical KV Cache Management
  2. 2026You Only Index Once: Cross-Layer Sparse Attention with Shared Routing
  3. 2026Stochastic Sparse Attention for Memory-Bound Inference
  4. 2026FlashMemory-DeepSeek-V4: Lightning Index Ultra-Long Context via Lookahead Sparse Attention
  5. 2026EntmaxKV: Support-Aware Decoding for Entmax Attention
  6. 2026Elastic Threshold Attention: Learned Contextual Sparsity for Long-Context Decoding
  7. 2026Vegas: Self-Speculative Decoding with Verification-Guided Sparse Attention
  8. 2025vAttention: Verified Sparse Attention