模仿人类思维规划,提升强化学习在新环境中的泛化能力。
Brain-Inspired Planning for Better Generalization in Reinforcement Learning
- 用顶层注意力机制实现决策时动态聚焦关键状态
- 框架可自动拆分复杂任务,提升长期规划鲁棒性
- 引入可行性评估器,避免幻觉目标导致的错误规划
现有强化学习系统在真实场景中面临严重泛化挑战,主要源于训练环境外的性能退化。本文受人类意识规划行为启发,提出一种自上而下的注意力机制,使智能体在决策时能动态聚焦环境状态中与当前意图相关的部分,称为“空间抽象”,显著提升零样本系统性泛化能力。在此基础上,构建了Skipper框架,自动将复杂任务分解为更易处理的子任务,通过关注环境中的时空相关元素,增强对分布偏移的鲁棒性及长期组合规划效率。此外,发现依赖生成模型生成状态目标的规划智能体普遍存在盲信幻觉目标的问题,导致误判和安全风险。受人脑拒绝妄想意图的启发,提出学习可行性评估器以拒绝不可行的目标,大幅改善多种规划智能体的表现。最后,展望未来研究方向,旨在实现通用任务抽象与完全抽象规划。
原文摘要 · Abstract (English)
Existing Reinforcement Learning (RL) systems encounter significant challenges when applied to real-world scenarios, primarily due to poor generalization across environments that differ from their training conditions. This thesis explores the direction of enhancing agents' zero-shot systematic generalization abilities by granting RL agents reasoning behaviors that are found to help systematic generalization in the human brain. Inspired by human conscious planning behaviors, we first introduced a top-down attention mechanism, which allows a decision-time planning agent to dynamically focus its reasoning on the most relevant aspects of the environmental state given its instantaneous intentions, a process we call "spatial abstraction". This approach significantly improves systematic generalization outside the training tasks. Subsequently, building on spatial abstraction, we developed the Skipper framework to automatically decompose complex tasks into simpler, more manageable sub-tasks. Skipper provides robustness against distributional shifts and efficacy in long-term, compositional planning by focusing on pertinent spatial and temporal elements of the environment. Finally, we identified a common failure mode and safety risk in planning agents that rely on generative models to generate state targets during planning. It is revealed that most agents blindly trust the targets they hallucinate, resulting in delusional planning behaviors. Inspired by how the human brain rejects delusional intentions, we propose learning a feasibility evaluator to enable rejecting hallucinated infeasible targets, which led to significant performance improvements in various kinds of planning agents. Finally, we suggest directions for future research, aimed at achieving general task abstraction and fully enabling abstract planning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。