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

Group Equivariant Deep Reinforcement Learning

Mcgill University(麦吉尔大学)

Arnab Kumar Mondal, Pratheeksha Nair, Kaleem Siddiqi

英文摘要:

In Reinforcement Learning (RL), Convolutional Neural Networks(CNNs) have been successfully applied as function approximators in Deep Q-Learning algorithms, which seek to learn action-value functions and policies in various environments. However, to date, there has been little work on the learning of symmetry-transformation equivariant representations of the input environment state. In this paper, we propose the use of Equivariant CNNs to train RL agents and study their inductive bias for transformation equivariant Q-value approximation. We demonstrate that equivariant architectures can dramatically enhance the performance and sample efficiency of RL agents in a highly symmetric environment while requiring fewer parameters. Additionally, we show that they are robust to changes in the environment caused by affine transformations.

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

↑