arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.09443cs.LGcs.AIstat.ML
收藏

RL中推广的深集

Deep Sets for Generalization in RL

INRIA(法国国家信息与自动化研究所)

Tristan Karch, Cédric Colas, Laetitia Teodorescu, Clément Moulin-Frier, Pierre-Yves Oudeyer

摘要:

本文研究了在语言引导的强化学习代理的奖励函数和策略架构的设计中编码以对象为中心的表示的思想。这是使用从深度集和门控注意力机制中获得灵感的逐对象排列不变网络的组合来完成的。在一个二维程序生成的世界中,代理的目标在自然语言导航和与对象交互,我们表明,这些架构表现出强大的泛化能力的分布目标。我们研究了在测试时对不同数量的对象的泛化,并进一步将以对象为中心的体系结构扩展到涉及关系推理的目标。

英文摘要:

This paper investigates the idea of encoding object-centered representations in the design of the reward function and policy architectures of a language-guided reinforcement learning agent. This is done using a combination of object-wise permutation invariant networks inspired from Deep Sets and gated-attention mechanisms. In a 2D procedurally-generated world where agents targeting goals in natural language navigate and interact with objects, we show that these architectures demonstrate strong generalization capacities to out-of-distribution goals. We study the generalization to varying numbers of objects at test time and further extend the object-centered architectures to goals involving relational reasoning.

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

↑