mlsanityData to 7 Oct 2026

Q-learning Decision Transformer: Leveraging Dynamic Programming for Conditional Sequence Modelling in Offline RL

arXiv 2209.03993 ↗First posted 8 Sep 2022cs.LGcited by 103

On the map: Offline RL and imitation learning from data → Decision transformers and sequence modeling

Abstract (arXiv)

Recent works have shown that tackling offline reinforcement learning (RL) with a conditional policy produces promising results. The Decision Transformer (DT) combines the conditional policy approach and a transformer architecture, showing competitive performance against several benchmarks. However, DT lacks stitching ability -- one of the critical abilities for offline RL to learn the optimal policy from sub-optimal trajectories. This issue becomes particularly significant when the offline dataset only contains sub-optimal trajectories. On the other hand, the conventional RL approaches based on Dynamic Programming (such as Q-learning) do not have the same limitation; however, they suffer from unstable learning behaviours, especially when they rely on function approximation in an off-policy learning setting. In this paper, we propose the Q-learning Decision Transformer (QDT) to address the shortcomings of DT by leveraging the benefits of Dynamic Programming (Q-learning). It utilises the Dynamic Programming results to relabel the return-to-go in the training data to then train the DT with the relabelled data. Our approach efficiently exploits the benefits of these two approaches and compensates for each other's shortcomings to achieve better performance. We empirically show these in both simple toy environments and the more complex D4RL benchmark, showing competitive performance gains.

Key facts. Cited by 103 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 36 of its 52 references that are in the map. First posted on arXiv 8 Sep 2022.

Builds on

36 of 52 references.

Closest ancestors · 2

Same direction, newest first

  1. 2021Generalized Decision Transformer for Offline Hindsight Information Matching19 Nov
  2. 2021Decision Transformer: Reinforcement Learning via Sequence Modeling2 Jun

Other references · 34

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2021Zero-Shot Text-to-Image Generationcited by 4,276
  4. 2020Conservative Q-Learning for Offline Reinforcement Learningcited by 1,777
  5. 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,406
  6. 2018Off-Policy Deep Reinforcement Learning without Explorationcited by 1,227
  7. 2021A Minimalist Approach to Offline Reinforcement Learningcited by 882
  8. 2019Stabilizing Off-Policy Q-Learning via Bootstrapping Error Reductioncited by 833
  9. 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 781
  10. 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
  11. 2019Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learningcited by 750
  12. 2019Behavior Regularized Offline Reinforcement Learningcited by 654
22 more
  1. 2020AWAC: Accelerating Online Reinforcement Learning with Offline Datasetscited by 647
  2. 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
  3. 2017End-to-end Driving via Conditional Imitation Learningcited by 479
  4. 2021COMBO: Conservative Offline Model-Based Policy Optimizationcited by 365
  5. 2019Learning Latent Plans from Playcited by 310
  6. 2020Critic Regularized Regressioncited by 269
  7. 2016Safe and Efficient Off-Policy Reinforcement Learningcited by 260
  8. 2022A Survey on Offline Reinforcement Learning: Taxonomy, Review, and Open Problemscited by 228
  9. 2020Keep Doing What Worked: Behavioral Modelling Priors for Offline Reinforcement Learningcited by 213
  10. 2021RvS: What is Essential for Offline RL via Supervised Learning?cited by 184
  11. 2020Hyperparameter Selection for Offline Reinforcement Learningcited by 113
  12. 2019Training Agents using Upside-Down Reinforcement Learningcited by 106
  13. 2019Reward-Conditioned Policiescited by 101
  14. 2022Don't Change the Algorithm, Change the Data: Exploratory Data for Offline Reinforcement Learningcited by 93
  15. 2019BAIL: Best-Action Imitation Learning for Batch Deep Reinforcement Learningcited by 90
  16. 2021Benchmarks for Deep Off-Policy Evaluationcited by 89
  17. 2022When Should We Prefer Offline Reinforcement Learning Over Behavioral Cloning?cited by 86
  18. 2021A Workflow for Offline Model-Free Robotic Reinforcement Learningcited by 68
  19. 2017Multi-step Reinforcement Learning: A Unifying Algorithmcited by 42
  20. 2019Understanding Multi-Step Deep Reinforcement Learning: A Systematic Study of the DQN Targetcited by 31
  21. 2013Off-policy Learning with Eligibility Traces: A Surveycited by 12
  22. 2021Quantile Filtered Imitation Learningcited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2022: 1; 2023: 19; 2024: 35; 2025: 26; 2026: 22.

103 papers.

Sort:

By area:

  1. 2025π*0.6: a VLA That Learns From Experiencecited by 280
  2. 2023Q-Transformer: Scalable Offline Reinforcement Learning via Autoregressive Q-Functionscited by 129
  3. 2024Policy Agnostic RL: Offline RL and Online RL Fine-Tuning of Any Class and Backbonecited by 71
  4. 2023Elastic Decision Transformercited by 66
  5. 2025Diffusion Guidance Is a Controllable Policy Improvement Operatorcited by 51
  6. 2023A Survey on Transformers in Reinforcement Learningcited by 50
  7. 2025ReinboT: Amplifying Robot Visual-Language Manipulation with Reinforcement Learningcited by 44
  8. 2023Generalization to New Sequential Decision Making Tasks with In-Context Learningcited by 39
  9. 2025Generative Trajectory Stitching through Diffusion Compositioncited by 38
  10. 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectorycited by 35
  11. 2024Diffusion World Model: Future Modeling Beyond Step-by-Step Rollout for Offline Reinforcement Learningcited by 31
  12. 2023Emergence of In-Context Reinforcement Learning from Noise Distillationcited by 26
