中文

ARROW:面向鲁棒世界模型的增强式回放

机器学习 2026-05-19 v2 人工智能

摘要

持续强化学习挑战在于在既保持先前学习技能的同时获取新技能,以提高在过去和未来任务中的性能。大多数现有方法依赖无模型方法配合回放缓冲区来缓解灾难性遗忘;然而,这些解决方案常因大内存需求而面临显著可扩展性挑战。drawing inspiration from neuroscience, where the brain replays experiences to a predictive World Model rather than directly to the policy, we present ARROW (Augmented Replay for RObust World models), a model-based continual RL algorithm that extends DreamerV3 with a memory-efficient, distribution-matching replay buffer. Unlike standard fixed-size FIFO buffers, ARROW maintains two complementary buffers: a short-term buffer for recent experiences and a long-term buffer that preserves task diversity through intelligent sampling. We evaluate ARROW on two challenging continual RL settings: Tasks without shared structure (Atari), and tasks with shared structure, where knowledge transfer is possible (Procgen CoinRun variants). Compared to model-free and model-based baselines with replay buffers of the same-size, ARROW demonstrates substantially less forgetting on tasks without shared structure, while maintaining comparable forward transfer. Our findings highlight the potential of model-based RL and bio-inspired approaches for continual reinforcement learning, warranting further research.

关键词

引用

@article{arxiv.2603.11395,
  title  = {ARROW: Augmented Replay for RObust World models},
  author = {Abdulaziz Alyahya and Abdallah Al Siyabi and Markus R. Ernst and Luke Yang and Levin Kuhlmann and Gideon Kowadlo},
  journal= {arXiv preprint arXiv:2603.11395},
  year   = {2026}
}

备注

36 pages and 11 figures (includes Appendix)