mlsanityData to 14 Oct 2026

Attention Once Is All You Need: Efficient Streaming Inference with Stateful Transformers

arXiv 2605.13784 ↗First posted 13 May 2026cs.LGcited by 2

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

Abstract (arXiv)

Conventional transformer inference engines are request-driven, paying an O(n) prefill cost on every query. In streaming workloads, where data arrives continuously and queries probe an ever-growing context, this cost is prohibitive. We introduce a data-driven computational model centred on stateful sessions: a persistent KV cache advanced incrementally as new data arrives, so prefill is moved off the critical path and query latency becomes O(|q|), independent of accumulated context size. Building on this, Flash Queries reclaim idle GPU cycles between data arrivals to pre-evaluate registered questions and return cached answers before the user asks, a pattern that is structurally impossible in stateless engines because they discard intermediate state between requests. A multi-tenant continuous-batching scheduler with cell-budget admission and prefix-aware grouped prefill lets dozens of stateful sessions coexist on a single GPU while preserving full quadratic self-attention. On streaming market-data benchmarks the reference implementation achieves up to 5.9x speedup over conventional inference engines (vLLM, SGLang, TensorRT-LLM, llama.cpp), holding query latency constant as accumulated context grows.

Key facts. Cited by 2 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 17 of its 21 references that are in the map. First posted on arXiv 13 May 2026.

Builds on

17 of 21 references.

Closest ancestors · 1

Same direction, newest first

  1. 2023Efficient Memory Management for Large Language Model Serving with PagedAttention12 Sep

Other references · 16

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
  3. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  4. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  5. 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
  6. 2020Longformer: The Long-Document Transformercited by 3,413
  7. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  8. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  9. 2020Big Bird: Transformers for Longer Sequencescited by 1,684
  10. 2019Generating Long Sequences with Sparse Transformerscited by 1,546
  11. 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,141
  12. 2023BloombergGPT: A Large Language Model for Financecited by 939
4 more
  1. 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
  2. 2024DistServe: Disaggregating Prefill and Decoding for Goodput-optimized Large Language Model Servingcited by 617
  3. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
  4. 2023Can ChatGPT Forecast Stock Price Movements? Return Predictability and Large Language Modelscited by 129

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Stream2LLM: Overlap Context Streaming and Prefill for Reduced Time-to-First-Token (TTFT)
  2. 2025Continuous Prompts: LLM-Augmented Pipeline Processing over Unstructured Streams
  3. 2025PrefillOnly: An Inference Engine for Prefill-only Workloads in Large Language Model Applications
  4. 2025Optimizing LLM Inference: Fluid-Guided Online Scheduling with Memory Constraints
  5. 2026GrowPage: On-Demand KV Budgeting for Efficient LLM Reasoning Serving
  6. 2024CORM: Cache Optimization with Recent Message for Large Language Model Inference
  7. 2025HeadInfer: Memory-Efficient LLM Inference by Head-wise Offloading
  8. 2024PipeInfer: Accelerating LLM Inference using Asynchronous Pipelined Speculation