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

Exchangeable Input Representations for Reinforcement Learning

Stanford University(斯坦福大学)

John Mern, Dorsa Sadigh, Mykel J. Kochenderfer

英文摘要:

Poor sample efficiency is a major limitation of deep reinforcement learning in many domains. This work presents an attention-based method to project neural network inputs into an efficient representation space that is invariant under changes to input ordering. We show that our proposed representation results in an input space that is a factor of $m!$ smaller for inputs of $m$ objects. We also show that our method is able to represent inputs over variable numbers of objects. Our experiments demonstrate improvements in sample efficiency for policy gradient methods on a variety of tasks. We show that our representation allows us to solve problems that are otherwise intractable when using naïve approaches.

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

↑