Unleashing the Power of Pre-trained Language Models for Offline Reinforcement Learning
On the map: Offline RL and imitation learning from data → Decision transformers and sequence modeling
Abstract (arXiv)
Offline reinforcement learning (RL) aims to find a near-optimal policy using pre-collected datasets. In real-world scenarios, data collection could be costly and risky; therefore, offline RL becomes particularly challenging when the in-domain data is limited. Given recent advances in Large Language Models (LLMs) and their few-shot learning prowess, this paper introduces Language Models for Motion Control (LaMo), a general framework based on Decision Transformers to effectively use pre-trained Language Models (LMs) for offline RL. Our framework highlights four crucial components: (1) Initializing Decision Transformers with sequentially pre-trained LMs, (2) employing the LoRA fine-tuning method, in contrast to full-weight fine-tuning, to combine the pre-trained knowledge from LMs and in-domain knowledge effectively, (3) using the non-linear MLP transformation instead of linear projections, to generate embeddings, and (4) integrating an auxiliary language prediction loss during fine-tuning to stabilize the LMs and retain their original abilities on languages. Empirical results indicate LaMo achieves excellent performance in sparse-reward tasks and closes the gap between value-based offline RL methods and decision transformers in dense-reward tasks. In particular, our method demonstrates superior performance in scenarios with limited data samples.
Key facts. Cited by 24 papers in the map (data through 7 Oct 2026). Placed in Decision transformers and sequence modeling, one of 13 directions in Offline RL and imitation learning from data. Builds on 59 of its 69 references that are in the map. First posted on arXiv 31 Oct 2023.
Builds on
59 of 69 references.
Closest ancestors · 8
Same direction, newest first
- 2023Future-conditioned Unsupervised Pretraining for Decision Transformer26 May
- 2023When should we prefer Decision Transformers for Offline Reinforcement Learning?23 May
- 2023Decision Transformer under Random Frame Dropping3 Mar
- 2022Q-learning Decision Transformer: Leveraging Dynamic Programming for Conditional Sequence Modelling in Offline RL8 Sep
- 2022Prompting Decision Transformer for Few-Shot Policy Generalization27 Jun
- 2022Can Wikipedia Help Offline Reinforcement Learning?28 Jan
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problem3 Jun
- 2021Decision Transformer: Reinforcement Learning via Sequence Modeling2 Jun
Other references · 51
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2023GPT-4 Technical Reportcited by 18,132
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
- 2022Large Language Models are Zero-Shot Reasonerscited by 4,323
38 more
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2022Scaling Instruction-Finetuned Language Modelscited by 2,997
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 2,993
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2023PaLM-E: An Embodied Multimodal Language Modelcited by 2,037
- 2022RT-1: Robotics Transformer for Real-World Control at Scalecited by 2,007
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2020Conservative Q-Learning for Offline Reinforcement Learningcited by 1,777
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,406
- 2021Offline Reinforcement Learning with Implicit Q-Learningcited by 1,245
- 2018Off-Policy Deep Reinforcement Learning without Explorationcited by 1,227
- 2022Language Models as Zero-Shot Planners: Extracting Actionable Knowledge for Embodied Agentscited by 1,029
- 2021Transformers in Vision: A Surveycited by 1,009
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2022Few-Shot Parameter-Efficient Fine-Tuning is Better and Cheaper than In-Context Learningcited by 886
- 2021A Minimalist Approach to Offline Reinforcement Learningcited by 882
- 2022A Generalist Agentcited by 802
- 2023LLaMA-Adapter: Efficient Fine-tuning of Language Models with Zero-init Attentioncited by 776
- 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 658
- 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learningcited by 543
- 2019Relay Policy Learning: Solving Long-Horizon Tasks via Imitation and Reinforcement Learningcited by 380
- 2022A Survey on Offline Reinforcement Learning: Taxonomy, Review, and Open Problemscited by 228
- 2023Text2Motion: From Natural Language Instructions to Feasible Planscited by 223
- 2022Pre-Trained Language Models for Interactive Decision-Makingcited by 220
- 2023TidyBot: Personalized Robot Assistance with Large Language Modelscited by 219
- 2022In-context Reinforcement Learning with Algorithm Distillationcited by 186
- 2023Revisiting the Minimalist Approach to Offline Reinforcement Learningcited by 121
- 2022CORL: Research-oriented Deep Offline Reinforcement Learning Librarycited by 117
- 2021d3rlpy: An Offline Deep Reinforcement Learning Librarycited by 84
- 2022On Pre-Training for Visuo-Motor Control: Revisiting a Learning-from-Scratch Baselinecited by 55
- 2022Visual Reinforcement Learning with Self-Supervised 3D Representationscited by 45
- 2023Prompt a Robot to Walk with Large Language Modelscited by 44
- 2023SayTap: Language to Quadrupedal Locomotioncited by 40
- 2023Emergent Agentic Transformer from Chain of Hindsight Experiencecited by 33
- 2023Prompt-Tuning Decision Transformer with Preference Rankingcited by 19
- 2020Reservoir Transformerscited by 18
50 most cited of 51 All 51 (JSON)
Built on by
Citing papers per year: 2024: 15; 2025: 7; 2026: 2.
24 papers.
Sort:
By area:
- 2024TinyVLA: Towards Fast, Data-Efficient Vision-Language-Action Models for Robotic Manipulationcited by 331
- 2024Agent AI: Surveying the Horizons of Multimodal Interactioncited by 167
- 2024Survey on Large Language Model-Enhanced Reinforcement Learning: Concept, Taxonomy, and Methodscited by 114
- 2024From Multimodal LLMs to Generalist Embodied Agents: Methods and Lessonscited by 42
- 2024Grounding Multimodal Large Language Models in Actionscited by 35
- 2024iLLM-TSC: Integration reinforcement learning and large language model for traffic signal control policy improvementcited by 23
- 2025Parallels Between VLA Model Post-Training and Human Motor Learning: Progress, Challenges, and Trendscited by 17
- 2024Decision Mamba: A Multi-Grained State Space Model with Self-Evolution Regularization for Offline RLcited by 15
- 2024Stop Wandering, Find the Keys: LLMs Discriminate Key States for Efficient Multi-Agent Explorationcited by 9
- 2024Tackling Data Corruption in Offline Reinforcement Learning via Sequence Modelingcited by 7
- 2024Pre-trained Language Models Improve the Few-shot Prompt Ability of Decision Transformercited by 6
- 2024Zero-shot Model-based Reinforcement Learning using Large Language Modelscited by 6
12 more
- 2024Decision Transformer as a Foundation Model for Partially Observable Continuous Controlcited by 5
- 2025Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervisioncited by 5
- 2025HCRMP: A LLM-Hinted Contextual Reinforcement Learning Framework for Autonomous Drivingcited by 4
- 2024Robust RL with LLM-Driven Data Synthesis and Policy Adaptation for Autonomous Drivingcited by 3
- 2024Guiding Reinforcement Learning Using Uncertainty-Aware Large Language Modelscited by 3
- 2025Prompt Tuning Decision Transformers with Structured and Scalable Banditscited by 2
- 2025Large Language Model-enhanced Reinforcement Learning for Low-Altitude Economy Networkingcited by 2
- 2025ADG: Ambient Diffusion-Guided Dataset Recovery for Corruption-Robust Offline Reinforcement Learningcited by 2
- 2026LLM-Guided Communication for Cooperative Multi-Agent Reinforcement Learningcited by 2
- 2025Agentic Episodic Controlcited by 1
- 2024Multi-State-Action Tokenisation in Decision Transformers for Multi-Discrete Action Spacescited by 0
- 2026MIRA: Memory-Integrated Reinforcement Learning Agent with Limited LLM Guidancecited by 0
- 2026LLM-Guided Communication for Cooperative Multi-Agent Reinforcement Learning18 May 2026
- 2026MIRA: Memory-Integrated Reinforcement Learning Agent with Limited LLM Guidance20 Feb 2026
- 2025Parallels Between VLA Model Post-Training and Human Motor Learning: Progress, Challenges, and Trends26 Jun 2025
- 2025Agentic Episodic Control2 Jun 2025
- 2025ADG: Ambient Diffusion-Guided Dataset Recovery for Corruption-Robust Offline Reinforcement Learning29 May 2025
- 2025Large Language Model-enhanced Reinforcement Learning for Low-Altitude Economy Networking27 May 2025
- 2025HCRMP: A LLM-Hinted Contextual Reinforcement Learning Framework for Autonomous Driving21 May 2025
- 2025Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervision21 Apr 2025
- 2025Prompt Tuning Decision Transformers with Structured and Scalable Bandits7 Feb 2025
- 2024From Multimodal LLMs to Generalist Embodied Agents: Methods and Lessons11 Dec 2024
- 2024Guiding Reinforcement Learning Using Uncertainty-Aware Large Language Models15 Nov 2024
- 2024Robust RL with LLM-Driven Data Synthesis and Policy Adaptation for Autonomous Driving16 Oct 2024
12 more
- 2024Zero-shot Model-based Reinforcement Learning using Large Language Models15 Oct 2024
- 2024Stop Wandering, Find the Keys: LLMs Discriminate Key States for Efficient Multi-Agent Exploration3 Oct 2024
- 2024TinyVLA: Towards Fast, Data-Efficient Vision-Language-Action Models for Robotic Manipulation19 Sep 2024
- 2024Pre-trained Language Models Improve the Few-shot Prompt Ability of Decision Transformer2 Aug 2024
- 2024iLLM-TSC: Integration reinforcement learning and large language model for traffic signal control policy improvement8 Jul 2024
- 2024Tackling Data Corruption in Offline Reinforcement Learning via Sequence Modeling5 Jul 2024
- 2024Multi-State-Action Tokenisation in Decision Transformers for Multi-Discrete Action Spaces1 Jul 2024
- 2024Grounding Multimodal Large Language Models in Actions12 Jun 2024
- 2024Decision Mamba: A Multi-Grained State Space Model with Self-Evolution Regularization for Offline RL8 Jun 2024
- 2024Decision Transformer as a Foundation Model for Partially Observable Continuous Control3 Apr 2024
- 2024Survey on Large Language Model-Enhanced Reinforcement Learning: Concept, Taxonomy, and Methods30 Mar 2024
- 2024Agent AI: Surveying the Horizons of Multimodal Interaction7 Jan 2024
- 2024Tackling Data Corruption in Offline Reinforcement Learning via Sequence Modelingcited by 7
- 2024Pre-trained Language Models Improve the Few-shot Prompt Ability of Decision Transformercited by 6
- 2024Decision Transformer as a Foundation Model for Partially Observable Continuous Controlcited by 5
- 2025Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervisioncited by 5
- 2025Prompt Tuning Decision Transformers with Structured and Scalable Banditscited by 2
- 2024Multi-State-Action Tokenisation in Decision Transformers for Multi-Discrete Action Spacescited by 0
- 2025Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervision21 Apr 2025
- 2025Prompt Tuning Decision Transformers with Structured and Scalable Bandits7 Feb 2025
- 2024Pre-trained Language Models Improve the Few-shot Prompt Ability of Decision Transformer2 Aug 2024
- 2024Tackling Data Corruption in Offline Reinforcement Learning via Sequence Modeling5 Jul 2024
- 2024Multi-State-Action Tokenisation in Decision Transformers for Multi-Discrete Action Spaces1 Jul 2024
- 2024Decision Transformer as a Foundation Model for Partially Observable Continuous Control3 Apr 2024
- 2024Agent AI: Surveying the Horizons of Multimodal Interactioncited by 167
- 2024Stop Wandering, Find the Keys: LLMs Discriminate Key States for Efficient Multi-Agent Explorationcited by 9
- 2026LLM-Guided Communication for Cooperative Multi-Agent Reinforcement Learningcited by 2
- 2025Agentic Episodic Controlcited by 1
- 2026MIRA: Memory-Integrated Reinforcement Learning Agent with Limited LLM Guidancecited by 0
- 2026LLM-Guided Communication for Cooperative Multi-Agent Reinforcement Learning18 May 2026
- 2026MIRA: Memory-Integrated Reinforcement Learning Agent with Limited LLM Guidance20 Feb 2026
- 2025Agentic Episodic Control2 Jun 2025
- 2024Stop Wandering, Find the Keys: LLMs Discriminate Key States for Efficient Multi-Agent Exploration3 Oct 2024
- 2024Agent AI: Surveying the Horizons of Multimodal Interaction7 Jan 2024
- 2024Survey on Large Language Model-Enhanced Reinforcement Learning: Concept, Taxonomy, and Methodscited by 114
- 2024iLLM-TSC: Integration reinforcement learning and large language model for traffic signal control policy improvementcited by 23
- 2024Zero-shot Model-based Reinforcement Learning using Large Language Modelscited by 6
- 2024Guiding Reinforcement Learning Using Uncertainty-Aware Large Language Modelscited by 3
- 2025Large Language Model-enhanced Reinforcement Learning for Low-Altitude Economy Networkingcited by 2
- 2025Large Language Model-enhanced Reinforcement Learning for Low-Altitude Economy Networking27 May 2025
- 2024Guiding Reinforcement Learning Using Uncertainty-Aware Large Language Models15 Nov 2024
- 2024Zero-shot Model-based Reinforcement Learning using Large Language Models15 Oct 2024
- 2024iLLM-TSC: Integration reinforcement learning and large language model for traffic signal control policy improvement8 Jul 2024
- 2024Survey on Large Language Model-Enhanced Reinforcement Learning: Concept, Taxonomy, and Methods30 Mar 2024
Similar, not cited
Close in wording, no citation either way
- 2024Unveiling Markov Heads in Pretrained Language Models for Offline Reinforcement Learning
- 2023Leftover Lunch: Advantage-based Offline Reinforcement Learning for Language Models
- 2024MORE-3S:Multimodal-based Offline Reinforcement Learning with Shared Semantic Spaces
- 2025Exploring the Potential of Offline RL for Reasoning in LLMs: A Preliminary Study
- 2026Leveraging Offline Supervision for Efficient and Generalizable Reinforcement Learning in Large-Scale Vision-Language-Action Models
- 2024Q-SFT: Q-Learning for Language Models via Supervised Fine-Tuning
- 2022Contextual Transformer for Offline Meta Reinforcement Learning
- 2025Recall-Extend Dynamics: Enhancing Small Language Models through Controlled Exploration and Refined Offline Integration