用优势加权重要性采样让平坦策略学会长程目标导航,无需分层结构。
Flattening Hierarchies with Policy Bootstrapping
- 用子目标条件策略自举训练扁平目标策略,避免复杂分层设计。
- 在多类任务中达到或超过现有最先进水平,可处理长程高维控制问题。
- 适合需要高效预训练通用策略的研究者和工程应用。
离线目标条件强化学习(GCRL)是利用大量无奖励轨迹预训练通用策略的有前景方法,类似于计算机视觉与自然语言处理中基础模型的自监督学习。然而,由于稀疏奖励与折扣效应的结合,将GCRL扩展到更长时序仍具挑战性,这使得远距离目标下基本动作的优势难以区分。分层强化学习虽在长程目标达成任务中表现优异,但其依赖模块化、时间尺度特定的策略及子目标生成,引入了显著复杂性,限制了在高维目标空间中的扩展。本文提出一种算法,通过优势加权重要性采样,从子目标条件策略自举训练扁平(非分层)目标条件策略。该方法无需对(子)目标空间构建生成模型,我们发现这是在大状态空间中实现高维控制可扩展性的关键。进一步表明,现有分层与自举方法均可视为本推导框架下的特定设计选择。在涵盖状态与像素级的移动与操作基准测试中,本方法达到或超越当前最优离线GCRL算法,并能扩展至先前方法失效的复杂长程任务。项目页面:https://johnlyzhou.github.io/saw/
原文摘要 · Abstract (English)
Offline goal-conditioned reinforcement learning (GCRL) is a promising approach for pretraining generalist policies on large datasets of reward-free trajectories, akin to the self-supervised objectives used to train foundation models for computer vision and natural language processing. However, scaling GCRL to longer horizons remains challenging due to the combination of sparse rewards and discounting, which obscures the comparative advantages of primitive actions with respect to distant goals. Hierarchical RL methods achieve strong empirical results on long-horizon goal-reaching tasks, but their reliance on modular, timescale-specific policies and subgoal generation introduces significant additional complexity and hinders scaling to high-dimensional goal spaces. In this work, we introduce an algorithm to train a flat (non-hierarchical) goal-conditioned policy by bootstrapping on subgoal-conditioned policies with advantage-weighted importance sampling. Our approach eliminates the need for a generative model over the (sub)goal space, which we find is key for scaling to high-dimensional control in large state spaces. We further show that existing hierarchical and bootstrapping-based approaches correspond to specific design choices within our derivation. Across a comprehensive suite of state- and pixel-based locomotion and manipulation benchmarks, our method matches or surpasses state-of-the-art offline GCRL algorithms and scales to complex, long-horizon tasks where prior approaches fail. Project page: https://johnlyzhou.github.io/saw/
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。