arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2201.12096cs.LG
收藏

基于掩码的潜在重构强化学习

Mask-based Latent Reconstruction for Reinforcement Learning

University of Science and Technology of China(中国科学技术大学);Microsoft Research Asia(微软亚洲研究院)

Tao Yu, Zhizheng Zhang, Cuiling Lan, Yan Lu, Zhibo Chen

摘要:

对于从像素进行深度强化学习(RL),学习有效的状态表示对于实现高性能至关重要。然而,在实践中,有限的经验和高维输入阻碍了有效的表示学习。为了解决这个问题,在其他研究领域的成功基于掩码的建模的动机,我们引入基于掩码的重建,以促进状态表示学习在RL。具体来说,我们提出了一个简单而有效的自我监督的方法,基于掩码的潜在重建(MLR),预测完整的状态表示在潜在的空间与空间和时间掩蔽像素的意见。MLR在学习状态表示时能够更好地使用上下文信息,使它们更具信息性,这有利于RL代理的训练。大量的实验表明,我们的MLR显着提高了RL的样本效率,并优于国家的最先进的样本效率RL方法在多个连续和离散控制基准。我们的代码可在https://github.com/microsoft/Mask-based-Latent-Reconstruction上获得。

英文摘要:

For deep reinforcement learning (RL) from pixels, learning effective state representations is crucial for achieving high performance. However, in practice, limited experience and high-dimensional inputs prevent effective representation learning. To address this, motivated by the success of mask-based modeling in other research fields, we introduce mask-based reconstruction to promote state representation learning in RL. Specifically, we propose a simple yet effective self-supervised method, Mask-based Latent Reconstruction (MLR), to predict complete state representations in the latent space from the observations with spatially and temporally masked pixels. MLR enables better use of context information when learning state representations to make them more informative, which facilitates the training of RL agents. Extensive experiments show that our MLR significantly improves the sample efficiency in RL and outperforms the state-of-the-art sample-efficient RL methods on multiple continuous and discrete control benchmarks. Our code is available at https://github.com/microsoft/Mask-based-Latent-Reconstruction.

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

↑