Scaling In-Context Online Learning Capability of LLMs via Cross-Episode Meta-RL
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) achieve strong performance when all task-relevant information is available upfront, as in static prediction and instruction-following problems. However, many real-world decision-making tasks are inherently online: crucial information must be acquired through interaction, feedback is delayed, and effective behavior requires balancing information collection and exploitation over time. While in-context learning enables adaptation without weight updates, existing LLMs often struggle to reliably leverage in-context interaction experience in such settings. In this work, we show that this limitation can be addressed through training. We introduce ORBIT, a multi-task, multi-episode meta-reinforcement learning framework that trains LLMs to learn from interaction in context. After meta-training, a relatively small open-source model (Qwen3-14B) demonstrates substantially improved in-context online learning on entirely unseen environments, matching the performance of GPT-5.2 and outperforming standard RL fine-tuning by a large margin. Scaling experiments further reveal consistent gains with model size, suggesting significant headroom for learn-at-inference-time decision-making agents. Code reproducing the results in the paper can be found at https://github.com/XiaofengLin7/ORBIT.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 26 of its 28 references that are in the map. First posted on arXiv 3 Feb 2026.
Builds on
26 of 28 references.
References · 26
- 2017Attention Is All You Needcited by 59,932
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,064
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,528
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,193
- 2022Transformers learn in-context by gradient descentcited by 714
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
14 more
- 2022In-context Reinforcement Learning with Algorithm Distillationcited by 186
- 2023Supervised Pretraining Can Learn In-Context Reinforcement Learningcited by 137
- 2023Transformers as Decision Makers: Provable In-Context Reinforcement Learning via Supervised Pretrainingcited by 82
- 2024Can large language models explore in-context?cited by 75
- 2023Generalization to New Sequential Decision Making Tasks with In-Context Learningcited by 39
- 2024Do LLM Agents Have Regret? A Case Study in Online Learning and Gamescited by 33
- 2024EVOLvE: Evaluating and Optimizing LLMs For In-Context Explorationcited by 33
- 2024Controlling Large Language Model Agents with Entropic Activation Steeringcited by 29
- 2025Training a Generally Curious Agentcited by 29
- 2024LLMs Are In-Context Bandit Reinforcement Learnerscited by 27
- 2026PACEvolve: Enabling Progress-Aware Consistent Evolutioncited by 21
- 2024On the Importance of Uncertainty in Decision-Making with Large Language Modelscited by 9
- 2025Large Language Model-Enhanced Multi-Armed Banditscited by 7
- 2025Post-Training LLMs as Better Decision-Making Agents: A Regret-Minimization Approachcited by 4
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Training AI Scientists to Replicate Researchcited by 4
- 2026Modularized Reinforcement Learning on LLMs: From MDP Creation to Exploration and Learningcited by 2
- 2026Stratified Consistency Distillation for Natural Language Formalizationcited by 1
- 2026Connect the Dots: Training LLMs for Long-Lifecycle Agents with Cross-Domain Generalization Via Reinforcement Learningcited by 0
- 2026MInTRL: Off-policy Intervention can boost On-policy RLcited by 0
- 2026Few-Shot Demonstrations Elicit the Use of In-Context World Representations in LLMscited by 0
- 2026Few-Shot Demonstrations Elicit the Use of In-Context World Representations in LLMs21 Sep 2026
- 2026MInTRL: Off-policy Intervention can boost On-policy RL11 Sep 2026
- 2026Stratified Consistency Distillation for Natural Language Formalization31 Aug 2026
- 2026Training AI Scientists to Replicate Research13 Aug 2026
- 2026Modularized Reinforcement Learning on LLMs: From MDP Creation to Exploration and Learning20 Jun 2026
- 2026Connect the Dots: Training LLMs for Long-Lifecycle Agents with Cross-Domain Generalization Via Reinforcement Learning18 Jun 2026