CCLF: A Contrastive-Curiosity-Driven Learning Framework for Sample-Efficient Reinforcement Learning

Abstract

In reinforcement learning (RL), it is challenging to learn directly from high-dimensional observations, where data augmentation has recently been shown to remedy this via encoding invariances from raw pixels. Nevertheless, we empirically find that not all samples are equally important and hence simply injecting more augmented inputs may instead cause instability in Q-learning. In this paper, we approach this problem systematically by developing a model-agnostic Contrastive-Curiosity-Driven Learning Framework (CCLF), which can fully exploit sample importance and improve learning efficiency in a self-supervised manner. Facilitated by the proposed contrastive curiosity, CCLF is capable of prioritizing the experience replay, selecting the most informative augmented inputs, and more importantly regularizing the Q-function as well as the encoder to concentrate more on under-learned data. Moreover, it encourages the agent to explore with a curiosity-based reward. As a result, the agent can focus on more informative samples and learn representation invariances more efficiently, with significantly reduced augmented inputs. We apply CCLF to several base RL algorithms and evaluate on the DeepMind Control Suite, Atari, and MiniGrid benchmarks, where our approach demonstrates superior sample efficiency and learning performances compared with other state-of-the-art methods.

Full paper with supplementary material, accepted by IJCAI 2022. Acknowledgements and affiliations are updated

References in corpus (19)

• Adam: A Method for Stochastic Optimization
• Representation Learning with Contrastive Predictive Coding
• Prioritized Experience Replay
• Soft Actor-Critic Algorithms and Applications
• Exact solutions to the nonlinear dynamics of learning in deep linear neural networks
• Noisy Networks for Exploration
• Model-Based Reinforcement Learning for Atari
• Reinforcement Learning with Augmented Data
• dm_control: Software and Tasks for Continuous Control
• A survey on intrinsic motivation in reinforcement learning
• Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learning
• Automatic Data Augmentation for Generalization in Deep Reinforcement Learning
• Network Randomization: A Simple Technique for Generalization in Deep Reinforcement Learning
• Improving Generalization in Reinforcement Learning with Mixture Regularization
• Curiosity-Driven Experience Prioritization via Density Estimation
• Sample-efficient Reinforcement Learning Representation Learning with Curiosity Contrastive Forward Dynamics Model
• From Psychological Curiosity to Artificial Curiosity: Curiosity-Driven Learning in Artificial Intelligence Tasks
• Sample Efficient Reinforcement Learning In Continuous State Spaces: A Perspective Beyond Linearity
• Towards More Sample Efficiency in Reinforcement Learning with Data Augmentation