arXiv:2608.29860cs.LG2026-09

用不确定性指导经验回放,让智能体学得更快更准。

Uncertainty-Driven Replay Memory for Reinforcement Learning

论文配图:Uncertainty-Driven Replay Memory for Reinforcement Learning
图 1 · 摘自论文原文
  • 根据模型预测不确定性动态调整回放缓冲区内容
  • 在多个环境中实现更高奖励与更快收敛
  • 适合追求高效训练的强化学习研究者

不确定性估计为强化学习(RL)智能体提供了强大能力。通过判断动作是否有助于探索已知或未知区域,可减少训练时间并提升长期奖励。本文提出一种新型经验回放机制——不确定性驱动回放记忆(UDRM),其根据训练中模型生成的不确定性估计动态更新缓冲区存储内容。不同于传统方法依赖时序差分误差或转移分布来更新回放缓冲区,本方案优先存储高不确定性样本,以增强智能体的泛化能力。实验表明,相较于其他不确定性感知的强化学习框架,本方法能显著提升智能体在训练过程中的累积奖励。

原文摘要 · Abstract (English)

Uncertainty estimation provides promising capabilities for reinforcement learning (RL) agents. Notably, estimating uncertainty can reduce the training time and enable agents to obtain greater rewards over time by exploiting information related to whether an action would facilitate exploration of portions of an environment that are well-known versus those that are relatively unknown. In this work, we propose a novel formulation of the experience replay buffer commonly used in RL that we call uncertainty-driven replay memory (UDRM), which entails an update scheme for internally stored memories based on uncertainty estimates obtained by an RL model during training. In contrast to existing forms of RL, which typically use temporal difference error or the distribution of transitions to update the replay memory buffer and train RL controllers, our scheme biases the memory buffer to store more uncertain transitions that will improve an RL agent's generalization throughout training. Experimental results demonstrate that our proposed uncertainty-aware replay buffer enables an RL agent to obtain higher rewards during training compared to other existing uncertainty-aware RL frameworks.

强化学习经验回放不确定性

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