Deep Reinforcement Learning in Large Discrete Action Spaces
Not on the map. Why?
Abstract (arXiv)
Being able to reason in an environment with a large number of discrete actions is essential to bringing reinforcement learning to a larger class of problems. Recommender systems, industrial plants and language models are only some of the many real-world tasks involving large numbers of discrete actions for which current methods are difficult or even often impossible to apply. An ability to generalize over the set of actions as well as sub-linear complexity relative to the size of the set are both necessary to handle such tasks. Current approaches are not able to provide both of these, which motivates the work in this paper. Our proposed approach leverages prior information about the actions to embed them in a continuous space upon which it can generalize. Additionally, approximate nearest-neighbor methods allow for logarithmic-time lookup complexity relative to the number of actions, which is necessary for time-wise tractable training. This combined approach allows reinforcement learning methods to be applied to large-scale learning problems previously intractable with current methods. We demonstrate our algorithm's abilities on a series of tasks having up to one million actions.
Key facts. Cited by 158 papers in the map (data through 7 Oct 2026). First posted on arXiv 24 Dec 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 1; 2021: 29; 2022: 33; 2023: 26; 2024: 33; 2025: 20; 2026: 15.
158 papers.
Sort:
By area:
- 2021Learning and Planning in Complex Action Spacescited by 76
- 2021DouZero: Mastering DouDizhu with Self-Play Deep Reinforcement Learningcited by 65
- 2023A Survey of Temporal Credit Assignment in Deep Reinforcement Learningcited by 56
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddingscited by 53
- 2024Value Augmented Sampling for Language Model Alignment and Personalizationcited by 49
- 2021Reinforced Neighborhood Selection Guided Multi-Relational Graph Neural Networkscited by 38
- 2021Learning Large Neighborhood Search Policy for Integer Programmingcited by 35
- 2024Grounding Multimodal Large Language Models in Actionscited by 35
- 2023Off-Policy Evaluation for Large Action Spaces via Conjunct Effect Modelingcited by 34
- 2024Efficient Reinforcement Learning with Large Language Model Priorscited by 30
- 2021NeurWIN: Neural Whittle Index Network For Restless Bandits Via Deep RLcited by 27
- 2022DEP-RL: Embodied Exploration for Reinforcement Learning in Overactuated and Musculoskeletal Systemscited by 27
38 more
- 2023Multi-Task Recommendations with Reinforcement Learningcited by 26
- 2024SynFlowNet: Design of Diverse and Novel Molecules with Synthesis Constraintscited by 26
- 2021Reinforcement Learning for Quantitative Tradingcited by 24
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
- 2023In-Context Reinforcement Learning for Variable Action Spacescited by 23
- 2022Multi-Task Fusion via Reinforcement Learning for Long-Term User Satisfaction in Recommender Systemscited by 22
- 2023A Survey on Physics Informed Reinforcement Learning: Review and Open Problemscited by 22
- 2021Continuous Control with Action Quantization from Demonstrationscited by 21
- 2024Generative Flows on Synthetic Pathway for Drug Designcited by 20
- 2021Deep hierarchical reinforcement agents for automated penetration testingcited by 19
- 2022Unsupervised Representation Learning in Deep Reinforcement Learning: A Reviewcited by 18
- 2023Two-Stage Constrained Actor-Critic for Short Video Recommendationcited by 16
- 2023B-Coder: Value-Based Deep Reinforcement Learning for Program Synthesiscited by 16
- 2024Graph Reinforcement Learning for Combinatorial Optimization: A Survey and Unifying Perspectivecited by 16
- 2024Hierarchical Multi-agent Reinforcement Learning for Cyber Network Defensecited by 16
- 2020Benchmarking End-to-End Behavioural Cloning on Video Gamescited by 15
- 2023UnifiedGesture: A Unified Gesture Synthesis Model for Multiple Skeletonscited by 15
- 2023Games for Artificial Intelligence Research: A Review and Perspectivescited by 14
- 2021Towards Return Parity in Markov Decision Processescited by 12
- 2022Pervasive Machine Learning for Smart Radio Environments Enabled by Reconfigurable Intelligent Surfacescited by 12
- 2023Generative Slate Recommendation with Reinforcement Learningcited by 12
- 2022Perceiving the World: Question-guided Reinforcement Learning for Text-based Gamescited by 10
- 2022Asking for Knowledge: Training RL Agents to Query External Knowledge Using Languagecited by 10
- 2022ResAct: Reinforcing Long-term Engagement in Sequential Recommendation with Residual Actorcited by 10
- 2023Generative Modelling of Stochastic Actions with Arbitrary Constraints in Reinforcement Learningcited by 10
- 2021Discrete-continuous Action Space Policy Gradient-based Attention for Image-Text Matchingcited by 8
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractionscited by 8
- 2019Action Assembly: Sparse Imitation Learning for Text Based Games with Combinatorial Action Spacescited by 7
- 2021Reversible Action Design for Combinatorial Optimization with Reinforcement Learningcited by 7
- 2021Don't Do What Doesn't Matter: Intrinsic Motivation with Action Usefulnesscited by 7
- 2025Structured Reinforcement Learning for Combinatorial Decision-Makingcited by 7
- 2021Deep Reinforcement Learning Optimizes Graphene Nanopores for Efficient Desalinationcited by 6
- 2021End-to-End Intersection Handling using Multi-Agent Deep Reinforcement Learningcited by 6
- 2021Learning to Operate an Electric Vehicle Charging Station Considering Vehicle-grid Integrationcited by 6
- 2022Control of Dual-Sourcing Inventory Systems using Recurrent Neural Networkscited by 6
- 2022SAFER: Data-Efficient and Safe Reinforcement Learning via Skill Acquisitioncited by 6
- 2022Should I send this notification? Optimizing push notifications decision making by modeling the futurecited by 6
- 2022Reward Shaping for User Satisfaction in a REINFORCE Recommendercited by 6
50 most cited of 158 All 158 (JSON)
- 2026Human-inspired, Task-Dimension-Guided Exploration for Efficient Learning in High Dimensions29 Sep 2026
- 2026Revisiting Overestimation Bias Problem of Q-learning: Settling Large Discrete Action Space via Action Intersection13 Aug 2026
- 2026Bellman-Taylor Score Decoding for Markov Decision Processes with State-Dependent Feasible Action Sets9 Jun 2026
- 2026SIRI: Self-Internalizing Reinforcement Learning with Intrinsic Skills for LLM Agent Training1 Jun 2026
- 2026Projecting Latent RL Actions: Towards Generalizable and Scalable Graph Combinatorial Optimization19 May 2026
- 2026Controllable Molecular Generative Foundation Models14 May 2026
- 2026Dynamic resource matching in manufacturing using deep reinforcement learning28 Mar 2026
- 2026Learning to Select Visual In-Context Demonstrations24 Mar 2026
- 2026Robust Post-Training for Generative Recommenders: Why Exponential Reward-Weighted SFT Outperforms RLHF10 Mar 2026
- 2026Overcoming Valid Action Suppression in Unmasked Policy Gradient Algorithms10 Mar 2026
- 2026A Scalable Approach to Solving Simulation-Based Network Security Games18 Feb 2026
- 2026Breaking the Grid: Distance-Guided Reinforcement Learning in Large Discrete Action Spaces9 Feb 2026
38 more
- 2026Latent Spherical Flow Policy for Reinforcement Learning with Combinatorial Actions29 Jan 2026
- 2026Combinatorial Optimization Augmented Machine Learning15 Jan 2026
- 2026Improving and Accelerating Offline RL in Large Discrete Action Spaces with Structured Policy Initialization7 Jan 2026
- 2025Probabilistic Hash Embeddings for Online Learning of Categorical Features25 Nov 2025
- 2025Vulnerable Agent Identification in Large-Scale Multi-Agent Reinforcement Learning18 Sep 2025
- 2025SEG-Parking: Towards Safe, Efficient, and Generalizable Autonomous Parking via End-to-End Offline Reinforcement Learning17 Sep 2025
- 2025Action Space Reduction Strategies for Reinforcement Learning in Autonomous Driving7 Jul 2025
- 2025Exploring Large Action Sets with Hyperspherical Embeddings using von Mises-Fisher Sampling1 Jul 2025
- 2025Learning Truthful Mechanisms without Discretization28 Jun 2025
- 2025Agentic Personalisation of Cross-Channel Marketing Experiences19 Jun 2025
- 2025Structured Reinforcement Learning for Combinatorial Decision-Making25 May 2025
- 2025Language Model Distillation: A Temporal Difference Imitation Learning Perspective24 May 2025
- 2025Embedded Mean Field Reinforcement Learning for Perimeter-defense Game20 May 2025
- 2025Multi-parameter Control for the (1+(λ,λ))-GA on OneMax via Deep Reinforcement Learning19 May 2025
- 2025SAINT: Attention-Based Policies for Discrete Combinatorial Action Spaces17 May 2025
- 2025Cache-Efficient Posterior Sampling for Reinforcement Learning with LLM-Derived Priors Across Discrete and Continuous Domains12 May 2025
- 2025Adaptive and Robust DBSCAN with Multi-agent Reinforcement Learning7 May 2025
- 2025Q-function Decomposition with Intervention Semantics with Factored Action Spaces30 Apr 2025
- 2025Self-Balancing, Memory Efficient, Dynamic Metric Space Data Maintenance, for Rapid Multi-Kernel Estimation25 Apr 2025
- 2025An Efficient Approach for Cooperative Multi-Agent Learning Problems7 Apr 2025
- 2025Reinforcement learning with combinatorial actions for coupled restless bandits1 Mar 2025
- 2025Communication Strategy on Macro-and-Micro Traffic State in Cooperative Deep Reinforcement Learning for Regional Traffic Signal Control18 Feb 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
- 2024BraVE: Offline Reinforcement Learning for Discrete Combinatorial Action Spaces28 Oct 2024
- 2024Hierarchical Multi-agent Reinforcement Learning for Cyber Network Defense22 Oct 2024
- 2024Mitigating Suboptimality of Deterministic Policy Gradients in Complex Q-functions15 Oct 2024
- 2024Deep Learning-driven Mobile Traffic Measurement Collection and Analysis14 Oct 2024
- 2024Efficient Reinforcement Learning with Large Language Model Priors10 Oct 2024
- 2024Learning in complex action spaces without policy gradients8 Oct 2024
- 2024Generative Flows on Synthetic Pathway for Drug Design6 Oct 2024
- 2024Distribution Guided Active Feature Acquisition4 Oct 2024
- 2024A Fresh Take on Stale Embeddings: Improving Dense Retriever Training with Corrector Networks3 Sep 2024
- 2024Enhancing Quantum Memory Lifetime with Measurement-Free Local Error Correction and Reinforcement Learning18 Aug 2024
- 2024Top K Enhanced Reinforcement Learning Attacks on Heterogeneous Graph Node Classification4 Aug 2024
- 2024Deep progressive reinforcement learning-based flexible resource scheduling framework for IRS and UAV-assisted MEC system2 Aug 2024
- 2024A Contextual Combinatorial Bandit Approach to Negotiation30 Jun 2024
- 2024Optimizing Novelty of Top-k Recommendations using Large Language Models and Reinforcement Learning20 Jun 2024
50 newest of 158 All 158 (JSON)
- 2022Multi-Task Fusion via Reinforcement Learning for Long-Term User Satisfaction in Recommender Systemscited by 22
- 2020Benchmarking End-to-End Behavioural Cloning on Video Gamescited by 15
- 2019Action Assembly: Sparse Imitation Learning for Text Based Games with Combinatorial Action Spacescited by 7
- 2021Value Penalized Q-Learning for Recommender Systemscited by 4
- 2022Fast Offline Policy Optimization for Large Scale Recommendationcited by 4
- 2024BraVE: Offline Reinforcement Learning for Discrete Combinatorial Action Spacescited by 3
- 2024An Investigation of Offline Reinforcement Learning in Factorisable Action Spacescited by 3
- 2024Amortized nonmyopic active search via deep imitation learningcited by 2
- 2025Language Model Distillation: A Temporal Difference Imitation Learning Perspectivecited by 2
- 2023Model-enhanced Contrastive Reinforcement Learning for Sequential Recommendationcited by 1
- 2025SEG-Parking: Towards Safe, Efficient, and Generalizable Autonomous Parking via End-to-End Offline Reinforcement Learningcited by 1
- 2026Improving and Accelerating Offline RL in Large Discrete Action Spaces with Structured Policy Initializationcited by 0
- 2026Improving and Accelerating Offline RL in Large Discrete Action Spaces with Structured Policy Initialization7 Jan 2026
- 2025SEG-Parking: Towards Safe, Efficient, and Generalizable Autonomous Parking via End-to-End Offline Reinforcement Learning17 Sep 2025
- 2025Language Model Distillation: A Temporal Difference Imitation Learning Perspective24 May 2025
- 2024An Investigation of Offline Reinforcement Learning in Factorisable Action Spaces17 Nov 2024
- 2024BraVE: Offline Reinforcement Learning for Discrete Combinatorial Action Spaces28 Oct 2024
- 2024Amortized nonmyopic active search via deep imitation learning23 May 2024
- 2023Model-enhanced Contrastive Reinforcement Learning for Sequential Recommendation25 Oct 2023
- 2022Multi-Task Fusion via Reinforcement Learning for Long-Term User Satisfaction in Recommender Systems9 Aug 2022
- 2022Fast Offline Policy Optimization for Large Scale Recommendation8 Aug 2022
- 2021Value Penalized Q-Learning for Recommender Systems15 Oct 2021
- 2020Benchmarking End-to-End Behavioural Cloning on Video Games2 Apr 2020
- 2019Action Assembly: Sparse Imitation Learning for Text Based Games with Combinatorial Action Spaces23 May 2019
- 2026SIRI: Self-Internalizing Reinforcement Learning with Intrinsic Skills for LLM Agent Trainingcited by 5
- 2024Knowing What Not to Do: Leverage Language Model Insights for Action Space Pruning in Multi-agent Reinforcement Learningcited by 3
- 2025Agentic Personalisation of Cross-Channel Marketing Experiencescited by 3
- 2026SIRI: Self-Internalizing Reinforcement Learning with Intrinsic Skills for LLM Agent Training1 Jun 2026
- 2025Agentic Personalisation of Cross-Channel Marketing Experiences19 Jun 2025
- 2024Knowing What Not to Do: Leverage Language Model Insights for Action Space Pruning in Multi-agent Reinforcement Learning27 May 2024
- 2024Grounding Multimodal Large Language Models in Actionscited by 35
- 2026Learning to Select Visual In-Context Demonstrationscited by 1
- 2026Learning to Select Visual In-Context Demonstrations24 Mar 2026
- 2024Grounding Multimodal Large Language Models in Actions12 Jun 2024
- 2021Learning and Planning in Complex Action Spacescited by 76
- 2021DouZero: Mastering DouDizhu with Self-Play Deep Reinforcement Learningcited by 65
- 2023A Survey of Temporal Credit Assignment in Deep Reinforcement Learningcited by 56
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddingscited by 53
- 2021Reinforced Neighborhood Selection Guided Multi-Relational Graph Neural Networkscited by 38
- 2021Learning Large Neighborhood Search Policy for Integer Programmingcited by 35
- 2023Off-Policy Evaluation for Large Action Spaces via Conjunct Effect Modelingcited by 34
- 2024Efficient Reinforcement Learning with Large Language Model Priorscited by 30
- 2021NeurWIN: Neural Whittle Index Network For Restless Bandits Via Deep RLcited by 27
- 2022DEP-RL: Embodied Exploration for Reinforcement Learning in Overactuated and Musculoskeletal Systemscited by 27
- 2023Multi-Task Recommendations with Reinforcement Learningcited by 26
- 2024SynFlowNet: Design of Diverse and Novel Molecules with Synthesis Constraintscited by 26
12 most cited of 134 All 134 (JSON)
- 2026Human-inspired, Task-Dimension-Guided Exploration for Efficient Learning in High Dimensions29 Sep 2026
- 2026Revisiting Overestimation Bias Problem of Q-learning: Settling Large Discrete Action Space via Action Intersection13 Aug 2026
- 2026Bellman-Taylor Score Decoding for Markov Decision Processes with State-Dependent Feasible Action Sets9 Jun 2026
- 2026Projecting Latent RL Actions: Towards Generalizable and Scalable Graph Combinatorial Optimization19 May 2026
- 2026Controllable Molecular Generative Foundation Models14 May 2026
- 2026Dynamic resource matching in manufacturing using deep reinforcement learning28 Mar 2026
- 2026Overcoming Valid Action Suppression in Unmasked Policy Gradient Algorithms10 Mar 2026
- 2026A Scalable Approach to Solving Simulation-Based Network Security Games18 Feb 2026
- 2026Breaking the Grid: Distance-Guided Reinforcement Learning in Large Discrete Action Spaces9 Feb 2026
- 2026Latent Spherical Flow Policy for Reinforcement Learning with Combinatorial Actions29 Jan 2026
- 2026Combinatorial Optimization Augmented Machine Learning15 Jan 2026
- 2025Probabilistic Hash Embeddings for Online Learning of Categorical Features25 Nov 2025
12 newest of 134 All 134 (JSON)