arXiv:2410.18082cs.LG2024-10ICLR被引 15

用生成模型动态重放关键经验,提升在线强化学习的样本效率。

Prioritized Generative Replay

  • 用条件扩散模型生成并优先重放高价值经验
  • 在多种环境中实现更优性能与更高更新/数据比
  • 适合追求高效在线学习的强化学习研究者

样本高效的在线强化学习常使用回放缓冲区重用经验以更新价值函数。然而,均匀回放效率低下,因为某些类型的转移对学习更为关键。虽然优先处理更有用的样本有益,但该策略可能引发过拟合,因有用样本往往更稀少。本文提出一种参数化、优先化的记忆机制,利用生成模型捕捉在线经验。该范式实现:(1) 过去经验的密度化,通过生成新样本利用生成模型的泛化能力;(2) 借助“相关性函数”族引导生成过程,使新样本聚焦于代理历史中更有效的部分。我们证明该方法可基于条件扩散模型与简单相关性度量(如好奇心或价值)实现。实验显示,该方法在状态与像素级任务中均显著提升性能与样本效率。我们揭示其增益机制:引导促进生成转移的多样性,减少过拟合。此外,该方法支持比以往更高的更新/数据比,为在线强化学习的规模化提供新路径。

原文摘要 · Abstract (English)

Sample-efficient online reinforcement learning often uses replay buffers to store experience for reuse when updating the value function. However, uniform replay is inefficient, since certain classes of transitions can be more relevant to learning. While prioritization of more useful samples is helpful, this strategy can also lead to overfitting, as useful samples are likely to be more rare. In this work, we instead propose a prioritized, parametric version of an agent's memory, using generative models to capture online experience. This paradigm enables (1) densification of past experience, with new generations that benefit from the generative model's generalization capacity and (2) guidance via a family of "relevance functions" that push these generations towards more useful parts of an agent's acquired history. We show this recipe can be instantiated using conditional diffusion models and simple relevance functions such as curiosity- or value-based metrics. Our approach consistently improves performance and sample efficiency in both state- and pixel-based domains. We expose the mechanisms underlying these gains, showing how guidance promotes diversity in our generated transitions and reduces overfitting. We also showcase how our approach can train policies with even higher update-to-data ratios than before, opening up avenues to better scale online RL agents.

强化学习生成模型在线学习经验回放

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