arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.04498cs.LG
收藏

AdaMemento: 自适应记忆辅助策略优化用于强化学习

AdaMemento: Adaptive Memory-Assisted Policy Optimization for Reinforcement Learning

Peking University(北京大学);Tsinghua University(清华大学)

Renye Yan, Yaozhong Gan, You Wu, Junliang Xing, Ling Liangn, Yeshang Zhu, Yimao Cai

AI总结:

针对稀疏奖励强化学习问题,提出AdaMemento框架,通过记忆反思模块利用正负经验、细粒度内在动机引导探索以及集成学习自适应协调利用与探索,显著提升性能。

摘要:

在强化学习的稀疏奖励场景中,记忆机制通过像人类一样反思过去经验,为策略优化提供了有前景的捷径。然而,当前基于记忆的强化学习方法仅仅存储和重用高价值策略,缺乏对多样化过去经验的深入精炼和过滤,从而限制了记忆的能力。在本文中,我们提出了AdaMemento,一种自适应记忆增强的强化学习框架。我们不仅记忆正面的过去经验,还设计了一个记忆反思模块,通过学习基于实时状态预测已知局部最优策略,同时利用正面和负面经验。为了有效收集信息丰富的轨迹用于记忆,我们进一步引入了一种细粒度的内在动机范式,其中相似状态中的细微差别可以被精确区分以引导探索。然后,通过集成学习自适应地协调对过去经验的利用和新策略的探索,以逼近全局最优。此外,我们从理论上证明了我们新的内在动机和集成机制的优势。通过59个定量和可视化实验,我们确认AdaMemento能够区分细微状态以更好地探索,并有效利用记忆中的过去经验,相比之前的方法取得了显著改进。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