mlsanityData to 14 Oct 2026

Towards Scalable Multi-Task Reinforcement Learning with Large Decision Models

arXiv 2606.24962 ↗First posted 23 Jun 2026cs.LGcited by 0

On the map: Deep reinforcement learning methods → Deep RL algorithms and representations

Abstract (arXiv)

Recent progress in large-scale sequence modeling has shown that a single model can learn useful representations across highly diverse data distributions. Inspired by these advances, we investigate whether a unified transformer policy can be trained across large collections of heterogeneous reinforcement learning environments. We introduce LDM-v0, a Large Decision Model trained offline on trajectories collected from thousands of environments spanning multiple domains and modalities. LDM-v0 is a multi-task, multi-modal transformer policy conditioned on histories of observations, actions, rewards, and termination signals, and trained through supervised next-action prediction over offline trajectories. We describe the environment infrastructure, automated data generation pipeline, model architecture, and training methodology used to build LDM-v0, and evaluate its performance across diverse environments. We show that a single pretrained model matches the performance of independently trained task-specific reference policies on approximately 1,000 environments including robotics, autonomous driving, inventory management, cybersecurity, trading, and video games. These results demonstrate the feasibility of large-scale offline pretraining across heterogeneous reinforcement learning environments using a single transformer policy.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Deep RL algorithms and representations, one of 15 directions in Deep reinforcement learning methods. Builds on 17 of its 18 references that are in the map. First posted on arXiv 23 Jun 2026.

Builds on

17 of 18 references.

Closest ancestors · 1

Same direction, newest first

  1. 2024AMAGO-2: Breaking the Multi-Task Barrier in Meta-Reinforcement Learning with Transformers17 Nov

Other references · 16

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
  4. 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,546
  5. 2022A Generalist Agentcited by 808
  6. 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 644
  7. 2016RL2: Fast Reinforcement Learning via Slow Reinforcement Learningcited by 503
  8. 2022In-context Reinforcement Learning with Algorithm Distillationcited by 190
  9. 2023Supervised Pretraining Can Learn In-Context Reinforcement Learningcited by 139
  10. 2023AMAGO: Scalable In-Context Reinforcement Learning for Adaptive Agentscited by 41
  11. 2023Generalization to New Sequential Decision Making Tasks with In-Context Learningcited by 39
  12. 2024Jack of All Trades, Master of Some, a Multi-Purpose Transformer Agentcited by 21
4 more
  1. 2024REGENT: A Retrieval-Augmented Generalist Agent That Can Act In-Context in New Environmentscited by 16
  2. 2022Data-Efficient Pipeline for Offline Reinforcement Learning with Limited Datacited by 12
  3. 2023Transformers in Reinforcement Learning: A Surveycited by 12
  4. 2024Understanding the Training and Generalization of Pretrained Transformer for Sequential Decision Makingcited by 5

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2023Rethinking Decision Transformer via Hierarchical Reinforcement Learning
  2. 2026Vintix II: Decision Pre-Trained Transformer is a Scalable In-Context Reinforcement Learner
  3. 2025Mastering Massive Multi-Task Reinforcement Learning via Mixture-of-Expert Decision Transformer
  4. 2023Multi-Objective Decision Transformers for Offline Reinforcement Learning
  5. 2026DecisionLLM: Large Language Models for Long Sequence Decision Exploration
  6. 2022Multi-Game Decision Transformers
  7. 2023Transformer-based World Models Are Happy With 100k Interactions
  8. 2021Offline Pre-trained Multi-Agent Decision Transformer: One Big Sequence Model Tackles All SMAC Tasks