arXiv:2410.04498cs.LG2024-10被引 2

通过自适应记忆机制,让强化学习更好利用过去经验。

AdaMemento: Adaptive Memory-Assisted Policy Optimization for Reinforcement Learning

  • 用正负经验联合反思,动态预测局部最优策略。
  • 细粒度内在动机引导探索,精准区分相似状态。
  • 集成学习协调经验利用与新策略探索,适合稀疏奖励场景。

在强化学习的稀疏奖励场景中,记忆机制可借鉴人类经验反思,加速策略优化。现有方法仅存储并复用高价值策略,缺乏对多样历史经验的深度提炼与筛选,限制了记忆能力。本文提出AdaMemento,一种自适应记忆增强的强化学习框架。不同于单纯记忆正面经验,我们设计记忆反思模块,基于实时状态学习预测已知局部最优策略,同时利用正负经验进行优化。为有效收集信息丰富的轨迹,引入细粒度内在动机机制,精确区分相似状态以指导探索。通过集成学习自适应协调过往经验利用与新策略探索,逼近全局最优。理论证明了新内在动机与集成机制的优势。59项定量与可视化实验表明,AdaMemento能精准区分细微状态以提升探索效率,并有效利用记忆中的经验,在多个基准上显著优于先前方法。

原文摘要 · Abstract (English)

In sparse reward scenarios of reinforcement learning (RL), the memory mechanism provides promising shortcuts to policy optimization by reflecting on past experiences like humans. However, current memory-based RL methods simply store and reuse high-value policies, lacking a deeper refining and filtering of diverse past experiences and hence limiting the capability of memory. In this paper, we propose AdaMemento, an adaptive memory-enhanced RL framework. Instead of just memorizing positive past experiences, we design a memory-reflection module that exploits both positive and negative experiences by learning to predict known local optimal policies based on real-time states. To effectively gather informative trajectories for the memory, we further introduce a fine-grained intrinsic motivation paradigm, where nuances in similar states can be precisely distinguished to guide exploration. The exploitation of past experiences and exploration of new policies are then adaptively coordinated by ensemble learning to approach the global optimum. Furthermore, we theoretically prove the superiority of our new intrinsic motivation and ensemble mechanism. From 59 quantitative and visualization experiments, we confirm that AdaMemento can distinguish subtle states for better exploration and effectively exploiting past experiences in memory, achieving significant improvement over previous methods.

强化学习记忆机制稀疏奖励

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。