arXiv:2503.20139cs.LGcs.AI2025-03被引 4

通过前瞻规划与不确定性感知,提升强化学习的样本效率与稳定性。

Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learning

  • 引入不确定性感知的多步前瞻规划,动态权衡模型误差与价值估计。
  • 在少样本区域主动采集数据,显著提升模型准确性和状态覆盖。
  • 适用于复杂机器人任务与游戏,交互次数更少但性能更强。

基于模型的强化学习(MBRL)相比无模型方法具有更高的样本效率,但模型不准确会引入偏差,导致错误轨迹。问题根源在于缺乏多样化训练数据,尤其在访问较少的不确定区域。现有方法仅被动量化不确定性,无法主动获取有助于提升状态覆盖和模型精度的样本。此外,MBRL常难以实现准确的多步预测,影响整体性能。为此,我们提出一种不确定性感知的模型化探索规划框架。在模型规划阶段,引入不确定性感知的k步前瞻规划,权衡模型不确定性与价值函数近似误差,优化每步动作选择;在策略优化阶段,采用不确定性驱动的探索策略,主动收集多样样本,提升模型精度与整体性能。该方法对不同状态/动作空间和奖励结构的任务均具灵活性。实验验证其在挑战性机器人操作任务和Atari游戏上的有效性,以更少交互次数超越现有最佳方法,带来显著性能提升。

原文摘要 · Abstract (English)

Model-based reinforcement learning (MBRL) has demonstrated superior sample efficiency compared to model-free reinforcement learning (MFRL). However, the presence of inaccurate models can introduce biases during policy learning, resulting in misleading trajectories. The challenge lies in obtaining accurate models due to limited diverse training data, particularly in regions with limited visits (uncertain regions). Existing approaches passively quantify uncertainty after sample generation, failing to actively collect uncertain samples that could enhance state coverage and improve model accuracy. Moreover, MBRL often faces difficulties in making accurate multi-step predictions, thereby impacting overall performance. To address these limitations, we propose a novel framework for uncertainty-aware policy optimization with model-based exploratory planning. In the model-based planning phase, we introduce an uncertainty-aware k-step lookahead planning approach to guide action selection at each step. This process involves a trade-off analysis between model uncertainty and value function approximation error, effectively enhancing policy performance. In the policy optimization phase, we leverage an uncertainty-driven exploratory policy to actively collect diverse training samples, resulting in improved model accuracy and overall performance of the RL agent. Our approach offers flexibility and applicability to tasks with varying state/action spaces and reward structures. We validate its effectiveness through experiments on challenging robotic manipulation tasks and Atari games, surpassing state-of-the-art methods with fewer interactions, thereby leading to significant performance improvements.

强化学习不确定性模型规划机器人

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