iQRL -- Implicitly Quantized Representations for Sample-efficient Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Learning representations for reinforcement learning (RL) has shown much promise for continuous control. We propose an efficient representation learning method using only a self-supervised latent-state consistency loss. Our approach employs an encoder and a dynamics model to map observations to latent states and predict future latent states, respectively. We achieve high performance and prevent representation collapse by quantizing the latent representation such that the rank of the representation is empirically preserved. Our method, named iQRL: implicitly Quantized Reinforcement Learning, is straightforward, compatible with any model-free RL algorithm, and demonstrates excellent performance by outperforming other recently proposed representation learning methods in continuous control benchmarks from DeepMind Control Suite.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 38 of its 51 references that are in the map. First posted on arXiv 4 Jun 2024.
Builds on
38 of 51 references.
References · 38
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2018Addressing Function Approximation Error in Actor-Critic Methodscited by 2,302
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2018DeepMind Control Suitecited by 935
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 761
- 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 745
- 2020Image Augmentation Is All You Need: Regularizing Deep Reinforcement Learning from Pixelscited by 573
- 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 482
- 2023Finite Scalar Quantization: VQ-VAE Made Simplecited by 425
26 more
- 2020Learning Invariant Representations for Reinforcement Learning without Reconstructioncited by 424
- 2022Temporal Difference Learning for Model Predictive Controlcited by 375
- 2021Understanding Dimensional Collapse in Contrastive Self-supervised Learningcited by 374
- 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 359
- 2020Data-Efficient Reinforcement Learning with Self-Predictive Representationscited by 309
- 2018Towards a Definition of Disentangled Representationscited by 294
- 2019Improving Sample Efficiency in Model-Free Reinforcement Learning from Imagescited by 268
- 2015Embed to Control: A Locally Linear Latent Dynamics Model for Control from Raw Imagescited by 266
- 2021Mastering Atari Games with Limited Datacited by 235
- 2019DeepMDP: Learning Continuous Latent Space Models for Representation Learningcited by 204
- 2015Deep Spatial Autoencoders for Visuomotor Learningcited by 149
- 2017DARLA: Improving Zero-Shot Transfer in Reinforcement Learningcited by 147
- 2019Unsupervised State Representation Learning in Ataricited by 144
- 2015Model Predictive Path Integral Control using Covariance Variable Importance Samplingcited by 122
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 73
- 2020Approximate information state for approximate planning and reinforcement learning in partially observed systemscited by 67
- 2017Near Optimal Behavior via Approximate State Abstractioncited by 63
- 2023TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learningcited by 56
- 2024Bridging State and History Representations: Understanding Self-Predictive RLcited by 52
- 2022The Mechanism of Prediction Head in Non-contrastive Self-supervised Learningcited by 38
- 2018Natural Environment Benchmarks for Reinforcement Learningcited by 32
- 2020Representations for Stable Off-Policy Reinforcement Learningcited by 31
- 2021Learning Representations for Pixel-based Control: What Matters and Why?cited by 24
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
- 2021Learning Dynamics Models for Model Predictive Agentscited by 21
- 2023Simplified Temporal Consistency Reinforcement Learningcited by 17
Built on by
Citing papers per year: 2025: 6; 2026: 3.
9 papers.
- 2025Towards General-Purpose Model-Free Reinforcement Learningcited by 40
- 2025Latent Action Learning Requires Supervision in the Presence of Distractorscited by 40
- 2025Hyperspherical Normalization for Scalable Deep Reinforcement Learningcited by 35
- 2025Simplicial Embeddings Improve Sample Efficiency in Actor-Critic Agentscited by 12
- 2025Discrete Codebook World Models for Continuous Controlcited by 9
- 2025Rethinking the Role of Dynamic Sparse Training for Scalable Deep Reinforcement Learningcited by 1
- 2026Debiased Model-based Representations for Sample-efficient Continuous Controlcited by 1
- 2026Temporal Consistency Improves Generalization in Contextual Offline Meta Reinforcement Learningcited by 0
- 2026RAY-TOLD: Ray-Based Latent Dynamics for Dense Dynamic Obstacle Avoidance with TDMPCcited by 0
- 2026Debiased Model-based Representations for Sample-efficient Continuous Control12 May 2026
- 2026RAY-TOLD: Ray-Based Latent Dynamics for Dense Dynamic Obstacle Avoidance with TDMPC30 Apr 2026
- 2026Temporal Consistency Improves Generalization in Contextual Offline Meta Reinforcement Learning3 Mar 2026
- 2025Simplicial Embeddings Improve Sample Efficiency in Actor-Critic Agents15 Oct 2025
- 2025Rethinking the Role of Dynamic Sparse Training for Scalable Deep Reinforcement Learning14 Oct 2025
- 2025Discrete Codebook World Models for Continuous Control1 Mar 2025
- 2025Hyperspherical Normalization for Scalable Deep Reinforcement Learning21 Feb 2025
- 2025Latent Action Learning Requires Supervision in the Presence of Distractors1 Feb 2025
- 2025Towards General-Purpose Model-Free Reinforcement Learning27 Jan 2025