用纯强化学习梯度在线微调决策变换器,突破了传统方法依赖监督学习的瓶颈。
Online Finetuning Decision Transformers with Pure RL Gradients
- 提出基于GRPO的纯强化学习更新策略,避免监督目标干扰
- 在多个基准上达到新SOTA,性能优于现有在线微调方法
- 适合追求高效在线学习与稳定训练的强化学习研究者
决策变换器(DTs)通过将离线强化学习建模为序列建模问题,成为序列决策的强大框架。然而,如何在纯强化学习梯度下实现在线微调仍缺乏探索,现有方法在在线阶段仍严重依赖监督学习目标。我们发现,事后回报重标注——在线DT中的标准组件——是阻碍纯强化学习微调的关键障碍:虽然对监督学习有益,但与基于重要性采样的强化学习算法(如GRPO)不兼容,导致训练不稳定。基于此洞察,我们提出新算法,支持使用纯强化学习梯度对决策变换器进行在线微调。通过将GRPO适配至DT框架,并引入子轨迹优化以改进信用分配、序列级似然目标以提升稳定性和效率、主动采样以鼓励在不确定区域探索。大量实验表明,所提方法优于现有在线DT基线,在多个基准上达到新状态最优表现,验证了纯强化学习梯度在决策变换器在线微调中的有效性。
原文摘要 · Abstract (English)
Decision Transformers (DTs) have emerged as a powerful framework for sequential decision making by formulating offline reinforcement learning (RL) as a sequence modeling problem. However, extending DTs to online settings with pure RL gradients remains largely unexplored, as existing approaches continue to rely heavily on supervised sequence-modeling objectives during online finetuning. We identify hindsight return relabeling -- a standard component in online DTs -- as a critical obstacle to RL-based finetuning: while beneficial for supervised learning, it is fundamentally incompatible with importance sampling-based RL algorithms such as GRPO, leading to unstable training. Building on this insight, we propose new algorithms that enable online finetuning of Decision Transformers using pure reinforcement learning gradients. We adapt GRPO to DTs and introduce several key modifications, including sub-trajectory optimization for improved credit assignment, sequence-level likelihood objectives for enhanced stability and efficiency, and active sampling to encourage exploration in uncertain regions. Through extensive experiments, we demonstrate that our methods outperform existing online DT baselines and achieve new state-of-the-art performance across multiple benchmarks, highlighting the effectiveness of pure-RL-based online finetuning for Decision Transformers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。