CRC-RL: A Novel Visual Feature Representation Architecture for Unsupervised Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
This paper addresses the problem of visual feature representation learning with an aim to improve the performance of end-to-end reinforcement learning (RL) models. Specifically, a novel architecture is proposed that uses a heterogeneous loss function, called CRC loss, to learn improved visual features which can then be used for policy learning in RL. The CRC-loss function is a combination of three individual loss functions, namely, contrastive, reconstruction and consistency loss. The feature representation is learned in parallel to the policy learning while sharing the weight updates through a Siamese Twin encoder model. This encoder model is augmented with a decoder network and a feature projection network to facilitate computation of the above loss components. Through empirical analysis involving latent feature visualization, an attempt is made to provide an insight into the role played by this loss function in learning new action-dependent features and how they are linked to the complexity of the problems being solved. The proposed architecture, called CRC-RL, is shown to outperform the existing state-of-the-art methods on the challenging Deep mind control suite environments by a significant margin thereby creating a new benchmark in this field.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 43 of its 58 references that are in the map. First posted on arXiv 31 Jan 2023.
Builds on
43 of 58 references.
References · 43
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2015Continuous control with deep reinforcement learningcited by 3,279
- 2013Playing Atari with Deep Reinforcement Learningcited by 3,211
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,218
- 2015Deep Reinforcement Learning with Double Q-learningcited by 1,704
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2018Soft Actor-Critic Algorithms and Applicationscited by 1,179
- 2019A Comprehensive Survey on Transfer Learningcited by 1,154
31 more
- 2020Meta-Learning in Neural Networks: A Surveycited by 944
- 2018DeepMind Control Suitecited by 935
- 2015End-to-End Training of Deep Visuomotor Policiescited by 847
- 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 745
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 672
- 2020Image Augmentation Is All You Need: Regularizing Deep Reinforcement Learning from Pixelscited by 573
- 2017A Brief Survey of Deep Reinforcement Learningcited by 466
- 2020Reinforcement Learning with Augmented Datacited by 452
- 2016Reinforcement Learning with Unsupervised Auxiliary Taskscited by 400
- 2016Benchmarking Deep Reinforcement Learning for Continuous Controlcited by 361
- 2016Target-driven Visual Navigation in Indoor Scenes using Deep Reinforcement Learningcited by 351
- 2016Deep Reinforcement Learning for Robotic Manipulation with Asynchronous Off-Policy Updatescited by 342
- 2019Improving Sample Efficiency in Model-Free Reinforcement Learning from Imagescited by 268
- 2020Decoupling Representation Learning from Reinforcement Learningcited by 263
- 2020Transfer Learning in Deep Reinforcement Learning: A Surveycited by 259
- 2019Revisiting Self-Supervised Visual Representation Learningcited by 250
- 2021Self-Supervised Representation Learning: Introduction, Advances and Challengescited by 147
- 2020Generalization in Reinforcement Learning by Soft Data Augmentationcited by 132
- 2019From Variational to Deterministic Autoencoderscited by 131
- 2018Visual Semantic Navigation using Scene Priorscited by 127
- 2019Observational Overfitting in Reinforcement Learningcited by 90
- 2020Automatic Data Augmentation for Generalization in Deep Reinforcement Learningcited by 71
- 2018Deep Reinforcement Learning for Vision-Based Robotic Grasping: A Simulated Comparative Evaluation of Off-Policy Methodscited by 32
- 2017CM-GANs: Cross-modal Generative Adversarial Networks for Common Representation Learningcited by 27
- 2020Robotic Grasping using Deep Reinforcement Learningcited by 27
- 2021Reinforcement Learning with Latent Flowcited by 18
- 2022Unsupervised Representation Learning in Deep Reinforcement Learning: A Reviewcited by 18
- 2016MARTA GANs: Unsupervised Representation Learning for Remote Sensing Image Classificationcited by 16
- 2018Intrinsically motivated reinforcement learning for human-robot interaction in the real-worldcited by 12
- 2018S-RL Toolbox: Environments, Datasets and Evaluation Metrics for State Representation Learningcited by 6
- 2018A Review of Meta-Reinforcement Learning for Deep Neural Networks Architecture Searchcited by 4
Built on by
Not cited yet.