仅用预训练策略实现高效在线强化学习微调,无需依赖预训练价值函数。
Efficient Online RL Fine Tuning with Offline Pre-trained Policy Only
- 仅利用预训练策略初始化在线阶段价值函数,避免保守估计问题。
- 在多个环境上达到与先进方法相当的性能,且训练效率更高。
- 特别适合仅有行为克隆策略而无价值函数的场景,如模仿学习后处理。
通过在线强化学习改进预训练策略的性能是一个关键但具有挑战性的问题。现有在线微调方法需持续使用离线预训练的价值函数以保证稳定性和性能,但这些价值函数通常对离线数据集外的状态-动作对存在保守估计,导致从离线到在线过渡时探索能力受限。此外,该要求限制了其在仅提供预训练策略而无预训练价值函数场景(如模仿学习预训练)中的应用。为此,我们提出一种仅依赖离线预训练策略的高效在线强化学习微调方法,即 PORL(Policy-Only Reinforcement Learning Fine-Tuning)。该方法在在线阶段从零开始快速初始化价值函数,避免有害的悲观主义。所提方法不仅在多个环境中表现媲美先进离线-在线强化学习算法及依赖先验数据或策略的在线强化学习方法,还首次实现了对行为克隆(BC)策略的直接微调。
原文摘要 · Abstract (English)
Improving the performance of pre-trained policies through online reinforcement learning (RL) is a critical yet challenging topic. Existing online RL fine-tuning methods require continued training with offline pretrained Q-functions for stability and performance. However, these offline pretrained Q-functions commonly underestimate state-action pairs beyond the offline dataset due to the conservatism in most offline RL methods, which hinders further exploration when transitioning from the offline to the online setting. Additionally, this requirement limits their applicability in scenarios where only pre-trained policies are available but pre-trained Q-functions are absent, such as in imitation learning (IL) pre-training. To address these challenges, we propose a method for efficient online RL fine-tuning using solely the offline pre-trained policy, eliminating reliance on pre-trained Q-functions. We introduce PORL (Policy-Only Reinforcement Learning Fine-Tuning), which rapidly initializes the Q-function from scratch during the online phase to avoid detrimental pessimism. Our method not only achieves competitive performance with advanced offline-to-online RL algorithms and online RL approaches that leverage data or policies prior, but also pioneers a new path for directly fine-tuning behavior cloning (BC) policies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。