mlsanityData to 7 Oct 2026

Efficient Planning in a Compact Latent Action Space

arXiv 2208.10291 ↗First posted 22 Aug 2022cs.LGcited by 57

Not on the map. Why?

Cited by 30 in Offline RL and imitation learning from data, but placed by its own abstract, which didn’t fit.

Abstract (arXiv)

Planning-based reinforcement learning has shown strong performance in tasks in discrete and low-dimensional continuous action spaces. However, planning usually brings significant computational overhead for decision-making, and scaling such methods to high-dimensional action spaces remains challenging. To advance efficient planning for high-dimensional continuous control, we propose Trajectory Autoencoding Planner (TAP), which learns low-dimensional latent action codes with a state-conditional VQ-VAE. The decoder of the VQ-VAE thus serves as a novel dynamics model that takes latent actions and current state as input and reconstructs long-horizon trajectories. During inference time, given a starting state, TAP searches over discrete latent actions to find trajectories that have both high probability under the training distribution and high predicted cumulative reward. Empirical evaluation in the offline RL setting demonstrates low decision latency which is indifferent to the growing raw action dimensionality. For Adroit robotic hand manipulation tasks with high-dimensional continuous action space, TAP surpasses existing model-based methods by a large margin and also beats strong model-free actor-critic baselines.

Key facts. Cited by 57 papers in the map (data through 7 Oct 2026). Builds on 41 of its 59 references that are in the map. First posted on arXiv 22 Aug 2022.

Builds on

41 of 59 references.

References · 41

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020Language Models are Few-Shot Learnerscited by 32,199
  4. 2017Neural Discrete Representation Learningcited by 4,650
  5. 2020Conservative Q-Learning for Offline Reinforcement Learningcited by 1,777
  6. 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,528
  7. 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,406
  8. 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,327
  9. 2021Offline Reinforcement Learning with Implicit Q-Learningcited by 1,245
  10. 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
  11. 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
  12. 2020Mastering Atari with Discrete World Modelscited by 990
29 more
  1. 2021A Minimalist Approach to Offline Reinforcement Learningcited by 882
  2. 2019Stabilizing Off-Policy Q-Learning via Bootstrapping Error Reductioncited by 833
  3. 2016Pixel Recurrent Neural Networkscited by 814
  4. 2022A Generalist Agentcited by 802
  5. 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 636
  6. 2020MOPO: Model-based Offline Policy Optimizationcited by 630
  7. 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
  8. 2016The Option-Critic Architecturecited by 433
  9. 2019Learning Latent Plans from Playcited by 310
  10. 2021Mastering Atari Games with Limited Datacited by 235
  11. 2022Online Decision Transformercited by 194
  12. 2015Learning Continuous Control Policies by Stochastic Value Gradientscited by 156
  13. 2020PLAS: Latent Action Space for Offline Reinforcement Learningcited by 133
  14. 2019Reinforcement Learning Upside Down: Don't Predict Rewards -- Just Map Them to Actionscited by 124
  15. 2018Neural probabilistic motor primitives for humanoid controlcited by 96
  16. 2019Exploring Model-based Planning with Policy Networkscited by 90
  17. 2020Learning Search Space Partition for Black-box Optimization using Monte Carlo Tree Searchcited by 87
  18. 2019IRIS: Implicit Reinforcement without Interaction at Scale for Learning Control from Offline Robot Manipulation Datacited by 85
  19. 2020On the role of planning in model-based deep reinforcement learningcited by 57
  20. 2018Self-Consistent Trajectory Autoencoder: Hierarchical Reinforcement Learning with Trajectory Embeddingscited by 52
  21. 2021Revisiting Design Choices in Offline Model-Based Reinforcement Learningcited by 50
  22. 2021Vector Quantized Models for Planningcited by 47
  23. 2020On the model-based stochastic value gradient for continuous reinforcement learningcited by 46
  24. 2022Bootstrapped Transformer for Offline Reinforcement Learningcited by 37
  25. 2021LASER: Learning a Latent Action Space for Efficient Reinforcement Learningcited by 32
  26. 2021Continuous Control with Action Quantization from Demonstrationscited by 21
  27. 2021Learning Latent Actions to Control Assistive Robotscited by 17
  28. 2021Learning Visually Guided Latent Actions for Assistive Teleoperationcited by 12
  29. 2021Learning Space Partitions for Path Planningcited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2023: 15; 2024: 19; 2025: 14; 2026: 9.

57 papers.

Sort:

By area:

  1. 2024OGBench: Benchmarking Offline Goal-Conditioned RLcited by 216
  2. 2023HIQL: Offline Goal-Conditioned RL with Latent States as Actionscited by 133
  3. 2024Foundation Policies with Hilbert Representationscited by 89
  4. 2025Token Assorted: Mixing Latent and Text Tokens for Improved Language Model Reasoningcited by 87
  5. 2023The Wisdom of Hindsight Makes Language Models Better Instruction Followerscited by 59
  6. 2024Efficient Online Reinforcement Learning Fine-Tuning Need Not Retain Offline Datacited by 58
  7. 2023Masked Trajectory Models for Prediction, Representation, and Controlcited by 40
  8. 2025What Do Latent Action Models Actually Learn?cited by 30
  9. 2023A Generalist Dynamics Model for Controlcited by 24
  10. 2024PRISE: LLM-Style Sequence Compression for Learning Temporal Action Abstractions in Controlcited by 21
  11. 2023Reparameterized Policy Learning for Multimodal Trajectory Optimizationcited by 20
  12. 2023Recurrent Action Transformer with Memorycited by 17
