用动态规划优化动作价值,让离线强化学习更稳更准
Q-value Regularized Decision ConvFormer for Offline Reinforcement Learning
- 用卷积Transformer建模轨迹,结合动态规划优化动作价值
- 在D4RL所有环境中表现优异,尤其擅长拼接低质量轨迹
- 适合研究离线强化学习和轨迹生成的从业者
作为数据驱动范式,离线强化学习(Offline RL)被建模为序列建模问题,决策变换器(Decision Transformer, DT)展现出卓越能力。与传统方法不同,DT基于期望回报、历史状态和动作,利用因果掩码Transformer生成最优动作。然而,单条轨迹内采样回报与多条轨迹间最优回报不一致,导致难以设定合理回报以生成最优动作并拼接次优轨迹。决策卷积变换器(DC)在马尔可夫决策过程上下文中对轨迹建模更直观。本文提出带Q值正则化的决策卷积变换器(QDC),融合了DC对轨迹的理解,并在训练中引入基于动态规划的动作值最大化项,确保采样动作的期望回报与全局最优回报一致。QDC在D4RL基准上表现卓越,在所有测试环境均达到或接近最优水平,尤其在轨迹拼接能力方面表现突出。
原文摘要 · Abstract (English)
As a data-driven paradigm, offline reinforcement learning (Offline RL) has been formulated as sequence modeling, where the Decision Transformer (DT) has demonstrated exceptional capabilities. Unlike previous reinforcement learning methods that fit value functions or compute policy gradients, DT adjusts the autoregressive model based on the expected returns, past states, and actions, using a causally masked Transformer to output the optimal action. However, due to the inconsistency between the sampled returns within a single trajectory and the optimal returns across multiple trajectories, it is challenging to set an expected return to output the optimal action and stitch together suboptimal trajectories. Decision ConvFormer (DC) is easier to understand in the context of modeling RL trajectories within a Markov Decision Process compared to DT. We propose the Q-value Regularized Decision ConvFormer (QDC), which combines the understanding of RL trajectories by DC and incorporates a term that maximizes action values using dynamic programming methods during training. This ensures that the expected returns of the sampled actions are consistent with the optimal returns. QDC achieves excellent performance on the D4RL benchmark, outperforming or approaching the optimal level in all tested environments. It particularly demonstrates outstanding competitiveness in trajectory stitching capability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。