mlsanityData to 14 Oct 2026

Task diversity produces systematic transfer but inhibits continual reinforcement learning

arXiv 2606.00880 ↗First posted 30 May 2026cs.LG cs.AIcited by 0

On the map: Deep reinforcement learning methods → Deep RL algorithms and representations

Abstract (arXiv)

Continual reinforcement learning (RL) aims to produce agents that never stop adapting to new tasks. A key question is how this interacts with the diversity of tasks an agent experiences. Prior work has shown that training on many diverse tasks leads to agents with strong zero-shot and in-context adaptation. However, this work evaluated agents after they'd stopped learning, i.e. with frozen weights. How task diversity affects an agent's ability to continue learning over a sequence of distribution shifts remains unclear. We introduce Banyan, a GPU-accelerated continual RL domain where one can parametrically control three independent axes that define a task: the map layouts an agent must navigate, the objects it must interact with, and the hierarchical structures of sub-goal dependencies. We find that increasing diversity along each axis induces systematic transfer -- that is, agents begin training on a new task distribution near the performance attained on the previous one, even when the shift changes the structure of the optimal policy. While increasing diversity improves systematic transfer, we find that too much diversity inhibits a learner's ability to continue adapting to new task distributions. As diversity increases, learners plateau in the success rate they achieve on new tasks, yet continue improving on old tasks -- even without further exposure to them. We find this phenomenon manifests across continual learning algorithms, memory architectures, architecture sizes, and in Kinetix -- a physics-based control domain. We release Banyan as a domain for running controlled experiments that study continual RL in the many-tasks regime. Code is available at https://github.com/nhshah15/banyan.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Deep RL algorithms and representations, one of 15 directions in Deep reinforcement learning methods. Builds on 41 of its 59 references that are in the map. First posted on arXiv 30 May 2026.

Builds on

41 of 59 references.

Closest ancestors · 1

Same direction, newest first

  1. 2024Simplifying Deep Temporal Difference Learning5 Jul

Other references · 40

  1. 2017Proximal Policy Optimization Algorithmscited by 15,215
  2. 2016Layer Normalizationcited by 5,669
  3. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
  4. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  5. 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
  6. 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 2,026
  7. 2017Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real Worldcited by 1,498
  8. 2016Progressive Neural Networkscited by 1,361
  9. 2017Continual Learning with Deep Generative Replaycited by 1,158
  10. 2019Solving Rubik's Cube with a Robot Handcited by 734
  11. 2017Sim-to-Real Transfer of Robotic Control with Dynamics Randomizationcited by 672
  12. 2019Leveraging Procedural Generation to Benchmark Reinforcement Learningcited by 483
28 more
  1. 2016The Option-Critic Architecturecited by 448
  2. 2018Quantifying Generalization in Reinforcement Learningcited by 352
  3. 2016Successor Features for Transfer in Reinforcement Learningcited by 340
  4. 2023Minigrid & Miniworld: Modular & Customizable Reinforcement Learning Environments for Goal-Oriented Taskscited by 327
  5. 2019Stabilizing Transformers for Reinforcement Learningcited by 270
  6. 2022The Primacy Bias in Deep Reinforcement Learningcited by 247
  7. 2017Distral: Robust Multitask Reinforcement Learningcited by 233
  8. 2016Modular Multitask Reinforcement Learning with Policy Sketchescited by 203
  9. 2021Open-Ended Learning Leads to Generally Capable Agentscited by 182
  10. 2023Human-Timescale Adaptation in an Open-Ended Task Spacecited by 138
  11. 2021Continual World: A Robotic Benchmark For Continual Reinforcement Learningcited by 116
  12. 2022General-Purpose In-Context Learning by Meta-Learning Transformerscited by 105
  13. 2023A Definition of Continual Reinforcement Learningcited by 101
  14. 2017Zero-Shot Task Generalization with Multi-Task Deep Reinforcement Learningcited by 98
  15. 2024Craftax: A Lightning-Fast Benchmark for Open-Ended Reinforcement Learningcited by 88
  16. 2023XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAXcited by 55
  17. 2019Environmental drivers of systematicity and generalization in a situated agentcited by 50
  18. 2020A Boolean Task Algebra for Reinforcement Learningcited by 48
  19. 2024Kinetix: Investigating the Training of General Agents through Open-Ended Physics-Based Control Taskscited by 45
  20. 2021Same State, Different Task: Continual Reinforcement Learning without Interferencecited by 43
  21. 2020Lifelong Learning of Compositional Structurescited by 35
  22. 2024GenRL: Multimodal-foundation world models for generalization in embodied agentscited by 31
  23. 2024MaestroMotif: Skill Design from Artificial Intelligence Feedbackcited by 24
  24. 2025Cross-environment Cooperation Enables Zero-shot Multi-agent Coordinationcited by 14
  25. 2023Combining Behaviors with the Successor Features Keyboardcited by 12
  26. 2022Task-Agnostic Continual Reinforcement Learning: Gaining Insights and Overcoming Challengescited by 11
  27. 2025MEAL: A Benchmark for Continual Multi-Agent Reinforcement Learningcited by 11
  28. 2023Composing Task Knowledge with Modular Successor Feature Approximatorscited by 7

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Beyond Single-Model Optimization: Preserving Plasticity in Continual Reinforcement Learning
  2. 2026Optimizing Team Behavior via Extremum-seeking Control in Multi-agent Reinforcement Learning
  3. 2026ContinualSkillBench: Can LLM Agents Truly Evolve Their Capabilities?
  4. 2024The impact of behavioral diversity in multi-agent reinforcement learning
  5. 2024Reset & Distill: A Recipe for Overcoming Negative Transfer in Continual Reinforcement Learning
  6. 2026Continual Reinforcement Learning with Neuroevolution
  7. 2026To Retain or to Adapt? Generalizing Continual Learning
  8. 2025Multi-Task Reinforcement Learning Enables Parameter Scaling