Improving Sample Efficiency in Model-Free Reinforcement Learning from Images
Not on the map. Why?
Cited by 23 in Offline RL and imitation learning from data, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Training an agent to solve control tasks directly from high-dimensional images with model-free reinforcement learning (RL) has proven difficult. A promising approach is to learn a latent representation together with the control policy. However, fitting a high-capacity encoder using a scarce reward signal is sample inefficient and leads to poor performance. Prior work has shown that auxiliary losses, such as image reconstruction, can aid efficient representation learning. However, incorporating reconstruction loss into an off-policy learning algorithm often leads to training instability. We explore the underlying reasons and identify variational autoencoders, used by previous investigations, as the cause of the divergence. Following these findings, we propose effective techniques to improve training stability. This results in a simple approach capable of matching state-of-the-art model-free and model-based algorithms on MuJoCo control tasks. Furthermore, our approach demonstrates robustness to observational noise, surpassing existing approaches in this setting. Code, results, and videos are anonymously available at https://sites.google.com/view/sac-ae/home.
Key facts. Cited by 268 papers in the map (data through 7 Oct 2026). First posted on arXiv 2 Oct 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 2; 2021: 56; 2022: 61; 2023: 57; 2024: 41; 2025: 37; 2026: 14.
268 papers.
Sort:
By area:
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 781
- 2021Domain Generalization: A Surveycited by 760
- 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 359
- 2022Masked Visual Pre-training for Motor Controlcited by 246
- 2022The Primacy Bias in Deep Reinforcement Learningcited by 242
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2021Mastering Atari Games with Limited Datacited by 235
- 2021Reinforcement Learning with Prototypical Representationscited by 179
- 2022Masked World Models for Visual Controlcited by 169
- 2021URLB: Unsupervised Reinforcement Learning Benchmarkcited by 138
- 2021Coarse-to-Fine Q-attention: Efficient Learning for Visual Robotic Manipulation via Discretisationcited by 120
38 more
- 2021State Entropy Maximization with Random Encoders for Efficient Explorationcited by 107
- 2022Reinforcement Learning with Action-Free Pre-Training from Videoscited by 107
- 2021Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentationcited by 101
- 2023UniDexGrasp++: Improving Dexterous Grasping Policy Learning via Geometry-aware Curriculum and Iterative Generalist-Specialist Learningcited by 101
- 2021Learning by Watching: Physical Imitation of Manipulation Skills from Human Videoscited by 87
- 2021S4RL: Surprisingly Simple Self-Supervision for Offline Reinforcement Learningcited by 85
- 2021Q-attention: Enabling Efficient Learning for Vision-based Robotic Manipulationcited by 85
- 2021Decoupling Value and Policy for Generalization in Reinforcement Learningcited by 81
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 73
- 2021Learning Vision-Guided Quadrupedal Locomotion End-to-End with Cross-Modal Transformerscited by 71
- 2022CIC: Contrastive Intrinsic Control for Unsupervised Skill Discoverycited by 66
- 2023When Do Transformers Shine in RL? Decoupling Memory from Credit Assignmentcited by 62
- 2021Learning Task Informed Abstractionscited by 58
- 2023TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learningcited by 56
- 2022On Pre-Training for Visuo-Motor Control: Revisiting a Learning-from-Scratch Baselinecited by 55
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Modelscited by 53
- 2024Bridging State and History Representations: Understanding Self-Predictive RLcited by 52
- 2024LLaRA: Supercharging Robot Learning Data for Vision-Language Policycited by 51
- 2022Look Closer: Bridging Egocentric and Third-Person Views with Transformers for Robotic Manipulationcited by 49
- 2021DRIVE: Deep Reinforced Accident Anticipation with Visual Explanationcited by 46
- 2022Learning Generalizable Dexterous Manipulation from Human Grasp Affordancecited by 46
- 2022Visual Reinforcement Learning with Self-Supervised 3D Representationscited by 45
- 2021MICo: Improved representations via sampling-based state similarity for Markov decision processescited by 43
- 2022Reinforcement Learning with Neural Radiance Fieldscited by 43
- 2022Visuo-Tactile Transformers for Manipulationcited by 43
- 2021Learning Markov State Abstractions for Deep Reinforcement Learningcited by 42
- 2021SECANT: Self-Expert Cloning for Zero-Shot Generalization of Visual Policiescited by 42
- 2023PLASTIC: Improving Input and Label Plasticity for Sample Efficient Reinforcement Learningcited by 42
- 2021PlayVirtual: Augmenting Cycle-Consistent Virtual Trajectories for Reinforcement Learningcited by 41
- 2021MADE: Exploration via Maximizing Deviation from Explored Regionscited by 40
- 2021AutoFL: Enabling Heterogeneity-Aware Energy Efficient Federated Learningcited by 39
- 2022Mask-based Latent Reconstruction for Reinforcement Learningcited by 39
- 2022Bisimulation Makes Analogies in Goal-Conditioned Reinforcement Learningcited by 38
- 2022Learning to Drive by Watching YouTube Videos: Action-Conditioned Contrastive Policy Pretrainingcited by 37
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learningcited by 37
- 2021Return-Based Contrastive Representation Learning for Reinforcement Learningcited by 36
- 2023AMAGO: Scalable In-Context Reinforcement Learning for Adaptive Agentscited by 36
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectorycited by 35
50 most cited of 268 All 268 (JSON)
- 2026Integrating Novelty and Surprise for Experience Prioritization and Exploration in Image-Based Reinforcement Learning18 Aug 2026
- 2026V-Simba: Unleashing the Architectural Potential of RL in Visual Continuous Control8 Aug 2026
- 2026Enhancing VLM Reward Models Through Structure-Aware Fine-Tuning4 Aug 2026
- 2026PAMD: Structured Adaptive Distances for Bisimulation Representations in Visual Reinforcement Learning20 Jul 2026
- 2026Mask-based Predictive Representations for Reinforcement Learning5 Jul 2026
- 2026Domain Adaptation with Adaptive Imagination for Visual Reinforcement Learning under Limited Target Data29 Jun 2026
- 2026A New Quaternion-Joint Cable-Driven Redundant Manipulator Configuration and its Control Through FABRIK and Residual Reinforcement Learning3 Jun 2026
- 2026R2R2: Robust Representation for Intensive Experience Reuse via Redundancy Reduction in Self-Predictive Learning13 May 2026
- 2026Higher Resolution, Better Generalization: Unlocking Visual Scaling in Deep Reinforcement Learning11 May 2026
- 2026Dream-MPC: Gradient-Based Model Predictive Control with Latent Imagination6 May 2026
- 2026Autonomous Diffractometry Enabled by Visual Reinforcement Learning13 Apr 2026
- 2026Morphology-Conditioned World Model for Cross-Embodiment Quadrupedal Locomotion9 Apr 2026
38 more
- 2026Squint: Fast Visual Reinforcement Learning for Sim-to-Real Robotics24 Feb 2026
- 2026Improving Policy Exploitation in Online Reinforcement Learning with Instant Retrospect Action27 Jan 2026
- 2025Spectral Representation-based Reinforcement Learning17 Dec 2025
- 2025Cloning Deterministic Worlds: The Critical Role of Latent Geometry in Long-Horizon World Models30 Oct 2025
- 2025Enhancing Tactile-based Reinforcement Learning for Robotic Control24 Oct 2025
- 2025Oracle-Guided Masked Contrastive Reinforcement Learning for Visuomotor Policies7 Oct 2025
- 2025Control of dynamical systems with neural networks6 Oct 2025
- 2025Hybrid Training for Vision-Language-Action Models1 Oct 2025
- 2025Sample Efficient Experience Replay in Non-stationary Environments18 Sep 2025
- 2025SegDAC: Visual Generalization in Reinforcement Learning via Dynamic Object Tokens12 Aug 2025
- 2025Scaling DRL for Decision Making: A Survey on Data, Network, and Training Budget Strategies5 Aug 2025
- 2025One-Step Flow Policy Mirror Descent31 Jul 2025
- 2025Revisiting Bisimulation Metric for Robust Representations in Reinforcement Learning24 Jul 2025
- 2025Chargax: A JAX Accelerated EV Charging Simulator2 Jul 2025
- 2025rQdia: Regularizing Q-Value Distributions With Image Augmentation26 Jun 2025
- 2025Learning Task Belief Similarity with Latent Dynamics for Meta-Reinforcement Learning24 Jun 2025
- 2025Harmonizing and Merging Source Models for CLIP-based Domain Generalization11 Jun 2025
- 2025Ego-centric Learning of Communicative World Models for Autonomous Driving9 Jun 2025
- 2025Self-Predictive Dynamics for Generalization of Vision-based Reinforcement Learning5 Jun 2025
- 2025Optimistic critics can empower small actors1 Jun 2025
- 2025Normalizing Flows are Capable Models for Continuous Control29 May 2025
- 2025Grounding Bodily Awareness in Visual Representations for Efficient Policy Learning24 May 2025
- 2025Mind the GAP! The Challenges of Scale in Pixel-based Deep Reinforcement Learning23 May 2025
- 2025Maximum Total Correlation Reinforcement Learning22 May 2025
- 2025Enhancing Robot Navigation Policies with Task-Specific Uncertainty Managements20 May 2025
- 2025Zero-Shot Visual Generalization in Robot Manipulation16 May 2025
- 2025Accelerating Visual-Policy Learning through Parallel Differentiable Simulation15 May 2025
- 2025Video-Enhanced Offline Reinforcement Learning: A Model-Based Approach10 May 2025
- 2025Pretraining a Shared Q-Network for Data-Efficient Offline Reinforcement Learning9 May 2025
- 2025Trajectory Entropy Reinforcement Learning for Robust Robot Motor Skill Learning7 May 2025
- 2025A Champion-level Vision-based Reinforcement Learning Agent for Competitive Racing in Gran Turismo 712 Apr 2025
- 2025Counterfactual experience augmented off-policy reinforcement learning18 Mar 2025
- 2025Enhance Exploration in Safe Reinforcement Learning with Contrastive Representation Learning13 Mar 2025
- 2025Studying the Interplay Between the Actor and Critic Representations in Reinforcement Learning8 Mar 2025
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models27 Feb 2025
- 2025Salience-Invariant Consistent Policy Learning for Generalization in Visual Reinforcement Learning12 Feb 2025
- 2025Efficient Reinforcement Learning Through Adaptively Pretrained Visual Encoder8 Feb 2025
- 2025RAPID: Robust and Agile Planner Using Inverse Reinforcement Learning for Vision-Based Drone Navigation4 Feb 2025
50 newest of 268 All 268 (JSON)
- 2021S4RL: Surprisingly Simple Self-Supervision for Offline Reinforcement Learningcited by 85
- 2022Learning Generalizable Dexterous Manipulation from Human Grasp Affordancecited by 46
- 2023For Pre-Trained Vision Models in Motor Control, Not All Policy Learning Methods are Created Equalcited by 22
- 2023Distance Weighted Supervised Learning for Offline Interaction Datacited by 18
- 2021Improving Zero-shot Generalization in Offline Reinforcement Learning using Generalized Similarity Functionscited by 17
- 2022Learning Bellman Complete Representations for Offline Policy Evaluationcited by 17
- 2023Constrained Policy Optimization with Explicit Behavior Density for Offline Reinforcement Learningcited by 16
- 2023Visual Imitation Learning with Patch Rewardscited by 16
- 2022Behavior Prior Representation learning for Offline Reinforcement Learningcited by 12
- 2023A Survey of Demonstration Learningcited by 12
- 2023Improving Behavioural Cloning with Positive Unlabeled Learningcited by 8
- 2024DMC-VB: A Benchmark for Representation Learning for Control with Visual Distractorscited by 5
12 most cited of 22 All 22 (JSON)
- 2025Zero-Shot Visual Generalization in Robot Manipulation16 May 2025
- 2025Video-Enhanced Offline Reinforcement Learning: A Model-Based Approach10 May 2025
- 2025Pretraining a Shared Q-Network for Data-Efficient Offline Reinforcement Learning9 May 2025
- 2025RAPID: Robust and Agile Planner Using Inverse Reinforcement Learning for Vision-Based Drone Navigation4 Feb 2025
- 2024Entropy Regularized Task Representation Learning for Offline Meta-Reinforcement Learning19 Dec 2024
- 2024DMC-VB: A Benchmark for Representation Learning for Control with Visual Distractors26 Sep 2024
- 2024Augmented Reality Demonstrations for Scalable Robot Imitation Learning20 Mar 2024
- 2024Visual Imitation Learning with Calibrated Contrastive Representation21 Jan 2024
- 2023Distance Weighted Supervised Learning for Offline Interaction Data26 Apr 2023
- 2023For Pre-Trained Vision Models in Motor Control, Not All Policy Learning Methods are Created Equal10 Apr 2023
- 2023A Survey of Demonstration Learning20 Mar 2023
- 2023Visual Imitation Learning with Patch Rewards2 Feb 2023
12 newest of 22 All 22 (JSON)
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2021Mastering Atari Games with Limited Datacited by 235
- 2022Masked World Models for Visual Controlcited by 169
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Modelscited by 53
- 2023HarmonyDream: Task Harmonization Inside World Modelscited by 22
- 2024Learning Latent Dynamic Robust Representations for World Modelscited by 20
- 2023ReCoRe: Regularized Contrastive Representation Learning of World Modelcited by 11
- 2024Simplifying Latent Dynamics with Softly State-Invariant World Modelscited by 10
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Modelscited by 10
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learningcited by 5
- 2023Model-Based Reinforcement Learning with Isolated Imaginationscited by 4
- 2023Go Beyond Imagination: Maximizing Episodic Reachability with World Modelscited by 3
12 most cited of 17 All 17 (JSON)
- 2026Morphology-Conditioned World Model for Cross-Embodiment Quadrupedal Locomotion9 Apr 2026
- 2025Cloning Deterministic Worlds: The Critical Role of Latent Geometry in Long-Horizon World Models30 Oct 2025
- 2025Ego-centric Learning of Communicative World Models for Autonomous Driving9 Jun 2025
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models27 Feb 2025
- 2024Learning Latent Dynamic Robust Representations for World Models10 May 2024
- 2024Simplifying Latent Dynamics with Softly State-Invariant World Models31 Jan 2024
- 2023ReCoRe: Regularized Contrastive Representation Learning of World Model14 Dec 2023
- 2023HarmonyDream: Task Harmonization Inside World Models30 Sep 2023
- 2023Go Beyond Imagination: Maximizing Episodic Reachability with World Models25 Aug 2023
- 2023Model-Based Reinforcement Learning with Multi-Task Offline Pretraining6 Jun 2023
- 2023Model-Based Reinforcement Learning with Isolated Imaginations27 Mar 2023
- 2023Continual Visual Reinforcement Learning with A Life-Long World Model12 Mar 2023
12 newest of 17 All 17 (JSON)
- 2023FastRLAP: A System for Learning High-Speed Driving via Deep RL and Autonomous Practicingcited by 13
- 2021Increasing the Efficiency of Policy Learning for Autonomous Vehicles by Multi-Task Representation Learningcited by 5
- 2021Vision Transformer for Learning Driving Policies in Complex Multi-Agent Environmentscited by 5
- 2023RLAD: Reinforcement Learning from Pixels for Autonomous Driving in Urban Environmentscited by 4
- 2024An Examination of Offline-Trained Encoders in Vision-Based Deep Reinforcement Learning for Autonomous Drivingcited by 0
- 2024An Examination of Offline-Trained Encoders in Vision-Based Deep Reinforcement Learning for Autonomous Driving2 Sep 2024
- 2023RLAD: Reinforcement Learning from Pixels for Autonomous Driving in Urban Environments29 May 2023
- 2023FastRLAP: A System for Learning High-Speed Driving via Deep RL and Autonomous Practicing19 Apr 2023
- 2021Vision Transformer for Learning Driving Policies in Complex Multi-Agent Environments14 Sep 2021
- 2021Increasing the Efficiency of Policy Learning for Autonomous Vehicles by Multi-Task Representation Learning26 Mar 2021
- 2025Hybrid Training for Vision-Language-Action Modelscited by 2
- 2025Hybrid Training for Vision-Language-Action Models1 Oct 2025
- 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 781
- 2021Domain Generalization: A Surveycited by 760
- 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 359
- 2022Masked Visual Pre-training for Motor Controlcited by 246
- 2022The Primacy Bias in Deep Reinforcement Learningcited by 242
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2021Reinforcement Learning with Prototypical Representationscited by 179
- 2021URLB: Unsupervised Reinforcement Learning Benchmarkcited by 138
- 2021Coarse-to-Fine Q-attention: Efficient Learning for Visual Robotic Manipulation via Discretisationcited by 120
- 2021State Entropy Maximization with Random Encoders for Efficient Explorationcited by 107
- 2022Reinforcement Learning with Action-Free Pre-Training from Videoscited by 107
- 2021Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentationcited by 101
12 most cited of 216 All 216 (JSON)
- 2026Integrating Novelty and Surprise for Experience Prioritization and Exploration in Image-Based Reinforcement Learning18 Aug 2026
- 2026V-Simba: Unleashing the Architectural Potential of RL in Visual Continuous Control8 Aug 2026
- 2026PAMD: Structured Adaptive Distances for Bisimulation Representations in Visual Reinforcement Learning20 Jul 2026
- 2026Mask-based Predictive Representations for Reinforcement Learning5 Jul 2026
- 2026Domain Adaptation with Adaptive Imagination for Visual Reinforcement Learning under Limited Target Data29 Jun 2026
- 2026A New Quaternion-Joint Cable-Driven Redundant Manipulator Configuration and its Control Through FABRIK and Residual Reinforcement Learning3 Jun 2026
- 2026R2R2: Robust Representation for Intensive Experience Reuse via Redundancy Reduction in Self-Predictive Learning13 May 2026
- 2026Higher Resolution, Better Generalization: Unlocking Visual Scaling in Deep Reinforcement Learning11 May 2026
- 2026Dream-MPC: Gradient-Based Model Predictive Control with Latent Imagination6 May 2026
- 2026Autonomous Diffractometry Enabled by Visual Reinforcement Learning13 Apr 2026
- 2026Squint: Fast Visual Reinforcement Learning for Sim-to-Real Robotics24 Feb 2026
- 2026Improving Policy Exploitation in Online Reinforcement Learning with Instant Retrospect Action27 Jan 2026
12 newest of 216 All 216 (JSON)