mlsanityData to 14 Oct 2026

Reflective Context Learning: Studying the Optimization Primitives of Context Space

arXiv 2604.03189 ↗First posted 3 Apr 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Generally capable agents must learn from experience in ways that generalize across tasks and environments. The fundamental problems of learning, including credit assignment, overfitting, forgetting, local optima, and high-variance learning signals, persist whether the learned object lies in parameter space or context space. While these challenges are well understood in classical machine learning optimization, they remain underexplored in context space, leading current methods to be fragmented and ad hoc. We present Reflective Context Learning (RCL), a unified framework for agents that learn through repeated interaction, reflection on behavior and failure modes, and iterative updates to context. In RCL, reflection converts trajectories and current context into a directional update signal analogous to gradients, while mutation applies that signal to improve future behavior in context space. We recast recent context-optimization approaches as instances of this shared learning problem and systematically extend them with classical optimization primitives, including batching, improved credit-assignment signal, auxiliary losses, failure replay, and grouped rollouts for variance reduction. On AppWorld, BrowseComp+, and RewardBench2, these primitives improve over strong baselines, with their relative importance shifting across task regimes. We further analyze robustness to initialization, the effects of batch size, sampling and curriculum strategy, optimizer-state variants, and the impact of allocating stronger or weaker models to different optimization components. Our results suggest that learning through context updates should be treated not as a set of isolated algorithms, but as an optimization problem whose mechanisms can be studied systematically and improved through transferable principles.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 32 of its 45 references that are in the map. First posted on arXiv 3 Apr 2026.

Builds on

32 of 45 references.

References · 32

  1. 2020Language Models are Few-Shot Learnerscited by 32,419
  2. 2022Training language models to follow instructions with human feedbackcited by 15,971
  3. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
  4. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  5. 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,276
  6. 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,915
  7. 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,795
  8. 2023Let's Verify Step by Stepcited by 3,789
  9. 2023Voyager: An Open-Ended Embodied Agent with Large Language Modelscited by 2,191
  10. 2015Prioritized Experience Replaycited by 1,099
  11. 2022Large Language Models Are Human-Level Prompt Engineerscited by 1,045
  12. 2023ExpeL: LLM Agents Are Experiential Learnerscited by 876
20 more
  1. 2023Large Language Models as Optimizerscited by 864
  2. 2023DSPy: Compiling Declarative Language Model Calls into Self-Improving Pipelinescited by 832
  3. 2023Automatic Prompt Optimization with "Gradient Descent" and Beam Searchcited by 653
  4. 2023Promptbreeder: Self-Referential Self-Improvement Via Prompt Evolutioncited by 551
  5. 2017Value-Decomposition Networks For Cooperative Multi-Agent Learningcited by 503
  6. 2016Reinforcement Learning with Unsupervised Auxiliary Taskscited by 404
  7. 2024AppWorld: A Controllable World of Apps and People for Benchmarking Interactive Coding Agentscited by 401
  8. 2025Agentic Context Engineering: Evolving Contexts for Self-Improving Language Modelscited by 398
  9. 2024Optimizing Instructions and Demonstrations for Multi-Stage Language Model Programscited by 291
  10. 2024TextGrad: Automatic "Differentiation" via Textcited by 273
  11. 2025BrowseComp-Plus: A More Fair and Transparent Evaluation Benchmark of Deep-Research Agentcited by 200
  12. 2025A Comprehensive Survey of Self-Evolving AI Agents: A New Paradigm Bridging Foundation Models and Lifelong Agentic Systemscited by 183
  13. 2025Dynamic Cheatsheet: Test-Time Learning with Adaptive Memorycited by 163
  14. 2025RewardBench 2: Advancing Reward Model Evaluationcited by 150
  15. 2025A Survey of Context Engineering for Large Language Modelscited by 126
  16. 2024Agent-Pro: Learning to Evolve via Policy-Level Reflection and Optimizationcited by 94
  17. 2025Training-Free Group Relative Policy Optimizationcited by 55
  18. 2025A Survey of Automatic Prompt Engineering: An Optimization Perspectivecited by 41
  19. 2024Efficient and Accurate Prompt Optimization: the Benefit of Memory in Exemplar-Guided Reflectioncited by 18
  20. 2025Scaling Textual Gradients via Sampling-Based Momentumcited by 6

Built on by

Not cited yet.