用大模型生成有序子目标,让离线强化学习更高效完成长任务
STO-RL: Offline RL under Sparse Rewards via LLM-Guided Subgoal Temporal Order
- 用大模型生成有时间顺序的子目标序列,构建状态到子目标阶段的映射
- 通过基于势函数的奖励设计,将稀疏奖励转化为密集且时序一致的信号
- 在多个稀疏奖励任务上表现优于现有方法,对错误子目标也有鲁棒性
离线强化学习(Offline RL)可从预收集数据中学习策略,避免在线交互的成本与风险,但在处理长时序、稀疏奖励任务时仍面临挑战。现有目标条件化和层次化离线RL方法虽分解任务并生成中间奖励,但常忽略子目标间的时序依赖,且依赖不精确的奖励塑造,导致策略性能不佳。为此,我们提出STO-RL(基于大模型引导子目标时序结构的离线强化学习),利用大语言模型(LLMs)生成具有时间顺序的子目标序列及对应的状态-子目标阶段映射。基于该时序结构,STO-RL采用基于势函数的奖励塑造,将稀疏的终态奖励转化为密集、时序一致的信号,促进子目标进展的同时避免次优解。由此生成的增强数据集支持高效离线训练高性能策略。在四个离散与连续稀疏奖励基准上的评估表明,STO-RL持续优于当前最优的离线目标条件化与层次化RL基线,实现更快收敛、更高成功率与更短轨迹。消融实验进一步验证其对不完美或噪声子目标序列的鲁棒性,证明大模型引导的子目标时序结构结合理论严谨的奖励塑造,为长时序离线强化学习提供了实用且可扩展的解决方案。
原文摘要 · Abstract (English)
Offline reinforcement learning (RL) enables policy learning from pre-collected datasets, avoiding costly and risky online interactions, but it often struggles with long-horizon tasks involving sparse rewards. Existing goal-conditioned and hierarchical offline RL methods decompose such tasks and generate intermediate rewards to mitigate limitations of traditional offline RL, but usually overlook temporal dependencies among subgoals and rely on imprecise reward shaping, leading to suboptimal policies. To address these issues, we propose STO-RL (Offline RL using LLM-Guided Subgoal Temporal Order), an offline RL framework that leverages large language models (LLMs) to generate temporally ordered subgoal sequences and corresponding state-to-subgoal-stage mappings. Using this temporal structure, STO-RL applies potential-based reward shaping to transform sparse terminal rewards into dense, temporally consistent signals, promoting subgoal progress while avoiding suboptimal solutions. The resulting augmented dataset with shaped rewards enables efficient offline training of high-performing policies. Evaluations on four discrete and continuous sparse-reward benchmarks demonstrate that STO-RL consistently outperforms state-of-the-art offline goal-conditioned and hierarchical RL baselines, achieving faster convergence, higher success rates, and shorter trajectories. Ablation studies further confirm STO-RL's robustness to imperfect or noisy LLM-generated subgoal sequences, demonstrating that LLM-guided subgoal temporal structures combined with theoretically grounded reward shaping provide a practical and scalable solution for long-horizon offline RL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。