arXiv:2412.06685cs.LGcs.AI2024-12被引 79

无需修改框架,一套方法可训练各类策略模型。

Policy Agnostic RL: Offline RL and Online RL Fine-Tuning of Any Class and Backbone

  • 用优化动作替代传统策略更新,适配不同策略类型。
  • 在真实机器人上40分钟将成功率从40%提升至70%。
  • 适合想快速迁移或微调复杂策略的研究者与工程师。

近期决策策略学习的进展主要源于使用表达能力强的策略模型,通常通过模仿学习实现。然而,模仿学习会丢弃非专家数据,而强化学习(RL)仍可利用次优数据。但新策略类别的训练常面临挑战:大多数深度RL工具与特定策略类别和主干网络绑定,导致更换策略后性能下降。例如,SAC依赖高斯策略的低方差重参数化梯度,对扩散策略不稳定,对自回归分类策略不可行。为此,我们提出无策略依赖的强化学习(PA-RL),可在不改变框架的前提下有效训练多种策略类别,包括不同架构和规模的扩散策略与Transformer策略。核心思想是:只要在“优化过的动作”上应用通用监督学习损失,即可替代RL中的策略改进步骤。通过从基础策略采样多个动作,并结合全局优化(用Q函数重排序动作样本)与局部优化(对动作样本进行梯度更新)最大化评论家得分,获取优化动作。PA-RL实现了完全基于演员-评论家的端到端训练,支持连续动作与自回归令牌输出。相比现有离线与在线微调方法,性能与样本效率最高提升2倍。首次实现以Cal-QL在线微调算法自主微调70亿参数的OpenVLA通用机器人策略,在真实世界中40分钟内成功率从40%提升至70%。

原文摘要 · Abstract (English)

Recent advances in learning decision-making policies can largely be attributed to training expressive policy models, largely via imitation learning. While imitation learning discards non-expert data, reinforcement learning (RL) can still learn from suboptimal data. However, instantiating RL training of a new policy class often presents a different challenge: most deep RL machinery is co-developed with assumptions on the policy class and backbone, resulting in poor performance when the policy class changes. For instance, SAC utilizes a low-variance reparameterization policy gradient for Gaussian policies, but this is unstable for diffusion policies and intractable for autoregressive categorical policies. To address this issue, we develop an offline RL and online fine-tuning approach called policy-agnostic RL (PA-RL) that can effectively train multiple policy classes, with varying architectures and sizes. We build off the basic idea that a universal supervised learning loss can replace the policy improvement step in RL, as long as it is applied on "optimized" actions. To obtain these optimized actions, we first sample multiple actions from a base policy, and run global optimization (i.e., re-ranking multiple action samples using the Q-function) and local optimization (i.e., running gradient steps on an action sample) to maximize the critic on these candidates. PA-RL enables fine-tuning diffusion and transformer policies with either autoregressive tokens or continuous action outputs, at different sizes, entirely via actor-critic RL. Moreover, PA-RL improves the performance and sample-efficiency by up to 2 times compared to existing offline RL and online fine-tuning methods. We show the first result that successfully fine-tunes OpenVLA, a 7B generalist robot policy, autonomously with Cal-QL, an online RL fine-tuning algorithm, improving from 40% to 70% in the real world in 40 minutes.

强化学习策略迁移机器人控制模型微调

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