arXiv:2411.01425cs.LGcs.AI2024-11被引 1

让强化学习自动发现有顺序要求的隐藏任务步骤。

Learning Hidden Subgoals under Temporal Ordering Constraints in Reinforcement Learning

  • 用对比学习和时间几何采样同时识别隐藏目标与顺序
  • 构建子目标树实现按序高效发现,提升样本效率
  • 适用于图像输入的复杂任务,可推广到未见场景

在现实应用中,完成任务的成功依赖于多个时间上相隔较远、且必须按固定顺序达成的关键步骤。例如烹饪食谱中的每一步都需按序执行。这些关键步骤可视为任务的子目标,其时间顺序称为时间顺序约束。然而,在许多实际问题中,子目标或关键状态隐藏在状态空间中,且其时间顺序未知,导致现有强化学习算法难以应对。为此,本文提出一种新算法LSTOC(Learning Hidden Subgoals under Temporal Ordering Constraints),通过首次到达表示与时间几何采样,设计新的对比学习目标,实现对隐藏子目标及其时间顺序的联合学习。进一步提出一种基于子目标树的样本高效学习策略,按顺序逐个发现子目标,有效提升轨迹收集效率,加快任务求解,并具备对未见任务的泛化能力。LSTOC在多个基于图像观测的环境中评估,显著优于基线方法。

原文摘要 · Abstract (English)

In real-world applications, the success of completing a task is often determined by multiple key steps which are distant in time steps and have to be achieved in a fixed time order. For example, the key steps listed on the cooking recipe should be achieved one-by-one in the right time order. These key steps can be regarded as subgoals of the task and their time orderings are described as temporal ordering constraints. However, in many real-world problems, subgoals or key states are often hidden in the state space and their temporal ordering constraints are also unknown, which make it challenging for previous RL algorithms to solve this kind of tasks. In order to address this issue, in this work we propose a novel RL algorithm for {\bf l}earning hidden {\bf s}ubgoals under {\bf t}emporal {\bf o}rdering {\bf c}onstraints (LSTOC). We propose a new contrastive learning objective which can effectively learn hidden subgoals (key states) and their temporal orderings at the same time, based on first-occupancy representation and temporal geometric sampling. In addition, we propose a sample-efficient learning strategy to discover subgoals one-by-one following their temporal order constraints by building a subgoal tree to represent discovered subgoals and their temporal ordering relationships. Specifically, this tree can be used to improve the sample efficiency of trajectory collection, fasten the task solving and generalize to unseen tasks. The LSTOC framework is evaluated on several environments with image-based observations, showing its significant improvement over baseline methods.

强化学习子目标发现顺序约束对比学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。