用自改进的搜索框架让模型学会通用解题,零样本泛化能力突破极限。
Learning to Search and Searching to Learn for Generalization in Planning
- 用图神经网络做启发式,结合WA*搜索循环优化策略
- 在488个积木的Blocksworld任务中零样本直接求解成功
- 适合研究通用规划与强化学习泛化问题的研究者
组合泛化仍是深度强化学习的核心挑战。经典规划通过显式关系描述提供了一个简单而富有挑战性的研究场景,无需从感知中学习。在稀疏奖励环境中,传统RL实时搜索探索效果差,基于学习的规划方法常依赖专家示范、事后重标记或从目标状态随机游走。相比之下,规划器依赖A*类最佳优先搜索从头求解。本文提出一种自改进的WA*学习框架,结合由关系图神经网络表示的价值启发式:启发式引导搜索,搜索数据再通过Q-learning更新启发式。该循环生成可作为通用策略的启发式,在无搜索情况下解决新实例,而传统DRL在此失效。我们在Sokoban、PushWorld、The Witness及2023国际规划竞赛基准上验证了该方法。尤为关键的是,训练于少于30个积木的Blocksworld实例的启发式,能零样本求解含488个积木的新实例。
原文摘要 · Abstract (English)
Combinatorial generalization remains a central challenge in Deep Reinforcement Learning (DRL). Classical planning provides a simple yet challenging setting to study this problem through explicit relational descriptions, without requiring learning from perception. In sparse-reward domains, standard RL exploration via real-time search is ineffective, and learning-based planning methods often rely on expert demonstrations, hindsight relabeling, or random walks from the goal state. In contrast, planners rely on best-first search methods such as $\mathrm{A}^\star$ to solve problems from scratch. We propose a self-improving $\mathrm{WA}^\star$ learning framework in combination with a value heuristic represented by a Relational Graph Neural Network: the heuristic guides search, and the resulting search data updates the heuristic via $Q$-learning. This loop yields heuristics that can function as general policies and solve new instances even without search, where DRL otherwise fails, as we show on puzzles such as Sokoban, PushWorld, The Witness, and the 2023 International Planning Competition benchmarks. Notably, we demonstrate strong zero-shot generalization: For example, heuristics trained on Blocksworld instances with fewer than 30 blocks successfully solve instances with 488 blocks without search.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。