Context Language Models

Rulin Shao, Shannon Zejiang Shen, Junjie Oscar Yin, Yuetai Li, Minheng Wang, Hamish Ivison, Radha Poovendran, Nathan Lambert and 5 others

University of Washington · Meta Superintelligence Labs · MIT · Trillium Labs

Published September 29, 2026 · on the feed September 30, 2026 · ▲ 38 · arXiv 2609.37725

The abstract

We introduce Context Language Models (CLMs), language models that natively manage their own context. We implement this by treating the context as a file and allowing the model to make unrestricted updates to this file. This allows the model to learn what is most important to maintain in context, and naturally extends to multi-agent systems where multiple agent contexts coexist as files. Building CLMs zero-shot with existing models outperforms SOTA context management strategies across a variety of tasks: 11.4% higher accuracy with 21.5% fewer FLOPs on BrowseComp-Plus, 5% higher scores with 59% fewer FLOPs on 12-hour EdgeBench, and 65% greater improvement with the same compute on a 24-hour multi-repository agent-swarm task. Moreover, by shifting context management from external harness control to intrinsic model behavior, CLMs naturally enable both in-context and parametric learning of context-management strategies. We show that CLMs can be steered with natural-language instructions evolved through a standard skill-optimization loop, improving held-out accuracy by up to 35.9 points on a context-management task while reducing compute. We also introduce an online reinforcement learning method for CLMs, improving Qwen3.5-9B performance on BrowseComp-Plus by 47.6% while using 12% fewer FLOPs. Finally, we co-design Suffix Cache Reuse for CLM serving, further reducing server-side compute by 35% relative to standard SGLang at matched performance.

The Brief

2:41

A short podcast summary of the paper. Written and voiced by AI.

Transcript

Context Language Models. By Rulin Shao and twelve others, from the University of Washington, Meta Superintelligence Labs, and two other institutions. Published on the twenty-ninth of September. Its code repository is worked out: about forty-five source files, with a readme and a licence.

A language model working through a long task has to decide what stays in its context and what goes. Usually that decision is not the model's to make. Context management sits in an external harness, a set of strategies wrapped around the model that control what it gets to see. The model itself never learns what is most important to keep.

Context Language Models, or CLMs, are language models that natively manage their own context. The implementation is simple to state. The context is treated as a file, and the model is allowed to make unrestricted updates to that file. Because the model is the one editing, it can learn what is most important to maintain. The same framing extends naturally to multi-agent systems, where several agents' contexts coexist side by side as files.

Moving context management from external harness control into intrinsic model behaviour opens two routes for learning it. One is in context: CLMs can be steered with natural-language instructions, evolved through a standard skill-optimisation loop. The other is parametric: the authors introduce an online reinforcement learning method for CLMs. Serving is co-designed too, with a technique called Suffix Cache Reuse aimed at the server-side cost of running models that rewrite their own context.

Even built zero-shot from existing models, CLMs outperform state-of-the-art context management strategies across a variety of tasks. On BrowseComp-Plus, accuracy is eleven point four percent higher with twenty-one point five percent fewer FLOPs. On a twelve-hour EdgeBench run, scores are five percent higher with fifty-nine percent fewer FLOPs. And on a twenty-four-hour agent-swarm task spanning multiple repositories, the improvement is sixty-five percent greater at the same compute.

Learning the strategy adds more. The evolved instructions improve held-out accuracy on a context-management task by up to thirty-five point nine points, while reducing compute. The reinforcement learning method lifts Qwen3.5-9B on BrowseComp-Plus by forty-seven point six percent, using twelve percent fewer FLOPs. On the serving side, Suffix Cache Reuse cuts server-side compute by thirty-five percent relative to standard SGLang, at matched performance.

Most of the gains here come in pairs: better results alongside fewer FLOPs. Underneath them is a change in where the decision lives. Once managing context belongs to the model rather than to the harness around it, it becomes something that can be taught, by instruction or by reward, and served more cheaply when the serving stack is built with it in mind.

Paper Quest · 7 chapters · about 41 min

Play this paper

Learn it by doing: make the calls the authors made, guess the result before you see it, and defend it in a viva graded on your own words. Chapter 1 is right here.

Chapter 1 🧩 · The problem

Every long task hits the same wall: the context fills up. Who decides what goes?

You run a language model agent on a long research task. Its history grows far faster than the window allows, and something has to go.

The road ahead

  1. 1 · The problem ~7 min On long tasks an agent's history outgrows its context window, and today fixed outside rules decide what to cut, often badly.
  2. 2 · What came before ~3 min
  3. 3 · The key idea ~9 min Let the model freely edit its own context, kept as a file, so it learns what to keep and that strategy can be improved.
  4. 4 · Choose your own research ~3 min
  5. 5 · Build the method ~8 min Mirror the context into a file, let the model edit it freely, train its strategy with reinforcement learning, and serve it with Suffix Cache Reuse.
  6. 6 · The experiment ~6 min You’ll guess: Built zero-shot from existing models, how much higher was CLM accuracy on BrowseComp-Plus than state-of-the-art context management strategies?
  7. 7 · The viva ~5 min Three questions from the examiners, answered in your own words. Win it for the paper’s trophy.
Join free to play the whole quest ›

Free, with Google or an email address. Your stars and progress are kept.

KipQuest: the day’s AI research, as podcasts and quests. All papers Android app Privacy Contact Episodes are written and voiced by AI from each paper. Summaries can contain mistakes: the paper is the source.