arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.04440cs.AIcs.LG
收藏

作为自模型进行规划的对偶策略

Dual policy as self-model for planning

Massachusetts Institute of Technology(麻省理工学院)

Jaesung Yoo, Fernanda de la Torre, Guangyu Robert Yang

AI总结:

受强化学习与神经科学启发,提出对偶策略智能体,使用蒸馏策略网络作为自模型进行规划,实验证明该方法能稳定训练、加速推理并促进探索,代价是需额外蒸馏新网络。

摘要:

规划是一种数据高效的决策策略,智能体通过探索可能的未来状态来选择候选动作。当存在高维动作空间时,为了模拟未来状态,必须利用自身决策策略的知识来限制待探索的动作数量。我们将用于模拟自身决策的模型称为智能体的自模型。尽管自模型通常与世界模型结合广泛用于隐式规划动作,但如何设计自模型仍不明确。受当前强化学习方法和神经科学的启发,我们探索了使用蒸馏策略网络作为自模型的益处与局限性。在此类对偶策略智能体中,无模型策略和蒸馏策略分别用于无模型动作和规划动作。我们在一个具有生态相关性的参数化环境上的结果表明,用于自模型的蒸馏策略网络能够稳定训练,比使用无模型策略具有更快的推理速度,促进更好的探索,并且能够学习对自身行为的全面理解,其代价是在无模型策略之外还需蒸馏一个新网络。

英文摘要:

Planning is a data efficient decision-making strategy where an agent selects candidate actions by exploring possible future states. To simulate future states when there is a high-dimensional action space, the knowledge of one's decision making strategy must be used to limit the number of actions to be explored. We refer to the model used to simulate one's decisions as the agent's self-model. While self-models are implicitly used widely in conjunction with world models to plan actions, it remains unclear how self-models should be designed. Inspired by current reinforcement learning approaches and neuroscience, we explore the benefits and limitations of using a distilled policy network as the self-model. In such dual-policy agents, a model-free policy and a distilled policy are used for model-free actions and planned actions, respectively. Our results on a ecologically relevant, parametric environment indicate that distilled policy network for self-model stabilizes training, has faster inference than using model-free policy, promotes better exploration, and could learn a comprehensive understanding of its own behaviors, at the cost of distilling a new network apart from the model-free policy.

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

↑