提出EXPO算法,让复杂策略在在线强化学习中更稳定高效。
EXPO: Stable Reinforcement Learning with Expressive Policies
- 用轻量编辑策略修正基础表达性策略的动作,避免梯度不稳
- 相比旧方法样本效率提升2-3倍,尤其在微调预训练模型时表现好
- 适合需要高表达力策略的在线强化学习任务,如复杂动作生成
我们研究在给定离线数据集的情况下,如何训练和微调具有表达力的在线强化学习(RL)策略。使用表达力强的策略(如扩散模型、流匹配模型)进行在线RL面临稳定价值最大化难题,因其长去噪链导致从动作到策略参数的梯度传播不稳定。我们的核心洞察是:可通过构建一个实时生成的策略来最大化Q值,而非直接对表达性策略优化价值。为此,我们提出表达性策略优化(EXPO),一种样本高效的在线RL算法,包含两个参数化策略:一个通过稳定模仿学习目标训练的大规模表达性基策略,以及一个轻量级高斯编辑策略,用于将基策略采样的动作调整至更高价值分布。该实时策略基于基策略动作与编辑后的动作,选择价值最高的动作用于采样和时序差分(TD)更新。实验显示,该方法在微调预训练策略及利用离线数据训练在线策略场景下,平均样本效率提升2-3倍。
原文摘要 · Abstract (English)
We study the problem of training and fine-tuning expressive policies with online reinforcement learning (RL) given an offline dataset. Training expressive policy classes with online RL present a unique challenge of stable value maximization. Unlike simpler Gaussian policies commonly used in online RL, expressive policies like diffusion and flow-matching policies are parameterized by a long denoising chain, which hinders stable gradient propagation from actions to policy parameters when optimizing against some value function. Our key insight is that we can address stable value maximization by avoiding direct optimization over value with the expressive policy and instead construct an on-the-fly RL policy to maximize Q-value. We propose Expressive Policy Optimization (EXPO), a sample-efficient online RL algorithm that utilizes an on-the-fly policy to maximize value with two parameterized policies -- a larger expressive base policy trained with a stable imitation learning objective and a light-weight Gaussian edit policy that edits the actions sampled from the base policy toward a higher value distribution. The on-the-fly policy optimizes the actions from the base policy with the learned edit policy and chooses the value maximizing action from the base and edited actions for both sampling and temporal-difference (TD) backup. Our approach yields up to 2-3x improvement in sample efficiency on average over prior methods both in the setting of fine-tuning a pretrained policy given offline data and in leveraging offline data to train online.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。