arXiv:2501.18093cs.LGcs.RO2025-01中稿 · presentation at th…被引 2

用奖励预测误差优化经验回放,加速强化学习训练

Reward Prediction Error Prioritisation in Experience Replay: The RPE-PER Method

  • 基于奖励预测误差动态挑选高价值经验
  • 在连续控制任务中提升学习速度与最终性能
  • 适合追求高效训练的强化学习研究者

强化学习算法通过与环境的迭代交互来学习最优控制策略。经验回放缓冲区是这一过程的关键组件,它存储历史经验,使算法能够从多样化的交互中学习,而不仅仅是最近的经验。在经验有限的动态环境中,这一点尤为重要。然而,如何高效地选择高价值经验以加速训练仍是一大挑战。受生物系统中奖励预测误差(RPE)在适应性行为和学习中的作用启发,本文提出奖励预测误差优先经验回放(RPE-PER)。该方法基于RPE对缓冲区中的经验进行优先级排序。具体而言,引入一个名为EMCN的批评家网络,不仅能输出标准批评家网络的Q值,还能预测奖励。实际奖励与预测奖励之间的差异即为RPE,作为经验优先级的信号。在多个连续控制任务上的实验表明,与基线方法相比,RPE-PER能有效提升离策略演员-批评家算法的学习速度与性能。

原文摘要 · Abstract (English)

Reinforcement Learning algorithms aim to learn optimal control strategies through iterative interactions with an environment. A critical element in this process is the experience replay buffer, which stores past experiences, allowing the algorithm to learn from a diverse range of interactions rather than just the most recent ones. This buffer is especially essential in dynamic environments with limited experiences. However, efficiently selecting high-value experiences to accelerate training remains a challenge. Drawing inspiration from the role of reward prediction errors (RPEs) in biological systems, where they are essential for adaptive behaviour and learning, we introduce Reward Predictive Error Prioritised Experience Replay (RPE-PER). This novel approach prioritises experiences in the buffer based on RPEs. Our method employs a critic network, EMCN, that predicts rewards in addition to the Q-values produced by standard critic networks. The discrepancy between these predicted and actual rewards is computed as RPE and utilised as a signal for experience prioritisation. Experimental evaluations across various continuous control tasks demonstrate RPE-PER's effectiveness in enhancing the learning speed and performance of off-policy actor-critic algorithms compared to baseline approaches.

强化学习经验回放奖励预测误差

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