KV-Fold: Unlocking Transformer Long Context

KV-Fold enables training-free, stable long-context inference up to 128K tokens with 100% retrieval accuracy, overcoming prior limitations.

Diagram illustrating the KV-Fold process of accumulating key-value pairs across sequence chunks.
Conceptual overview of the KV-Fold long context inference protocol.
Visual TL;DR
Long context challengeDriver
training demands high resources or architectural changes for extended context
From the articleKV-Fold provides a path to maintaining long-range retrieval capabilities through a series of tractable forward passes, making KV-Fold long context inference a compelling development.
Introducing KV-FoldCore
From the article 4 mentionsA recent arXiv preprint introduces KV-Fold, a novel, training-free approach to long-context inference that ingeniously repurposes the key-value (KV) cache.
Functional foldingContext
KV cache treated as accumulator in left fold across sequence chunks
From the articleThis simple, one-step update mirrors the behavior of `foldl` in functional programming, establishing a stable, chunk-to-chunk recurrence.
Stable recurrenceContext
chunk-to-chunk updates mirror foldl, per-step drift quickly saturates
From the article 2 mentionsThis simple, one-step update mirrors the behavior of `foldl` in functional programming, establishing a stable, chunk-to-chunk recurrence.
Robustness observedContext
From the articleThe researchers observed that per-step drift quickly saturates and remains stable, demonstrating robustness across numerical precision, chunk sizes, and model families.
High-fidelity retrievalEffect
enables long-range retrieval with 100% accuracy
From the article 2 mentionsOn a needle-in-a-haystack benchmark, the method achieved 100% exact-match retrieval across numerous trials, spanning contexts from 16K to 128K tokens.
128K token contextOutcome
achieves stable inference up to 128,000 tokens
From the articleOn a needle-in-a-haystack benchmark, the method achieved 100% exact-match retrieval across numerous trials, spanning contexts from 16K to 128K tokens.

The quest for extended context windows in large language models has historically demanded significant computational resources or architectural modifications. A recent arXiv preprint introduces KV-Fold, a novel, training-free approach to long-context inference that ingeniously repurposes the key-value (KV) cache.

Stable Recurrence via Functional Folding

KV-Fold treats the KV cache as an accumulator in a left fold operation across sequence chunks. At each step, the model processes a new chunk, conditioned on the accumulated cache from previous chunks. This cache is then appended with the newly generated keys and values, and the enlarged cache is passed forward. This simple, one-step update mirrors the behavior of `foldl` in functional programming, establishing a stable, chunk-to-chunk recurrence. The researchers observed that per-step drift quickly saturates and remains stable, demonstrating robustness across numerical precision, chunk sizes, and model families.

Unlocking High-Fidelity Long-Range Retrieval

The practical impact of KV-Fold is demonstrated through its exceptional performance on challenging long-context tasks. On a needle-in-a-haystack benchmark, the method achieved 100% exact-match retrieval across numerous trials, spanning contexts from 16K to 128K tokens. Crucially, this was accomplished on a Llama-3.1-8B model while staying within the memory constraints of a single 40GB GPU, and supporting chain depths up to 511. This stands in contrast to streaming methods that often sacrifice fidelity for memory boundedness. KV-Fold provides a path to maintaining long-range retrieval capabilities through a series of tractable forward passes, making KV-Fold long context inference a compelling development.

© 2026 StartupHub.ai. All rights reserved. You may not republish this article in full without a license. Search engines and AI research tools may crawl and summarize for reference. Bulk reproduction or model training requires a license. See our terms.

Watch

The videos behind the news, one flip at a time

The feed mixes the videos from our articles with startup demos and you flip through them like TV channels for free with no signup.

Daniel Singer

Written by

Daniel Singer

Editor, StartupHub.ai

Daniel Singer is the editor of StartupHub.ai, a technology expert and thought leader on AI and its applications across sectors, from fintech and healthcare to developer tooling and consumer software. He writes and tests the tools covered here thoroughly and regularly, and built StartupHub.ai to give founders, operators and buyers a clearer read on what they are actually being sold.