38 more
  1. 2024Closing the Gap between TD Learning and Supervised Learning -- A Generalisation Point of Viewcited by 26
  2. 2023Critic-Guided Decision Transformer for Offline Reinforcement Learningcited by 25
  3. 2024Reinformer: Max-Return Sequence Modeling for Offline RLcited by 25
  4. 2023ACT: Empowering Decision Transformer with Dynamic Programming via Advantage Conditioningcited by 24
  5. 2023Towards Robust Offline Reinforcement Learning under Diverse Data Corruptioncited by 24
  6. 2023Unleashing the Power of Pre-trained Language Models for Offline Reinforcement Learningcited by 24
  7. 2023Waypoint Transformer: Reinforcement Learning via Supervised Learning with Intermediate Targetscited by 22
  8. 2023Zero-Shot Reinforcement Learning from Low Quality Datacited by 21
  9. 2023Graph Decision Transformercited by 20
  10. 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directionscited by 20
  11. 2025floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RLcited by 20
  12. 2023Prompt-Tuning Decision Transformer with Preference Rankingcited by 19
  13. 2024TOP-ERL: Transformer-based Off-Policy Episodic Reinforcement Learningcited by 18
  14. 2023Recurrent Action Transformer with Memorycited by 17
  15. 2024Decision Mamba: A Multi-Grained State Space Model with Self-Evolution Regularization for Offline RLcited by 15
  16. 2023Rethinking Decision Transformer via Hierarchical Reinforcement Learningcited by 14
  17. 2024Latent Plan Transformer for Trajectory Abstraction: Planning as Latent Space Inferencecited by 14
  18. 2024HarmoDT: Harmony Multi-Task Decision Transformer for Offline Reinforcement Learningcited by 14
  19. 2023Transformers in Reinforcement Learning: A Surveycited by 11
  20. 2023Free from Bellman Completeness: Trajectory Stitching via Model-based Return-conditioned Supervised Learningcited by 11
  21. 2024Temporal Logic Specification-Conditioned Decision Transformer for Offline Safe Reinforcement Learningcited by 10
  22. 2024A2PO: Towards Effective Offline Reinforcement Learning from an Advantage-aware Perspectivecited by 10
  23. 2024Rethinking Transformers in Solving POMDPscited by 9
  24. 2025Self-Predictive Representations for Combinatorial Generalization in Behavioral Cloningcited by 9
  25. 2024Reinforcement Learning Gradients as Vitamin for Online Finetuning Decision Transformerscited by 8
  26. 2025Yes, Q-learning Helps Offline In-Context RLcited by 8
  27. 2024Adaptive Q-Aid for Conditional Supervised Learning in Offline Reinforcement Learningcited by 7
  28. 2024Tackling Data Corruption in Offline Reinforcement Learning via Sequence Modelingcited by 7
  29. 2024PlanDQ: Hierarchical Plan Orchestration via D-Conductor and Q-Performercited by 6
  30. 2024Predictive Coding for Decision Transformercited by 6
  31. 2025NavQ: Learning a Q-Model for Foresighted Vision-and-Language Navigationcited by 6
  32. 2025Adaptive Context Length Optimization with Low-Frequency Truncation for Multi-Agent Reinforcement Learningcited by 6
  33. 2026Flow Actor-Critic for Offline Reinforcement Learningcited by 6
  34. 2023PASTA: Pretrained Action-State Transformer Agentscited by 5
  35. 2024Decision Transformer as a Foundation Model for Partially Observable Continuous Controlcited by 5
  36. 2024GARLIC: GPT-Augmented Reinforcement Learning with Intelligent Control for Vehicle Dispatchingcited by 5
  37. 2025Scalable In-Context Q-Learningcited by 5
  38. 2024Safe and Robust Reinforcement Learning: Principles and Practicecited by 4

50 most cited of 103 All 103 (JSON)

Similar, not cited

Close in wording, no citation either way

  1. 2024Q-value Regularized Transformer for Offline Reinforcement Learning
  2. 2025A Comparison Between Decision Transformers and Traditional Offline Reinforcement Learning Algorithms
  3. 2023When should we prefer Decision Transformers for Offline Reinforcement Learning?
  4. 2024Return Augmented Decision Transformer for Off-Dynamics Reinforcement Learning
  5. 2021Offline Reinforcement Learning as One Big Sequence Modeling Problem
  6. 2023Multi-Objective Decision Transformers for Offline Reinforcement Learning
  7. 2023Decision S4: Efficient Sequence-Based RL via State Spaces Layers
  8. 2022Bootstrapped Transformer for Offline Reinforcement Learning