Efficient decorrelation of features using Gramian in Reinforcement Learning

Abstract

Learning good representations is a long standing problem in reinforcement learning (RL). One of the conventional ways to achieve this goal in the supervised setting is through regularization of the parameters. Extending some of these ideas to the RL setting has not yielded similar improvements in learning. In this paper, we develop an online regularization framework for decorrelating features in RL and demonstrate its utility in several test environments. We prove that the proposed algorithm converges in the linear function approximation setting and does not change the main objective of maximizing cumulative reward. We demonstrate how to scale the approach to deep RL using the Gramian of the features achieving linear computational complexity in the number of features and squared complexity in size of the batch. We conduct an extensive empirical study of the new approach on Atari 2600 games and show a significant improvement in sample efficiency in 40 out of 49 games.

References in corpus (6)

• Understanding deep learning requires rethinking generalization
• Rainbow: Combining Improvements in Deep Reinforcement Learning
• Discovering Hidden Factors of Variation in Deep Networks
• Investigating Generalisation in Continuous Deep Reinforcement Learning
• Using PCA to Efficiently Represent State Spaces
• Deep Reinforcement Learning with Decorrelation