arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.09686cs.AIcs.SYeess.SY
收藏

通过隐式规划实现带逻辑规约的组合任务的泛化

Generalization of Compositional Tasks with Logical Specification via Implicit Planning

Georgia Institute of Technology(佐治亚理工学院)

Duo Xu, Faramarz Fekri

AI总结:

针对逻辑规约定义的组合任务强化学习泛化难、收敛慢的问题,提出带隐式规划器的分层RL框架,用GNN在潜空间规划并引导低层策略,在效率和最优性上均优于现有方法。

摘要:

本研究旨在解决为逻辑规约定义的组合任务学习可泛化策略的挑战。这类任务由多个时间维度上延伸的子任务构成。由于长程任务中存在子任务相互依赖及奖励稀疏的问题,现有的强化学习(RL)方法,如任务条件策略和目标条件策略,在泛化到组合任务时仍面临收敛缓慢和性能次优的困境。为克服这些局限,我们提出了一种全新的分层强化学习框架,以提升任务泛化的效率和最优性。在高层,我们设计了专门用于组合任务泛化的隐式规划器,该规划器选择下一个子任务,并估计从当前状态完成剩余任务的多步回报。它学习一个潜在转移模型,并利用图神经网络(GNN)在潜在空间中执行规划。随后,高层规划器选定的子任务引导低层智能体有效处理长程任务,同时多步回报促使低层策略考虑未来子任务的依赖关系,从而提升其最优性。我们开展了全面的实验,证明该框架在效率和最优性两方面均优于现有方法。

英文摘要:

In this study, we address the challenge of learning generalizable policies for compositional tasks defined by logical specifications. These tasks consist of multiple temporally extended sub-tasks. Due to the sub-task inter-dependencies and sparse reward issue in long-horizon tasks, existing reinforcement learning (RL) approaches, such as task-conditioned and goal-conditioned policies, continue to struggle with slow convergence and sub-optimal performance in generalizing to compositional tasks. To overcome these limitations, we introduce a new hierarchical RL framework that enhances the efficiency and optimality of task generalization. At the high level, we present an implicit planner specifically designed for generalizing compositional tasks. This planner selects the next sub-task and estimates the multi-step return for completing the remaining task to complete from the current state. It learns a latent transition model and performs planning in the latent space by using a graph neural network (GNN). Subsequently, the high-level planner's selected sub-task guides the low-level agent to effectively handle long-horizon tasks, while the multi-step return encourages the low-level policy to account for future sub-task dependencies, enhancing its optimality. We conduct comprehensive experiments to demonstrate the framework's advantages over previous methods in terms of both efficiency and optimality.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