arXiv:2412.18855cs.LG2024-12NeurIPS被引 12

提出通用离线到在线强化学习方法,解决预训练策略迁移难题

Optimistic Critic Reconstruction and Constrained Fine-Tuning for General Offline-to-Online RL

  • 用乐观重评与校准修正离线评价偏差
  • 通过约束微调应对在线学习分布偏移
  • 适配任意离线方法,提升多任务在线性能

离线到在线(O2O)强化学习通过使用离线预训练策略作为初始化,可在少量在线交互中快速提升性能。现有方法通常针对特定离线RL算法设计微调策略,无法实现从任意离线方法到任意在线方法的通用迁移。本文揭示了离线数据集与在线环境间存在评估与改进不匹配问题,阻碍了预训练策略的直接应用。为此,我们提出同时处理这两类不匹配的方法:在在线微调前,以乐观方式重新评估离线训练的悲观价值函数,并利用可靠的离线行为策略校准错位的价值函数,避免错误更新;获得乐观且对齐的价值函数后,采用约束微调对抗在线学习中的分布偏移。实验表明,该方法在多个模拟任务上相比当前最优方法实现了稳定且高效的性能提升。

原文摘要 · Abstract (English)

Offline-to-online (O2O) reinforcement learning (RL) provides an effective means of leveraging an offline pre-trained policy as initialization to improve performance rapidly with limited online interactions. Recent studies often design fine-tuning strategies for a specific offline RL method and cannot perform general O2O learning from any offline method. To deal with this problem, we disclose that there are evaluation and improvement mismatches between the offline dataset and the online environment, which hinders the direct application of pre-trained policies to online fine-tuning. In this paper, we propose to handle these two mismatches simultaneously, which aims to achieve general O2O learning from any offline method to any online method. Before online fine-tuning, we re-evaluate the pessimistic critic trained on the offline dataset in an optimistic way and then calibrate the misaligned critic with the reliable offline actor to avoid erroneous update. After obtaining an optimistic and and aligned critic, we perform constrained fine-tuning to combat distribution shift during online learning. We show empirically that the proposed method can achieve stable and efficient performance improvement on multiple simulated tasks when compared to the state-of-the-art methods.

强化学习离线到在线策略迁移

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