用动态规划改进稀疏奖励下的目标导向强化学习,提升效率与鲁棒性。
Q-WSL: Optimizing Goal-Conditioned RL with Weighted Supervised Learning via Dynamic Programming
- 结合Q-learning动态规划思想优化加权监督学习策略
- 在复杂任务中实现更高性能与样本效率,优于现有方法
- 适合处理稀疏奖励和随机环境的强化学习研究者
一种新型算法——目标导向加权监督学习(GCWSL)近年来被提出,用于应对目标导向强化学习中的稀疏奖励问题。其凭借简单、高效和稳定,在多种目标达成任务中表现优异。然而,GCWSL缺乏轨迹拼接能力,当测试时遇到未见过的技能时难以学习最优策略,尤其在回放缓冲区主要包含次优轨迹时更为明显。相比之下,基于TD的强化学习方法如Q-learning利用动态规划,虽无此问题但常因价值函数近似困难而出现不稳定性。本文提出Q-WSL框架,融合动态规划优势,通过回放缓冲区内不同轨迹上的(状态, 目标)对输出最优动作,实现Q-learning与GCWSL的优势互补,有效缓解各自缺陷。在具有挑战性的目标达成任务上,实验表明Q-WSL在性能与样本效率方面均超越其他目标导向方法,并在二元奖励结构和环境随机性下表现出显著鲁棒性。
原文摘要 · Abstract (English)
A novel class of advanced algorithms, termed Goal-Conditioned Weighted Supervised Learning (GCWSL), has recently emerged to tackle the challenges posed by sparse rewards in goal-conditioned reinforcement learning (RL). GCWSL consistently delivers strong performance across a diverse set of goal-reaching tasks due to its simplicity, effectiveness, and stability. However, GCWSL methods lack a crucial capability known as trajectory stitching, which is essential for learning optimal policies when faced with unseen skills during testing. This limitation becomes particularly pronounced when the replay buffer is predominantly filled with sub-optimal trajectories. In contrast, traditional TD-based RL methods, such as Q-learning, which utilize Dynamic Programming, do not face this issue but often experience instability due to the inherent difficulties in value function approximation. In this paper, we propose Q-learning Weighted Supervised Learning (Q-WSL), a novel framework designed to overcome the limitations of GCWSL by incorporating the strengths of Dynamic Programming found in Q-learning. Q-WSL leverages Dynamic Programming results to output the optimal action of (state, goal) pairs across different trajectories within the replay buffer. This approach synergizes the strengths of both Q-learning and GCWSL, effectively mitigating their respective weaknesses and enhancing overall performance. Empirical evaluations on challenging goal-reaching tasks demonstrate that Q-WSL surpasses other goal-conditioned approaches in terms of both performance and sample efficiency. Additionally, Q-WSL exhibits notable robustness in environments characterized by binary reward structures and environmental stochasticity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。