用Transformer实现离线策略的轨迹强化学习,提升机器人训练效率
TOP-ERL: Transformer-based Off-Policy Episodic Reinforcement Learning
- 将长动作序列分段,用Transformer critic评估每段价值
- 在复杂机器人环境上性能显著优于现有方法
- 适合需要高效探索的长期任务,如机器人控制
本文提出基于Transformer的离线策略轨迹强化学习(TOP-ERL),突破传统轨迹强化学习仅支持在线策略的限制。在轨迹强化学习中,策略生成多步动作序列而非单步动作,通常由运动基元(MP)参数化,可实现长程平滑探索并捕捉时序相关性。然而,由于难以评估完整动作序列的状态-动作值,现有方法多局限于在线策略框架,影响样本效率且无法使用更高效的离线策略结构。TOP-ERL通过将长动作序列分段,并结合n步回报估计与Transformer架构的评论家模型,实现对每段序列的价值评估,从而支持稳定高效的离线策略更新。在复杂机器人学习环境中的实验表明,该方法显著超越当前最优强化学习算法。充分的消融实验进一步验证了关键设计选择对性能的影响。
原文摘要 · Abstract (English)
This work introduces Transformer-based Off-Policy Episodic Reinforcement Learning (TOP-ERL), a novel algorithm that enables off-policy updates in the ERL framework. In ERL, policies predict entire action trajectories over multiple time steps instead of single actions at every time step. These trajectories are typically parameterized by trajectory generators such as Movement Primitives (MP), allowing for smooth and efficient exploration over long horizons while capturing high-level temporal correlations. However, ERL methods are often constrained to on-policy frameworks due to the difficulty of evaluating state-action values for entire action sequences, limiting their sample efficiency and preventing the use of more efficient off-policy architectures. TOP-ERL addresses this shortcoming by segmenting long action sequences and estimating the state-action values for each segment using a transformer-based critic architecture alongside an n-step return estimation. These contributions result in efficient and stable training that is reflected in the empirical results conducted on sophisticated robot learning environments. TOP-ERL significantly outperforms state-of-the-art RL methods. Thorough ablation studies additionally show the impact of key design choices on the model performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。