Dynamics-aware Embeddings
Not on the map. Why?
Abstract (arXiv)
In this paper we consider self-supervised representation learning to improve sample efficiency in reinforcement learning (RL). We propose a forward prediction objective for simultaneously learning embeddings of states and action sequences. These embeddings capture the structure of the environment's dynamics, enabling efficient policy learning. We demonstrate that our action embeddings alone improve the sample efficiency and peak performance of model-free RL on control from low-dimensional states. By combining state and action embeddings, we achieve efficient learning of high-quality policies on goal-conditioned continuous control from pixel observations in only 1-2 million environment steps.
Key facts. Cited by 34 papers in the map (data through 14 Oct 2026). First posted on arXiv 25 Aug 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 11; 2022: 5; 2023: 11; 2024: 4; 2025: 1; 2026: 2.
34 papers.
Sort:
By area:
- 2021Demonstration-Guided Reinforcement Learning with Learned Skillscited by 76
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 75
- 2021Accelerating Robotic Reinforcement Learning via Parameterized Action Primitivescited by 65
- 2023PlayFusion: Skill Acquisition via Diffusion from Language-Annotated Playcited by 50
- 2024HRP: Human Affordances for Robotic Pre-Trainingcited by 44
- 2024DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Controlcited by 44
- 2023An Unbiased Look at Datasets for Visuo-Motor Pre-Trainingcited by 42
- 2021Hierarchical Skills for Efficient Explorationcited by 38
- 2021HyAR: Addressing Discrete-Continuous Action Reinforcement Learning via Hybrid Action Representationcited by 35
- 2021LASER: Learning a Latent Action Space for Efficient Reinforcement Learningcited by 32
- 2023Inverse Dynamics Pretraining Learns Good Representations for Multitask Imitationcited by 31
- 2021JueWu-MC: Playing Minecraft with Sample-efficient Hierarchical Reinforcement Learningcited by 28
22 more
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 27
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatchcited by 25
- 2022Unsupervised Representation Learning in Deep Reinforcement Learning: A Reviewcited by 18
- 2021Hierarchical Neural Dynamic Policiescited by 17
- 2021Decoupled Exploration and Exploitation Policies for Sample-Efficient Reinforcement Learningcited by 14
- 2022One After Another: Learning Incremental Skills for a Changing Worldcited by 10
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractionscited by 9
- 2023Learning minimal representations of stochastic processes with variational autoencoderscited by 7
- 2022SkillS: Adaptive Skill Sequencing for Efficient Temporally-Extended Explorationcited by 6
- 2023Leveraging Jumpy Models for Planning and Fast Learning in Robotic Domainscited by 6
- 2021Physical Reasoning Using Dynamics-Aware Modelscited by 4
- 2023Dynamic Neighborhood Construction for Structured Large Discrete Action Spacescited by 4
- 2022Leveraging Demonstrations with Latent Space Priorscited by 3
- 2025Reinforcement-Learning Portfolio Allocation with Dynamic Embedding of Market Informationcited by 2
- 2021A Self-Supervised Auxiliary Loss for Deep RL in Partially Observable Settingscited by 1
- 2021Learning Routines for Effective Off-Policy Reinforcement Learningcited by 0
- 2023Learning Embeddings for Sequential Tasks Using Population of Agentscited by 0
- 2023DCT: Dual Channel Training of Action Embeddings for Reinforcement Learning with Large Discrete Action Spacescited by 0
- 2023A Central Motor System Inspired Pre-training Reinforcement Learning for Robotic Controlcited by 0
- 2024ATraDiff: Accelerating Online Reinforcement Learning with Imaginary Trajectoriescited by 0
- 2026Breaking the Grid: Distance-Guided Reinforcement Learning in Large Discrete Action Spacescited by 0
- 2026DynaWM: Dynamics-Aware Distillation with World Model and Momentum Targets for Smooth Locomotion over Continuous Stairscited by 0
- 2026DynaWM: Dynamics-Aware Distillation with World Model and Momentum Targets for Smooth Locomotion over Continuous Stairs23 Jun 2026
- 2026Breaking the Grid: Distance-Guided Reinforcement Learning in Large Discrete Action Spaces9 Feb 2026
- 2025Reinforcement-Learning Portfolio Allocation with Dynamic Embedding of Market Information29 Jan 2025
- 2024DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Control18 Sep 2024
- 2024HRP: Human Affordances for Robotic Pre-Training26 Jul 2024
- 2024ATraDiff: Accelerating Online Reinforcement Learning with Imaginary Trajectories6 Jun 2024
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatch24 May 2024
- 2023PlayFusion: Skill Acquisition via Diffusion from Language-Annotated Play7 Dec 2023
- 2023A Central Motor System Inspired Pre-training Reinforcement Learning for Robotic Control14 Nov 2023
- 2023An Unbiased Look at Datasets for Visuo-Motor Pre-Training13 Oct 2023
- 2023Learning minimal representations of stochastic processes with variational autoencoders21 Jul 2023
- 2023DCT: Dual Channel Training of Action Embeddings for Reinforcement Learning with Large Discrete Action Spaces28 Jun 2023
22 more
- 2023Learning Embeddings for Sequential Tasks Using Population of Agents5 Jun 2023
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learning4 Jun 2023
- 2023Dynamic Neighborhood Construction for Structured Large Discrete Action Spaces31 May 2023
- 2023Inverse Dynamics Pretraining Learns Good Representations for Multitask Imitation26 May 2023
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractions9 May 2023
- 2023Leveraging Jumpy Models for Planning and Fast Learning in Robotic Domains24 Feb 2023
- 2022SkillS: Adaptive Skill Sequencing for Efficient Temporally-Extended Exploration24 Nov 2022
- 2022Leveraging Demonstrations with Latent Space Priors26 Oct 2022
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradient15 Sep 2022
- 2022Unsupervised Representation Learning in Deep Reinforcement Learning: A Review27 Aug 2022
- 2022One After Another: Learning Incremental Skills for a Changing World21 Mar 2022
- 2021JueWu-MC: Playing Minecraft with Sample-efficient Hierarchical Reinforcement Learning7 Dec 2021
- 2021Accelerating Robotic Reinforcement Learning via Parameterized Action Primitives28 Oct 2021
- 2021Hierarchical Skills for Efficient Exploration20 Oct 2021
- 2021HyAR: Addressing Discrete-Continuous Action Reinforcement Learning via Hybrid Action Representation12 Sep 2021
- 2021Demonstration-Guided Reinforcement Learning with Learned Skills21 Jul 2021
- 2021Hierarchical Neural Dynamic Policies12 Jul 2021
- 2021Learning Routines for Effective Off-Policy Reinforcement Learning5 Jun 2021
- 2021A Self-Supervised Auxiliary Loss for Deep RL in Partially Observable Settings17 Apr 2021
- 2021LASER: Learning a Latent Action Space for Efficient Reinforcement Learning29 Mar 2021
- 2021Physical Reasoning Using Dynamics-Aware Models20 Feb 2021
- 2021Decoupled Exploration and Exploitation Policies for Sample-Efficient Reinforcement Learning23 Jan 2021
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatchcited by 25
- 2025Reinforcement-Learning Portfolio Allocation with Dynamic Embedding of Market Informationcited by 2
- 2024ATraDiff: Accelerating Online Reinforcement Learning with Imaginary Trajectoriescited by 0
- 2026Breaking the Grid: Distance-Guided Reinforcement Learning in Large Discrete Action Spacescited by 0
- 2026Breaking the Grid: Distance-Guided Reinforcement Learning in Large Discrete Action Spaces9 Feb 2026
- 2025Reinforcement-Learning Portfolio Allocation with Dynamic Embedding of Market Information29 Jan 2025
- 2024ATraDiff: Accelerating Online Reinforcement Learning with Imaginary Trajectories6 Jun 2024
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatch24 May 2024
- 2021Demonstration-Guided Reinforcement Learning with Learned Skillscited by 76
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 75
- 2021Accelerating Robotic Reinforcement Learning via Parameterized Action Primitivescited by 65
- 2024HRP: Human Affordances for Robotic Pre-Trainingcited by 44
- 2023An Unbiased Look at Datasets for Visuo-Motor Pre-Trainingcited by 42
- 2021Hierarchical Skills for Efficient Explorationcited by 38
- 2021HyAR: Addressing Discrete-Continuous Action Reinforcement Learning via Hybrid Action Representationcited by 35
- 2021LASER: Learning a Latent Action Space for Efficient Reinforcement Learningcited by 32
- 2023Inverse Dynamics Pretraining Learns Good Representations for Multitask Imitationcited by 31
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 27
- 2022Unsupervised Representation Learning in Deep Reinforcement Learning: A Reviewcited by 18
- 2021Hierarchical Neural Dynamic Policiescited by 17
12 most cited of 26 All 26 (JSON)
- 2024HRP: Human Affordances for Robotic Pre-Training26 Jul 2024
- 2023A Central Motor System Inspired Pre-training Reinforcement Learning for Robotic Control14 Nov 2023
- 2023An Unbiased Look at Datasets for Visuo-Motor Pre-Training13 Oct 2023
- 2023Learning minimal representations of stochastic processes with variational autoencoders21 Jul 2023
- 2023DCT: Dual Channel Training of Action Embeddings for Reinforcement Learning with Large Discrete Action Spaces28 Jun 2023
- 2023Learning Embeddings for Sequential Tasks Using Population of Agents5 Jun 2023
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learning4 Jun 2023
- 2023Dynamic Neighborhood Construction for Structured Large Discrete Action Spaces31 May 2023
- 2023Inverse Dynamics Pretraining Learns Good Representations for Multitask Imitation26 May 2023
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractions9 May 2023
- 2023Leveraging Jumpy Models for Planning and Fast Learning in Robotic Domains24 Feb 2023
- 2022SkillS: Adaptive Skill Sequencing for Efficient Temporally-Extended Exploration24 Nov 2022
12 newest of 26 All 26 (JSON)