UltraQuant: 4-bit KV Caching for Context-Heavy Agents
- lab arXiv
- lab arXivLabs
- product CDNA4
- product CatalyzeX
- product TurboQuant
- product UltraQuant
- product alphaXiv
- product vLLM
A new 4-bit key-value cache compression method called UltraQuant reduces time-to-first-token by up to 3.47x for context-heavy, multi-turn agent workloads, according to a paper posted to arXiv on June 18, 2026 [1][2]. The paper, titled “UltraQuant: 4-bit KV Caching for Context-Heavy Agents,” targets a specific bottleneck in large language model serving: the key-value (KV) cache. In agentic workloads, long prefixes are reused across many short turns, placing unusual pressure on memory and making it difficult to keep GPUs fully utilized [1][2]. The authors frame 4-bit KV-cache compression around these multi-round agent workloads, where task quality, cache residency, and serving throughput must be measured jointly [1][2]. UltraQuant is described as an FP4 approximation path that uses FP8 queries, FP4 KV tensors, UE8M0 group scales, and native scaled-MFMA support on AMD’s CDNA4 architecture [1][2]. The design draws on TurboQuant-style rotation and codebook quantization as a quality anchor, while using vLLM FP8 KV caching as the deployment anchor [1][2]. The paper also details practical design choices to make the 4-bit path robust, including asymmetric key/value treatment, Walsh-Hadamard rotation, QJL removal, and block-scale variants [1][2]. On a long-context, multi-turn agentic workload, UltraQuant cut the P50 time-to-first-token by 3.47x in the cache-pressured late rounds, and by 2.3x across all rounds, compared to an FP8 KV baseline [1][2]. Output throughput rose by 1.63x over the same baseline [1][2]. The work includes optimized decode-attention kernels developed specifically for AMD GPUs [1][2]. The paper was submitted to arXiv’s Machine Learning section (cs.LG). arXiv, which began on August 14, 1991, is an open-access repository of electronic preprints that are moderated but not peer-reviewed [6]. As of November 2024, the repository receives about 24,000 submissions per month and has surpassed two million total articles [6]. The platform also hosts arXivLabs, a framework for community-developed tools that appear on article record pages, though the Labs program is currently pausing new proposals while the arXiv team focuses on a cloud migration [3][4].
applicationresearch-paper
Background sources we checked (7)
- arxiv.org ↗ Context-heavy agents place unusual pressure on the key-value (KV) cache: long prefixes are reused across many short turns, while concurrency determines whether the serving system can keep GPUs utilized. We study 4-bit KV-cache compression for this setting, using TurboQuant-style …
- info.arxiv.org ↗ arXiv Labs - arXiv info | arXiv e-print repository Skip to content # arXiv Labs Attention arXiv Users: arXiv Labs is pausing new proposals ## What are arXiv Labs? arXiv Labs are a way for the community to contribute new, useful features to arXiv. These integrations are avail…
- blog.arxiv.org ↗ arXivLabs: a space for community innovation – arXiv blog arXiv has launched a new, formalized framework enabling innovative collaborations with individuals and organizations. “Members of our community want to contribute tools that enhance the arXiv experience, and we val…
- info.arxiv.org ↗ arXivLabs: Showcase - arXiv info | arXiv e-print repository ... # arXivLabs: Showcase ... arXiv is surrounded by a community of researchers and developers working at the cutting edge of information science and technology. ... While the arXiv team is focused on our core mission—pr…
- en.wikipedia.org ↗ arXiv (pronounced as "archive"—the X represents the Greek letter chi ⟨χ⟩) is an open-access repository of electronic preprints and postprints (known as e-prints) approved for posting after moderation, but not peer reviewed. It consists of scientific papers in the fields of mathem…
- en.wikipedia.org ↗ 14 (fourteen) is the natural number following 13 and preceding 15.…
- en.wikipedia.org ↗ A large language model (LLM) is a type of machine learning model designed for natural language processing tasks such as language generation. LLMs are language models with many parameters, and are trained with self-supervised learning on a vast amount of text.…
Sources
- export.arxiv.org — UltraQuant: 4-bit KV Caching for Context-Heavy Agents ↗