Latent Action Reparameterization for Efficient Agent Inference
On the map: LLM agents and tool use → General LLM agent frameworks and benchmarks
Abstract (arXiv)
Large language model (LLM) agents often rely on long sequences of low-level textual actions, resulting in large effective decision horizons and high inference cost. While prior work has focused on improving inference efficiency through system-level optimizations or prompt engineering, we argue that a key bottleneck lies in the representation of the action space itself. We propose Latent Action Reparameterization (LAR), a framework that learns a compact latent action space in which each latent action corresponds to a multi-step semantic behavior. By reparameterizing agent actions into latent units, LAR enables decision making over a shorter effective horizon while preserving the expressiveness of the original action space. Unlike hand-crafted macros or hierarchical controllers, latent actions are learned from agent trajectories and integrated directly into the model, allowing both planning and execution to operate over abstract action representations. Across a range of LLM-based agent benchmarks, LAR significantly reduces the effective action horizon and improves inference efficiency under fixed compute budgets. As a consequence, our approach achieves substantial reductions in action tokens and corresponding wall-clock inference time, while maintaining or improving task success rates. These results suggest that action representation learning is a critical and underexplored factor in scaling efficient LLM agent inference, complementary to advances in model architecture and hardware.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in General LLM agent frameworks and benchmarks, one of 16 directions in LLM agents and tool use. Builds on 47 of its 51 references that are in the map. First posted on arXiv 18 May 2026.
Builds on
47 of 51 references.
Closest ancestors · 7
Same direction, newest first
- 2026Harnessing Agentic Evolution13 May
- 2025CostBench: Evaluating Multi-Turn Cost-Optimal Planning and Adaptation in Dynamic Environments for LLM Tool-Use Agents4 Nov
- 2025ReflAct: World-Grounded Decision Making in LLM Agents via Goal-State Reflection21 May
- 2025Robotouille: An Asynchronous Planning Benchmark for LLM Agents6 Feb
- 2024AgentQuest: A Modular Benchmark Framework to Measure Progress and Improve LLM Agents9 Apr
- 2023A Survey on Large Language Model based Autonomous Agents22 Aug
- 2023AgentBench: Evaluating LLMs as Agents7 Aug
Other references · 40
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
- 2025Qwen3 Technical Reportcited by 8,195
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,064
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2023Generative Agents: Interactive Simulacra of Human Behaviorcited by 3,500
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
28 more
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2022Least-to-Most Prompting Enables Complex Reasoning in Large Language Modelscited by 1,397
- 2023Mind2Web: Towards a Generalist Agent for the Webcited by 1,210
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
- 2024A Survey on the Memory Mechanism of Large Language Model based Agentscited by 622
- 2016The Option-Critic Architecturecited by 433
- 2024AFlow: Automating Agentic Workflow Generationcited by 395
- 2025Towards Reasoning Era: A Survey of Long Chain-of-Thought for Reasoning Large Language Modelscited by 381
- 2023Prompt Cache: Modular Attention Reuse for Low-Latency Inferencecited by 255
- 2025TokenSkip: Controllable Chain-of-Thought Compression in LLMscited by 246
- 2024A Survey on Efficient Inference for Large Language Modelscited by 197
- 2023Efficient Large Language Models: A Surveycited by 183
- 2022Distilling Reasoning Capabilities into Smaller Language Modelscited by 155
- 2025ACON: Optimizing Context Compression for Long-horizon LLM Agentscited by 126
- 2025KodCode: A Diverse, Challenging, and Verifiable Synthetic Dataset for Codingcited by 115
- 2023Guiding Large Language Models via Directional Stimulus Promptingcited by 94
- 2022Critic-Guided Decoding for Controlled Text Generationcited by 45
- 2024Enhancing Decision-Making for LLM Agents via Step-Level Q-Value Modelscited by 28
- 2024PRISE: LLM-Style Sequence Compression for Learning Temporal Action Abstractions in Controlcited by 21
- 2025AutoEnv: Automated Environments for Measuring Cross-Environment Agent Learningcited by 18
- 2024Hardware-Aware Parallel Prompt Decoding for Memory-Efficient Acceleration of LLM Inferencecited by 16
- 2024Efficient Sequential Decision Making with Large Language Modelscited by 16
- 2025FastMTP: Accelerating LLM Inference with Enhanced Multi-Token Predictioncited by 13
- 2025From What to Why: A Multi-Agent System for Evidence-based Chemical Reaction Condition Reasoningcited by 10
- 2025ARIA: Training Language Agents with Intention-Driven Reward Aggregationcited by 9
- 2025ConciseHint: Boosting Efficient Reasoning via Continuous Concise Hints during Generationcited by 9
- 2026Dual Latent Memory for Visual Multi-agent Systemcited by 9
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Multi-Head Latent Control: A Unified Interface for LLM Agent Decision Making
- 2025SAND: Boosting LLM Agents with Self-Taught Action Deliberation
- 2025Pre-Act: Multi-Step Planning and Reasoning Improves Acting in LLM Agents
- 2024Controlling Large Language Model Agents with Entropic Activation Steering
- 2026OLIVIA: Online Learning via Inference-time Action Adaptation for Decision Making in LLM ReAct Agents
- 2026AGORA: Adapter-Grounded Observation-Action Retention for Inference-Free Prompt Compression in LLM Agents
- 2024Empowering Large Language Model Agents through Action Learning
- 2026Why We Need World Models for AGI: Where LLMs Fail and How World Models May Outperform