Learning Action Representations for Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Most model-free reinforcement learning methods leverage state representations (embeddings) for generalization, but either ignore structure in the space of actions or assume the structure is provided a priori. We show how a policy can be decomposed into a component that acts in a low-dimensional space of action representations and a component that transforms these representations into actual actions. These representations improve generalization over large, finite action sets by allowing the agent to infer the outcomes of actions similar to actions already taken. We provide an algorithm to both learn and use action representations and provide conditions for its convergence. The efficacy of the proposed method is demonstrated on large-scale real-world problems.
Key facts. Cited by 75 papers in the map (data through 7 Oct 2026). First posted on arXiv 1 Feb 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 16; 2022: 15; 2023: 14; 2024: 8; 2025: 14; 2026: 8.
75 papers.
Sort:
By area:
- 2022Tutorial on amortized optimizationcited by 83
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 73
- 2022Supervised Training of Conditional Monge Mapscited by 60
- 2021Exploration in Deep Reinforcement Learning: From Single-Agent to Multiagent Domaincited by 57
- 2023TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learningcited by 56
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddingscited by 53
- 2021Learning Large Neighborhood Search Policy for Integer Programmingcited by 35
- 2023Off-Policy Evaluation for Large Action Spaces via Conjunct Effect Modelingcited by 34
- 2021HyAR: Addressing Discrete-Continuous Action Reinforcement Learning via Hybrid Action Representationcited by 33
- 2021LASER: Learning a Latent Action Space for Efficient Reinforcement Learningcited by 32
- 2025What Do Latent Action Models Actually Learn?cited by 30
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
38 more
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatchcited by 24
- 2024PRISE: LLM-Style Sequence Compression for Learning Temporal Action Abstractions in Controlcited by 21
- 2021Deep hierarchical reinforcement agents for automated penetration testingcited by 19
- 2022Unsupervised Representation Learning in Deep Reinforcement Learning: A Reviewcited by 18
- 2024Off-Policy Evaluation of Slate Bandit Policies via Optimizing Abstractioncited by 17
- 2025A Survey of State Representation Learning for Deep Reinforcement Learningcited by 17
- 2024POTEC: Off-Policy Learning for Large Action Spaces via Two-Stage Policy Decompositioncited by 13
- 2023Generative Slate Recommendation with Reinforcement Learningcited by 12
- 2025Action Tokenizer Matters in In-Context Imitation Learningcited by 12
- 2022Learn the Time to Learn: Replay Scheduling in Continual Learningcited by 11
- 2021Uncertainty Estimation Using Riemannian Model Dynamics for Offline Reinforcement Learningcited by 10
- 2022Perceiving the World: Question-guided Reinforcement Learning for Text-based Gamescited by 10
- 2022ResAct: Reinforcing Long-term Engagement in Sequential Recommendation with Residual Actorcited by 10
- 2024AD3: Implicit Action is the Key for World Models to Distinguish the Diverse Visual Distractorscited by 10
- 2021Action Redundancy in Reinforcement Learningcited by 9
- 2022Reinforcement Learning in Presence of Discrete Markovian Context Evolutioncited by 9
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractionscited by 8
- 2021Solving Large-Scale Extensive-Form Network Security Games via Neural Fictitious Self-Playcited by 7
- 2023Learning Action Embeddings for Off-Policy Evaluationcited by 7
- 2025Structured Reinforcement Learning for Combinatorial Decision-Makingcited by 7
- 2021Toward Robust Long Range Policy Transfercited by 6
- 2021How to Sense the World: Leveraging Hierarchy in Multimodal Perception for Robust Reinforcement Learning Agentscited by 6
- 2022SAFER: Data-Efficient and Safe Reinforcement Learning via Skill Acquisitioncited by 6
- 2022Simple Emergent Action Representations from Multi-Task Policy Trainingcited by 6
- 2023No Prior Mask: Eliminate Redundant Action for Deep Reinforcement Learningcited by 6
- 2023Dynamic Neighborhood Construction for Structured Large Discrete Action Spacescited by 4
- 2023Representation-Driven Reinforcement Learningcited by 4
- 2025Learning Actionable World Models for Industrial Process Controlcited by 4
- 2021Locality Matters: A Scalable Value Decomposition Approach for Cooperative Multi-Agent Reinforcement Learningcited by 3
- 2023Action Pick-up in Dynamic Action Space Reinforcement Learningcited by 3
- 2024Action abstractions for amortized samplingcited by 3
- 2024An Investigation of Offline Reinforcement Learning in Factorisable Action Spacescited by 3
- 2021Low-Dimensional State and Action Representation Learning with MDP Homomorphism Metricscited by 2
- 2021Learning Natural Language Generation from Scratchcited by 2
- 2022Reinforcement Learning with Large Action Spaces for Neural Machine Translationcited by 2
- 2022Representation Learning for Continuous Action Spaces is Beneficial for Efficient Policy Learningcited by 2
- 2023Distributional Off-Policy Evaluation for Slate Recommendationscited by 2
- 2023Self-supervised Pretraining for Decision Foundation Model: Formulation, Pipeline and Challengescited by 2
50 most cited of 75 All 75 (JSON)
- 2026Human-inspired, Task-Dimension-Guided Exploration for Efficient Learning in High Dimensions29 Sep 2026
- 2026LLM-Guided Contextual Action Evaluation for Operational Decisions in Industrial Processes25 Aug 2026
- 2026Bellman-Taylor Score Decoding for Markov Decision Processes with State-Dependent Feasible Action Sets9 Jun 2026
- 2026CODA: Coordination via On-Policy Diffusion for Multi-Agent Offline Reinforcement Learning25 Apr 2026
- 2026A Hierarchical Spatiotemporal Action Tokenizer for In-Context Imitation Learning in Robotics16 Apr 2026
- 2026Vocabulary Dropout for Curriculum Diversity in LLM Co-Evolution3 Apr 2026
- 2026QSIM: Mitigating Overestimation in Multi-Agent Reinforcement Learning via Action Similarity Weighted Q-Learning26 Feb 2026
- 2026Breaking the Grid: Distance-Guided Reinforcement Learning in Large Discrete Action Spaces9 Feb 2026
- 2025Scaling DRL for Decision Making: A Survey on Data, Network, and Training Budget Strategies5 Aug 2025
- 2025BEAVER: Building Environments with Assessable Variation for Evaluating Multi-Objective Reinforcement Learning10 Jul 2025
- 2025A Survey of State Representation Learning for Deep Reinforcement Learning20 Jun 2025
- 2025Action-Adaptive Continual Learning: Enabling Policy Generalization under Dynamic Action Spaces6 Jun 2025
38 more
- 2025Training RL Agents for Multi-Objective Network Defense Tasks28 May 2025
- 2025What Do Latent Action Models Actually Learn?27 May 2025
- 2025Beyond the Proxy: Trajectory-Distilled Guidance for Offline GFlowNet Training26 May 2025
- 2025Structured Reinforcement Learning for Combinatorial Decision-Making25 May 2025
- 2025Embedded Mean Field Reinforcement Learning for Perimeter-defense Game20 May 2025
- 2025LOD: Latent Objective Discovery in Heterogeneous Multi-Agent Reinforcement Learning14 May 2025
- 2025Enhancing Player Enjoyment with a Two-Tier DRL and LLM-Based Agent System for Fighting Games10 Apr 2025
- 2025Learning Actionable World Models for Industrial Process Control3 Mar 2025
- 2025Action Tokenizer Matters in In-Context Imitation Learning3 Mar 2025
- 2025Reducing Action Space for Deep Reinforcement Learning via Causal Effect Estimation24 Jan 2025
- 2024An Investigation of Offline Reinforcement Learning in Factorisable Action Spaces17 Nov 2024
- 2024Action abstractions for amortized sampling19 Oct 2024
- 2024Addressing Large Action Spaces in 3D Floorplanning via Spatial Generalization15 Jun 2024
- 2024Cross-Domain Policy Adaptation by Capturing Representation Mismatch24 May 2024
- 2024AD3: Implicit Action is the Key for World Models to Distinguish the Diverse Visual Distractors15 Mar 2024
- 2024PRISE: LLM-Style Sequence Compression for Learning Temporal Action Abstractions in Control16 Feb 2024
- 2024POTEC: Off-Policy Learning for Large Action Spaces via Two-Stage Policy Decomposition9 Feb 2024
- 2024Off-Policy Evaluation of Slate Bandit Policies via Optimizing Abstraction3 Feb 2024
- 2023Self-supervised Pretraining for Decision Foundation Model: Formulation, Pipeline and Challenges29 Dec 2023
- 2023No Prior Mask: Eliminate Redundant Action for Deep Reinforcement Learning11 Dec 2023
- 2023Distributional Off-Policy Evaluation for Slate Recommendations27 Aug 2023
- 2023DCT: Dual Channel Training of Action Embeddings for Reinforcement Learning with Large Discrete Action Spaces28 Jun 2023
- 2023TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learning22 Jun 2023
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learning4 Jun 2023
- 2023Representation-Driven Reinforcement Learning31 May 2023
- 2023Dynamic Neighborhood Construction for Structured Large Discrete Action Spaces31 May 2023
- 2023Off-Policy Evaluation for Large Action Spaces via Conjunct Effect Modeling14 May 2023
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractions9 May 2023
- 2023Learning Action Embeddings for Off-Policy Evaluation6 May 2023
- 2023Action Pick-up in Dynamic Action Space Reinforcement Learning3 Apr 2023
- 2023Diverse Policy Optimization for Structured Action Space23 Feb 2023
- 2023Generative Slate Recommendation with Reinforcement Learning20 Jan 2023
- 2022Representation Learning for Continuous Action Spaces is Beneficial for Efficient Policy Learning23 Nov 2022
- 2022Simple Emergent Action Representations from Multi-Task Policy Training18 Oct 2022
- 2022Reinforcement Learning with Large Action Spaces for Neural Machine Translation6 Oct 2022
- 2022Learn the Time to Learn: Replay Scheduling in Continual Learning18 Sep 2022
- 2022Selective Token Generation for Few-shot Natural Language Generation17 Sep 2022
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradient15 Sep 2022
50 newest of 75 All 75 (JSON)
- 2024PRISE: LLM-Style Sequence Compression for Learning Temporal Action Abstractions in Controlcited by 21
- 2025Action Tokenizer Matters in In-Context Imitation Learningcited by 12
- 2021Uncertainty Estimation Using Riemannian Model Dynamics for Offline Reinforcement Learningcited by 10
- 2024An Investigation of Offline Reinforcement Learning in Factorisable Action Spacescited by 3
- 2026A Hierarchical Spatiotemporal Action Tokenizer for In-Context Imitation Learning in Roboticscited by 0
- 2026CODA: Coordination via On-Policy Diffusion for Multi-Agent Offline Reinforcement Learningcited by 0
- 2026CODA: Coordination via On-Policy Diffusion for Multi-Agent Offline Reinforcement Learning25 Apr 2026
- 2026A Hierarchical Spatiotemporal Action Tokenizer for In-Context Imitation Learning in Robotics16 Apr 2026
- 2025Action Tokenizer Matters in In-Context Imitation Learning3 Mar 2025
- 2024An Investigation of Offline Reinforcement Learning in Factorisable Action Spaces17 Nov 2024
- 2024PRISE: LLM-Style Sequence Compression for Learning Temporal Action Abstractions in Control16 Feb 2024
- 2021Uncertainty Estimation Using Riemannian Model Dynamics for Offline Reinforcement Learning22 Feb 2021
- 2023Diverse Policy Optimization for Structured Action Spacecited by 1
- 2023Diverse Policy Optimization for Structured Action Space23 Feb 2023
- 2022Tutorial on amortized optimizationcited by 83
- 2023For SALE: State-Action Representation Learning for Deep Reinforcement Learningcited by 73
- 2022Supervised Training of Conditional Monge Mapscited by 60
- 2021Exploration in Deep Reinforcement Learning: From Single-Agent to Multiagent Domaincited by 57
- 2023TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learningcited by 56
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddingscited by 53
- 2021Learning Large Neighborhood Search Policy for Integer Programmingcited by 35
- 2023Off-Policy Evaluation for Large Action Spaces via Conjunct Effect Modelingcited by 34
- 2021HyAR: Addressing Discrete-Continuous Action Reinforcement Learning via Hybrid Action Representationcited by 33
- 2021LASER: Learning a Latent Action Space for Efficient Reinforcement Learningcited by 32
- 2025What Do Latent Action Models Actually Learn?cited by 30
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
12 most cited of 65 All 65 (JSON)
- 2026Human-inspired, Task-Dimension-Guided Exploration for Efficient Learning in High Dimensions29 Sep 2026
- 2026LLM-Guided Contextual Action Evaluation for Operational Decisions in Industrial Processes25 Aug 2026
- 2026Bellman-Taylor Score Decoding for Markov Decision Processes with State-Dependent Feasible Action Sets9 Jun 2026
- 2026Vocabulary Dropout for Curriculum Diversity in LLM Co-Evolution3 Apr 2026
- 2026QSIM: Mitigating Overestimation in Multi-Agent Reinforcement Learning via Action Similarity Weighted Q-Learning26 Feb 2026
- 2026Breaking the Grid: Distance-Guided Reinforcement Learning in Large Discrete Action Spaces9 Feb 2026
- 2025Scaling DRL for Decision Making: A Survey on Data, Network, and Training Budget Strategies5 Aug 2025
- 2025BEAVER: Building Environments with Assessable Variation for Evaluating Multi-Objective Reinforcement Learning10 Jul 2025
- 2025A Survey of State Representation Learning for Deep Reinforcement Learning20 Jun 2025
- 2025Action-Adaptive Continual Learning: Enabling Policy Generalization under Dynamic Action Spaces6 Jun 2025
- 2025Training RL Agents for Multi-Objective Network Defense Tasks28 May 2025
- 2025What Do Latent Action Models Actually Learn?27 May 2025
12 newest of 65 All 65 (JSON)