mlsanityData to 7 Oct 2026

Scaling In-Context Online Learning Capability of LLMs via Cross-Episode Meta-RL

arXiv 2602.04089 ↗First posted 3 Feb 2026cs.AI cs.CL cs.LGcited by 6

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2022Training language models to follow instructions with human feedbackcited by 15,712
  4. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  5. 2025Qwen3 Technical Reportcited by 8,195
  6. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  7. 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,064
  8. 2017Deep reinforcement learning from human preferencescited by 3,846
  9. 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,528
  10. 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,193
  11. 2022Transformers learn in-context by gradient descentcited by 714
  12. 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
14 more
  1. 2022In-context Reinforcement Learning with Algorithm Distillationcited by 186
  2. 2023Supervised Pretraining Can Learn In-Context Reinforcement Learningcited by 137
  3. 2023Transformers as Decision Makers: Provable In-Context Reinforcement Learning via Supervised Pretrainingcited by 82
  4. 2024Can large language models explore in-context?cited by 75
  5. 2023Generalization to New Sequential Decision Making Tasks with In-Context Learningcited by 39
  6. 2024Do LLM Agents Have Regret? A Case Study in Online Learning and Gamescited by 33
  7. 2024EVOLvE: Evaluating and Optimizing LLMs For In-Context Explorationcited by 33
  8. 2024Controlling Large Language Model Agents with Entropic Activation Steeringcited by 29
  9. 2025Training a Generally Curious Agentcited by 29
  10. 2024LLMs Are In-Context Bandit Reinforcement Learnerscited by 27
  11. 2026PACEvolve: Enabling Progress-Aware Consistent Evolutioncited by 21
  12. 2024On the Importance of Uncertainty in Decision-Making with Large Language Modelscited by 9
  13. 2025Large Language Model-Enhanced Multi-Armed Banditscited by 7
  14. 2025Post-Training LLMs as Better Decision-Making Agents: A Regret-Minimization Approachcited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 6.

6 papers.