Continuous Latent Contexts Enable Efficient Online Learning in Transformers
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) exhibit a strong capacity for in-context learning: Given labeled examples, they can generate good predictions without parameter updates. However, many interactive settings go beyond static prediction to online decision-making, in which effective behavior demands adaptation over long multi-turn horizons in response to feedback, and efficient algorithms in these domains must use compact representations of what they have learned. Recently, continuous transformer architectures with latent chain of thought have shown promise for offline iterative tasks such as directed graph-reachability. Motivated by this, we study whether continuous latent context tokens equip transformers to more effectively realize online learning. We give explicit constructions of constant-depth transformers that implement two foundational online decision-making procedures -- the weighted majority algorithm and Q-learning -- by storing their algorithmic state as linear combinations of feature embeddings, using a small number of latent context tokens. We further train a small GPT-2-style transformer with latent contexts using a multi-curriculum objective that does not directly supervise the latent states. On long synthetic online prediction sequences, this model outperforms larger and more complex LLMs, including Qwen-3-14B and DeepSeek-V3. Our results suggest that continuous latent contexts provide a simple and effective persistent state for transformers to implement online learning algorithms.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 44 of its 56 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
44 of 56 references.
References · 44
- 2017Attention Is All You Needcited by 59,932
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2025Qwen3 Technical Reportcited by 8,195
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 671
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
- 2023Language Models Represent Space and Timecited by 306
- 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 293
- 2023Function Vectors in Large Language Modelscited by 292
- 2024Chain of Thought Empowers Transformers to Solve Inherently Serial Problemscited by 256
- 2023Think before you speak: Training Language Models With Pause Tokenscited by 254
- 2022In-context Reinforcement Learning with Algorithm Distillationcited by 186
32 more
- 2024Let's Think Dot by Dot: Hidden Computation in Transformer Language Modelscited by 186
- 2023What Formal Languages Can Transformers Express? A Surveycited by 120
- 2024Exploratory Preference Optimization: Harnessing Implicit Q*-Approximation for Sample-Efficient RLHFcited by 106
- 2020Improper Learning for Non-Stochastic Controlcited by 83
- 2025Reasoning by Superposition: A Theoretical Perspective on Chain of Continuous Thoughtcited by 73
- 2023Tighter Bounds on the Expressivity of Transformer Encoderscited by 65
- 2024Can Looped Transformers Learn to Implement Multi-step Gradient Descent for In-context Learning?cited by 60
- 2020Q-learning with Logarithmic Regretcited by 59
- 2023Guiding Language Model Reasoning with Planning Tokenscited by 58
- 2024Longhorn: State Space Models are Amortized Online Learnerscited by 58
- 2025A Little Depth Goes a Long Way: The Expressive Power of Log-Depth Transformerscited by 49
- 2017Learning Linear Dynamical Systems via Spectral Filteringcited by 46
- 2024Theoretical limitations of multi-layer Transformercited by 43
- 2025Transformers Learn to Implement Multi-step Gradient Descent with Chain of Thoughtcited by 32
- 2024Transformers Can Learn Temporal Difference Methods for In-Context Reinforcement Learningcited by 22
- 2024Interpretable User Satisfaction Estimation for Conversational Systems with Large Language Modelscited by 19
- 2025Latent Thinking Optimization: Your Latent Reasoning Language Model Secretly Encodes Reward Signals in Its Latent Thoughtscited by 18
- 2025A Comedy of Estimators: On KL Regularization in RL Training of LLMscited by 18
- 2025Emergence of Superposition: Unveiling the Training Dynamics of Chain of Continuous Thoughtcited by 16
- 2022Online Adaptive Policy Selection in Time-Varying Systems: No-Regret via Contractive Perturbationscited by 14
- 2024Latent Plan Transformer for Trajectory Abstraction: Planning as Latent Space Inferencecited by 14
- 2024Training Nonlinear Transformers for Chain-of-Thought Inference: A Theoretical Generalization Analysiscited by 14
- 2024Mean-Field Sampling for Cooperative Multi-Agent Reinforcement Learningcited by 9
- 2024Efficient Reinforcement Learning for Global Decision Making in the Presence of Local Agents at Scalecited by 8
- 2024Online Policy Optimization in Unknown Nonlinear Systemscited by 8
- 2026Latent Reasoning with Supervised Thinking Statescited by 8
- 2026Thinking Without Words: Efficient Latent Reasoning with Abstract Chain-of-Thoughtcited by 7
- 2025Feel-Good Thompson Sampling for Contextual Bandits: a Markov Chain Monte Carlo Showdowncited by 4
- 2026Unsupervised Decomposition and Recombination with Discriminator-Driven Diffusion Modelscited by 3
- 2026Learning Approximate Nash Equilibria in Cooperative Multi-Agent Reinforcement Learning via Mean-Field Subsamplingcited by 2
- 2026Transformers in the Dark: Navigating Unknown Search Spaces via Bandit Feedbackcited by 2
- 2025A Simple Generalisation of the Implicit Dynamics of In-Context Learningcited by 1
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Hob-VL: A Benchmark for Visually Grounded Boolean Reasoning1 Oct 2026
- 2026SMat-Attention: Structured Long-Context Sequence Modeling28 Sep 2026
- 2026Role Steering of Language Models for Social Simulations9 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2025Next-Latent Prediction Transformers Learn Compact World Models
- 2026Why Struggle with Continuous Latents? Interpretable Discrete Latent Reasoning via Rendered Compression
- 2026Learn from your own latents and not from tokens: A sample-complexity theory
- 2026DecisionLLM: Large Language Models for Long Sequence Decision Exploration
- 2026Improving Latent Generalization Using Test-time Compute
- 2026Grounding latent algorithm routing in transformer reasoning
- 2025Learning When to Stop: Adaptive Latent Reasoning via Reinforcement Learning
- 2026Language Models Struggle to Use Representations Learned In-Context