Cache-Efficient Posterior Sampling for Reinforcement Learning with LLM-Derived Priors Across Discrete and Continuous Domains
Not on the map. Why?
Abstract (arXiv)
Integrating large language models (LLMs) as priors in reinforcement learning (RL) offers significant advantages but comes with substantial computational costs. We present a principled cache-efficient framework for posterior sampling with LLM-derived priors that dramatically reduces these costs while maintaining high performance. At the core of our approach is an adaptive caching mechanism, where cache parameters are meta-optimized using surrogate gradients derived from policy performance. This design enables efficient inference across both discrete text environments (e.g., TextWorld, ALFWorld) and continuous control domains (e.g., MuJoCo), achieving a 3.8--4.7× reduction in LLM queries and 4.0--12.0× lower median latencies (85--93,ms on a consumer GPU) while retaining 96--98% of uncached performance. Our theoretical analysis provides KL divergence bounds on approximation quality, validated empirically. The framework extends to offline RL, where our CQL-Prior variant improves performance by 14--29% and reduces training time by 38--40%. Extensive evaluations across a diverse suite of eight tasks demonstrate the generalizability and practical viability of LLM-guided RL in resource-constrained settings.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 31 of its 45 references that are in the map. First posted on arXiv 12 May 2025.
Builds on
31 of 45 references.
References · 31
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 3,080
- 2022Do As I Can, Not As I Say: Grounding Language in Robotic Affordancescited by 2,348
- 2023Voyager: An Open-Ended Embodied Agent with Large Language Modelscited by 2,191
- 2020Conservative Q-Learning for Offline Reinforcement Learningcited by 1,809
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
- 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,546
- 2023Mastering Diverse Domains through World Modelscited by 1,253
- 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,205
- 2019Habitat: A Platform for Embodied AI Researchcited by 1,191
19 more
- 2022Inner Monologue: Embodied Reasoning through Planning with Language Modelscited by 1,081
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,061
- 2022Language Models as Zero-Shot Planners: Extracting Actionable Knowledge for Embodied Agentscited by 1,035
- 2023Reasoning with Language Model is Planning with World Modelcited by 861
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 774
- 2023Eureka: Human-Level Reward Design via Coding Large Language Modelscited by 571
- 2022Temporal Difference Learning for Model Predictive Controlcited by 397
- 2019Efficient Off-Policy Meta-Reinforcement Learning via Probabilistic Context Variablescited by 389
- 2018Quantifying Generalization in Reinforcement Learningcited by 352
- 2018TextWorld: A Learning Environment for Text-based Gamescited by 310
- 2023Reward Design with Language Modelscited by 261
- 2023Building Cooperative Embodied Agents Modularly with Large Language Modelscited by 237
- 2021ManiSkill: Generalizable Manipulation Skill Benchmark with Large-Scale Demonstrationscited by 197
- 2023Grounding Large Language Models in Interactive Environments with Online Reinforcement Learningcited by 195
- 2023Guiding Pretraining in Reinforcement Learning with Large Language Modelscited by 187
- 2015Deep Reinforcement Learning in Large Discrete Action Spacescited by 161
- 2018Meta-Gradient Reinforcement Learningcited by 159
- 2024Efficient Reinforcement Learning with Large Language Model Priorscited by 30
- 2022Retro-RL: Reinforcing Nominal Controller With Deep Reinforcement Learning for Tilting-Rotor Dronescited by 2
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026DistillCache: KL-Guided Adaptive KV-Cache Eviction for Memory-Efficient LLM Inference
- 2026A Survey of Reinforcement Learning for Large Language Models under Data Scarcity: Challenges and Solutions
- 2025Toward Efficient Exploration by Large Language Model Agents
- 2025Enhancing Efficiency and Exploration in Reinforcement Learning for LLMs
- 2025Improving Data Efficiency for LLM Reinforcement Fine-tuning Through Difficulty-targeted Online Data Selection and Rollout Replay
- 2025d2Cache: Accelerating Diffusion-Based LLMs via Dual Adaptive Caching
- 2026Small RL Controller, Large Language Model: RL-Guided Adaptive Sampling for Test-Time Scaling
- 2026Sparse-RL: Breaking the Memory Wall in LLM Reinforcement Learning via Stable Sparse Rollouts