arXiv:2410.08896cs.LG2024-10ICLR被引 16

用世界模型生成数据,让高更新率强化学习更稳定。

MAD-TD: Model-Augmented Data stabilizes High Update Ratio RL

  • 用小型世界模型生成数据,增强TD学习的稳定性
  • 在深度思维控制套件中实现高更新率下的竞争性性能
  • 适合需要持续学习的现实强化学习场景

构建能在少量样本下找到良好策略的深度强化学习智能体极为困难。为提升样本效率,近期工作尝试对每个新样本执行大量梯度更新。尽管高更新比(UTD)表现出强劲的实证性能,但训练过程易不稳定。以往方法需周期性重置神经网络参数来缓解,但重启训练在许多真实场景中不可行,且需调参重置间隔。本文聚焦于低样本下稳定训练的核心难题:学习到的价值函数无法泛化至未观测的策略动作。我们通过在离策略强化学习中引入少量由学习到的世界模型生成的数据,直接缓解此问题。提出的方法——基于时序差分学习的模型增强数据(MAD-TD),利用少量生成数据稳定高UTD训练,在深度思维控制套件中最难任务上达到竞争性表现。实验进一步表明,高质量模型生成数据至关重要,MAD-TD能有效抑制价值过估计,并显著提升持续学习中的实际稳定性。

原文摘要 · Abstract (English)

Building deep reinforcement learning (RL) agents that find a good policy with few samples has proven notoriously challenging. To achieve sample efficiency, recent work has explored updating neural networks with large numbers of gradient steps for every new sample. While such high update-to-data (UTD) ratios have shown strong empirical performance, they also introduce instability to the training process. Previous approaches need to rely on periodic neural network parameter resets to address this instability, but restarting the training process is infeasible in many real-world applications and requires tuning the resetting interval. In this paper, we focus on one of the core difficulties of stable training with limited samples: the inability of learned value functions to generalize to unobserved on-policy actions. We mitigate this issue directly by augmenting the off-policy RL training process with a small amount of data generated from a learned world model. Our method, Model-Augmented Data for TD Learning (MAD-TD), uses small amounts of generated data to stabilize high UTD training and achieve competitive performance on the most challenging tasks in the DeepMind control suite. Our experiments further highlight the importance of employing a good model to generate data, MAD-TD's ability to combat value overestimation, and its practical stability gains for continued learning.

强化学习高更新率世界模型稳定性

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