提出ASTRO框架,用动态引导拼接轨迹提升离线强化学习性能
ASTRO: Adaptive Stitching via Dynamics-Guided Trajectory Rollouts
- 通过时序距离表征识别可到达的拼接目标点
- 利用滚动偏差反馈自适应生成动作序列,确保轨迹符合动态规律
- 在OGBench和D4RL上显著提升策略学习效果,适合复杂环境离线训练
离线强化学习依赖预收集数据集学习最优策略,但其中包含次优和碎片化轨迹,导致奖励传播困难,价值估计不准,策略性能下降。现有生成模型的轨迹拼接方法常受限于行为策略支持范围或违背系统动力学,难以有效改进策略。本文提出ASTRO框架,通过学习时序距离表示识别可到达的拼接目标,并设计动态引导的拼接规划器,基于预测动作与实际到达状态间的滚动偏差反馈(Rollout Deviation Feedback),自适应生成符合动力学特性的连接动作序列,从而实现分布新颖且动力学一致的轨迹增强。该方法显著提升了拼接可行性与可达性,有效促进策略学习。在多个算法上优于现有离线强化学习增强方法,在挑战性OGBench基准上表现突出,并在标准D4RL基准上实现持续提升。
原文摘要 · Abstract (English)
Offline reinforcement learning (RL) enables agents to learn optimal policies from pre-collected datasets. However, datasets containing suboptimal and fragmented trajectories present challenges for reward propagation, resulting in inaccurate value estimation and degraded policy performance. While trajectory stitching via generative models offers a promising solution, existing augmentation methods frequently produce trajectories that are either confined to the support of the behavior policy or violate the underlying dynamics, thereby limiting their effectiveness for policy improvement. We propose ASTRO, a data augmentation framework that generates distributionally novel and dynamics-consistent trajectories for offline RL. ASTRO first learns a temporal-distance representation to identify distinct and reachable stitch targets. We then employ a dynamics-guided stitch planner that adaptively generates connecting action sequences via Rollout Deviation Feedback, defined as the gap between target state sequence and the actual arrived state sequence by executing predicted actions, to improve trajectory stitching's feasibility and reachability. This approach facilitates effective augmentation through stitching and ultimately enhances policy learning. ASTRO outperforms prior offline RL augmentation methods across various algorithms, achieving notable performance gain on the challenging OGBench suite and demonstrating consistent improvements on standard offline RL benchmarks such as D4RL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。