38 more
  1. 2024Model-based Offline Reinforcement Learning with Lower Expectile Q-Learningcited by 17
  2. 2024TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representationscited by 17
  3. 2023IQL-TD-MPC: Implicit Q-Learning for Hierarchical Model Predictive Controlcited by 14
  4. 2023Rethinking Decision Transformer via Hierarchical Reinforcement Learningcited by 14
  5. 2024Latent Plan Transformer for Trajectory Abstraction: Planning as Latent Space Inferencecited by 14
  6. 2024Leveraging Skills from Unlabeled Prior Data for Efficient Online Explorationcited by 13
  7. 2025STAR: Learning Diverse Robot Skill Abstractions through Rotation-Augmented Vector Quantizationcited by 13
  8. 2025Continually Evolving Skill Knowledge in Vision Language Action Modelcited by 13
  9. 2023Instructed Diffuser with Temporal Condition Guidance for Offline Reinforcement Learningcited by 12
  10. 2025Scalable Offline Model-Based RL with Action Chunkscited by 12
  11. 2023Efficient Planning with Latent Diffusioncited by 11
  12. 2023H-GAP: Humanoid Control with a Generalist Plannercited by 11
  13. 2024Unsupervised-to-Online Reinforcement Learningcited by 11
  14. 2023Goal-Conditioned Predictive Coding for Offline Reinforcement Learningcited by 10
  15. 2024Premier-TACO is a Few-Shot Policy Learner: Pretraining Multitask Representation via Temporal Action-Driven Contrastive Losscited by 10
  16. 2024AD3: Implicit Action is the Key for World Models to Distinguish the Diverse Visual Distractorscited by 10
  17. 2025Penalizing Infeasible Actions and Reward Scaling in Reinforcement Learning with Offline Datacited by 7
  18. 2024PlanDQ: Hierarchical Plan Orchestration via D-Conductor and Q-Performercited by 6
  19. 2024Predictive Coding for Decision Transformercited by 6
  20. 2025Temporal Distance-aware Transition Augmentation for Offline Model-based Reinforcement Learningcited by 5
  21. 2025Offline Goal-Conditioned Reinforcement Learning with Projective Quasimetric Planningcited by 5
  22. 2026Chain-of-Goals Hierarchical Policy for Long-Horizon Offline Goal-Conditioned RLcited by 5
  23. 2023FoMo Rewards: Can we cast foundation models as reward functions?cited by 4
  24. 2024Planning Transformer: Long-Horizon Offline Reinforcement Learning with Planning Tokenscited by 4
  25. 2024Few-Shot Task Learning through Inverse Generative Modelingcited by 4
  26. 2025Scalable Decision-Making in Stochastic Environments through Learned Temporal Abstractioncited by 4
  27. 2023Subwords as Skills: Tokenization for Sparse-Reward Reinforcement Learningcited by 3
  28. 2023Emergence and Causality in Complex Systems: A Survey on Causal Emergence and Related Quantitative Studiescited by 3
  29. 2024Constrained Latent Action Policies for Model-Based Offline Reinforcement Learningcited by 3
  30. 2026Chunk-Guided Q-Learningcited by 3
  31. 2024Goal-Reaching Policy Learning from Non-Expert Observations via Effective Subgoal Guidancecited by 2
  32. 2024Navigation with QPHIL: Quantizing Planner for Hierarchical Implicit Q-Learningcited by 2
  33. 2025Analytic Energy-Guided Policy Optimization for Offline Reinforcement Learningcited by 2
  34. 2025Learning Skills from Action-Free Videoscited by 2
  35. 2026Occupancy Reward Shaping: Improving Credit Assignment for Offline Goal-Conditioned Reinforcement Learningcited by 1
  36. 2026Motion Planning in Compressed Representation Spacescited by 1
  37. 2024Are Expressive Models Truly Necessary for Offline RL?cited by 0
  38. 2025Decision Flow Policy Optimizationcited by 0

50 most cited of 57 All 57 (JSON)

Similar, not cited

Close in wording, no citation either way

  1. 2026TapSampling: Inference-Time Sampling with a Task-Progress-Understanding Verifier for Robotic Manipulation
  2. 2024RoLD: Robot Latent Diffusion for Multi-task Policy Modeling
  3. 2026AtomVLA: Scalable Post-Training for Robotic Manipulation via Predictive Latent World Models
  4. 2026LAMP: Latent Motion Prior-Guided Real-World Learning for Dexterous Hand Manipulation
  5. 2026GE-Act 2.0: Pretraining and Scaling a World-Action Model for Robotic Manipulation
  6. 2026What Matters for Latent Actions in Robot Learning
  7. 2025SITCOM: Scaling Inference-Time COMpute for VLAs
  8. 2026Representation World Model: Learning States, Transition and Executable Plans in Representation