提出双重温和泛化机制,提升离线强化学习性能与稳定性
Doubly Mild Generalization for Offline Reinforcement Learning
- 在数据集附近温和拓展动作空间,避免过泛化
- 有效控制值函数过估计,实测在多种任务上达顶尖水平
- 兼具离线与在线学习优势,适合需要稳定泛化的场景
离线强化学习面临外推误差与值函数过估计问题,根源在于价值函数或策略对分布外动作的过度泛化。尽管现有方法试图完全避免泛化,本文指出在特定条件下,适度泛化仍可信赖并用于提升性能。为此提出双重温和泛化(DMG),包含:(i) 动作层面的温和泛化,即在数据集附近邻域选择最大化Q值的动作;(ii) 泛化传播的温和控制,防止错误泛化在自举过程中累积放大,同时保留学习信号传播。理论上,DMG在理想泛化情形下优于最优样本内策略;即使在最坏泛化情况下,也能控制过估计并提供性能下界。实验表明,DMG在Gym-MuJoCo运动任务和具有挑战性的AntMaze任务中均取得领先表现。此外,其在泛化灵活性上的优势使其能平滑过渡至在线学习,并实现优异的在线微调效果。
原文摘要 · Abstract (English)
Offline Reinforcement Learning (RL) suffers from the extrapolation error and value overestimation. From a generalization perspective, this issue can be attributed to the over-generalization of value functions or policies towards out-of-distribution (OOD) actions. Significant efforts have been devoted to mitigating such generalization, and recent in-sample learning approaches have further succeeded in entirely eschewing it. Nevertheless, we show that mild generalization beyond the dataset can be trusted and leveraged to improve performance under certain conditions. To appropriately exploit generalization in offline RL, we propose Doubly Mild Generalization (DMG), comprising (i) mild action generalization and (ii) mild generalization propagation. The former refers to selecting actions in a close neighborhood of the dataset to maximize the Q values. Even so, the potential erroneous generalization can still be propagated, accumulated, and exacerbated by bootstrapping. In light of this, the latter concept is introduced to mitigate the generalization propagation without impeding the propagation of RL learning signals. Theoretically, DMG guarantees better performance than the in-sample optimal policy in the oracle generalization scenario. Even under worst-case generalization, DMG can still control value overestimation at a certain level and lower bound the performance. Empirically, DMG achieves state-of-the-art performance across Gym-MuJoCo locomotion tasks and challenging AntMaze tasks. Moreover, benefiting from its flexibility in both generalization aspects, DMG enjoys a seamless transition from offline to online learning and attains strong online fine-tuning performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。