Distral: Robust Multitask Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Most deep reinforcement learning algorithms are data inefficient in complex and rich environments, limiting their applicability to many scenarios. One direction for improving data efficiency is multitask learning with shared neural network parameters, where efficiency may be improved through transfer across related tasks. In practice, however, this is not usually observed, because gradients from different tasks can interfere negatively, making learning unstable and sometimes even less data efficient. Another issue is the different reward schemes between tasks, which can easily lead to one task dominating the learning of a shared model. We propose a new approach for joint training of multiple tasks, which we refer to as Distral (Distill & transfer learning). Instead of sharing parameters between the different workers, we propose to share a "distilled" policy that captures common behaviour across tasks. Each worker is trained to solve its own task while constrained to stay close to the shared policy, while the shared policy is trained by distillation to be the centroid of all task policies. Both aspects of the learning process are derived by optimizing a joint objective function. We show that our approach supports efficient transfer on complex 3D environments, outperforming several related methods. Moreover, the proposed learning process is more robust and more stable---attributes that are critical in deep reinforcement learning.
Key facts. Cited by 231 papers in the map (data through 7 Oct 2026). First posted on arXiv 13 Jul 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 0; 2020: 2; 2021: 48; 2022: 57; 2023: 34; 2024: 27; 2025: 32; 2026: 30.
231 papers.
Sort:
By area:
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
- 2022The Primacy Bias in Deep Reinforcement Learningcited by 242
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 232
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scalecited by 226
- 2023RVT: Robotic View Transformer for 3D Object Manipulationcited by 224
- 2020Keep Doing What Worked: Behavioral Modelling Priors for Offline Reinforcement Learningcited by 213
- 2020Language Conditioned Imitation Learning over Unstructured Datacited by 208
- 2021Open-Ended Learning Leads to Generally Capable Agentscited by 177
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021Goal-Conditioned Reinforcement Learning with Imagined Subgoalscited by 129
- 2022Understanding and Preventing Capacity Loss in Reinforcement Learningcited by 124
38 more
- 2023Learning Agile Soccer Skills for a Bipedal Robot with Deep Reinforcement Learningcited by 111
- 2021Why Generalization in RL is Difficult: Epistemic POMDPs and Implicit Partial Observabilitycited by 102
- 2023UniDexGrasp++: Improving Dexterous Grasping Policy Learning via Geometry-aware Curriculum and Iterative Generalist-Specialist Learningcited by 101
- 2023Confronting Reward Model Overoptimization with Constrained RLHFcited by 96
- 2021Reset-Free Reinforcement Learning via Multi-Task Learning: Learning Dexterous Manipulation Behaviors without Human Interventioncited by 90
- 2021From Motor Control to Team Play in Simulated Humanoid Footballcited by 87
- 2022Pre-Training for Robots: Offline RL Enables Learning New Tasks from a Handful of Trialscited by 78
- 2022Offline Q-Learning on Diverse Multi-Task Data Both Scales And Generalizescited by 72
- 2023Independent Component Alignment for Multi-Task Learningcited by 69
- 2022In Defense of the Unitary Scalarization for Deep Multi-Task Learningcited by 65
- 2021Conservative Data Sharing for Multi-Task Offline Reinforcement Learningcited by 64
- 2022Federated Reinforcement Learning with Environment Heterogeneitycited by 61
- 2021Learning to Synthesize Programs as Interpretable and Generalizable Policiescited by 59
- 2021Robot Learning from Randomized Simulations: A Reviewcited by 56
- 2022Retrieval-Augmented Reinforcement Learningcited by 53
- 2021Temporal Abstraction in Reinforcement Learning with the Successor Representationcited by 52
- 2021Generalization in Dexterous Manipulation via Geometry-Aware Multi-Task Learningcited by 49
- 2021AdaRL: What, Where, and How to Adapt in Transfer Reinforcement Learningcited by 46
- 2021Near-optimal Representation Learning for Linear Bandits and Linear RLcited by 45
- 2021Bootstrapped Meta-Learningcited by 45
- 2023Multi-Task Reinforcement Learning with Mixture of Orthogonal Expertscited by 45
- 2021SECANT: Self-Expert Cloning for Zero-Shot Generalization of Visual Policiescited by 42
- 2024Sharing Knowledge in Multi-Task Deep Reinforcement Learningcited by 42
- 2024RLDG: Robotic Generalist Policy Distillation via Reinforcement Learningcited by 42
- 2022PaCo: Parameter-Compositional Multi-Task Reinforcement Learningcited by 40
- 2022Hypernetworks in Meta-Reinforcement Learningcited by 39
- 2021Interesting Object, Curious Agent: Learning Task-Agnostic Explorationcited by 38
- 2021Reinforcement Learning for Adaptive Mesh Refinementcited by 37
- 2022Skill-based Meta-Reinforcement Learningcited by 36
- 2023Scaling Pareto-Efficient Decision Making Via Offline Multi-Objective RLcited by 36
- 2021Efficient Transformers in Reinforcement Learning using Actor-Learner Distillationcited by 35
- 2021Lifelong Robotic Reinforcement Learning by Retaining Experiencescited by 34
- 2021On the Power of Multitask Representation Learning in Linear MDPcited by 30
- 2021Universal Trading for Order Execution with Oracle Policy Distillationcited by 29
- 2022Provable Benefit of Multitask Representation Learning in Reinforcement Learningcited by 29
- 2022Don't Start From Scratch: Leveraging Prior Data to Automate Robotic Reinforcement Learningcited by 28
- 2022Towards Versatile Embodied Navigationcited by 26
- 2021Block Contextual MDPs for Continual Learningcited by 25
50 most cited of 231 All 231 (JSON)
- 2026TelecomGPT-R1: Unified Post-Training for Reasoning Across Heterogeneous Telecom Tasks21 Sep 2026
- 2026HarnessBandit: Joint Learnability-Transferability Scheduling for Multi-Harness Agentic Reinforcement Learning12 Sep 2026
- 2026PAC: Progress-Augmented Advantage Curriculum for Multi-Task Reinforcement Learning of LLMs31 Aug 2026
- 2026Task Specialization Fine-Tuning for Contextual Reinforcement Learning17 Aug 2026
- 2026Multi-Task Learning for Heterogeneous Prediction from Video Game State with Transfer Learning23 Jul 2026
- 2026Building2Building: A Large Scale Benchmark for Generalizable Real-World Reinforcement Learning17 Jul 2026
- 2026Rethinking Transfer in Continual Learning: A Replay-Based Realisation17 Jul 2026
- 2026Regularized Reward-Punishment Reinforcement Learning26 Jun 2026
- 2026TerraTransfer: Learning End-to-End Driving Policies Without Expert Demonstrations16 Jun 2026
- 2026Representation Learning Enables Scalable Multitask Deep Reinforcement Learning4 Jun 2026
- 2026AgentJet: A Distributed Swarm Training Framework for Agentic Reinforcement Learning3 Jun 2026
- 2026Task diversity produces systematic transfer but inhibits continual reinforcement learning30 May 2026
38 more
- 2026Decoupled Behavioral Cloning for Scalable Inductive Generalization in RL from Specifications30 May 2026
- 2026FedQHD: Closed-Form Function-Space Federated Reinforcement Learning27 May 2026
- 2026Your Language Model is Its Own Critic: Reinforcement Learning with Value Estimation from Actor's Internal States8 May 2026
- 2026Market-Alignment Risk in Pricing Agents: Trace Diagnostics and Trace-Prior RL under Hidden Competitor State7 May 2026
- 2026Schedule-and-Calibrate: Utility-Guided Multi-Task Reinforcement Learning for Code LLMs7 May 2026
- 2026LANTERN: LLM-Augmented Neurosymbolic Transfer with Experience-Gated Reasoning Networks6 May 2026
- 2026Power Distribution Bridges Sampling, Self-Reward RL, and Self-Distillation6 May 2026
- 2026Contextual Multi-Task Reinforcement Learning for Autonomous Reef Monitoring14 Apr 2026
- 2026Provable Multi-Task Reinforcement Learning: A Representation Learning Framework with Low Rank Rewards4 Apr 2026
- 2026Neural Operators for Multi-Task Control and Adaptation3 Apr 2026
- 2026Reward-Conditioned Reinforcement Learning5 Mar 2026
- 2026Scaling Tasks, Not Samples: Mastering Humanoid Control through Multi-Task Model-Based Reinforcement Learning2 Mar 2026
- 2026Principled Fast and Meta Knowledge Learners for Continual Reinforcement Learning1 Mar 2026
- 2026OmniSapiens: A Foundation Model for Social Behavior Processing via Heterogeneity-Aware Relative Policy Optimization11 Feb 2026
- 2026Embodiment-Aware Generalist Specialist Distillation for Unified Humanoid Whole-Body Control3 Feb 2026
- 2026Probabilistic Performance Guarantees for Multi-Task Reinforcement Learning2 Feb 2026
- 2026CADENT: Gated Hybrid Distillation for Sample-Efficient Transfer in Reinforcement Learning28 Jan 2026
- 2026Structure Detection for Contextual Reinforcement Learning13 Jan 2026
- 2025Conservative Bias in Multi-Teacher Learning: Why Agents Prefer Low-Reward Advisors19 Dec 2025
- 2025Driving Beyond Privilege: Distilling Dense-Reward Knowledge into Sparse-Reward Policies3 Dec 2025
- 2025Soft Conflict-Resolution Decision Transformer for Offline Multi-Task Reinforcement Learning17 Nov 2025
- 2025Scalable Multi-Objective and Meta Reinforcement Learning via Gradient Estimation16 Nov 2025
- 2025Multi-Task Reinforcement Learning with Language-Encoded Gated Policy Networks7 Oct 2025
- 2025GRACE: A Language Model Framework for Explainable Inverse Reinforcement Learning2 Oct 2025
- 2025Simple Optimizers for Convex Aligned Multi-Objective Optimization6 Sep 2025
- 2025Arnold: A multi-task, multi-embodiment muscle transformer policy25 Aug 2025
- 2025Sparsity-Driven Plasticity in Multi-Task Reinforcement Learning9 Aug 2025
- 2025BEAVER: Building Environments with Assessable Variation for Evaluating Multi-Objective Reinforcement Learning10 Jul 2025
- 2025Efficient Multi-Task Reinforcement Learning with Cross-Task Policy Guidance9 Jul 2025
- 2025Integrating Diffusion-based Multi-task Learning with Online Reinforcement Learning for Robust Quadruped Robot Control8 Jul 2025
- 2025TD-MPC-Opt: Distilling Model-Based Multi-Task Reinforcement Learning Agents2 Jul 2025
- 2025Learning Task Belief Similarity with Latent Dynamics for Meta-Reinforcement Learning24 Jun 2025
- 2025CARoL: Context-aware Adaptation for Robot Learning8 Jun 2025
- 2025Delphos: A reinforcement learning framework for assisting discrete choice model specification6 Jun 2025
- 2025Interpretability by Design for Efficient Multi-Objective Reinforcement Learning4 Jun 2025
- 2025Multi-task Learning for Heterogeneous Multi-source Block-Wise Missing Data30 May 2025
- 2025Bigger, Regularized, Categorical: High-Capacity Value Functions are Efficient Multi-Task Learners29 May 2025
- 2025Pessimism Principle Can Be Effective: Towards a Framework for Zero-Shot Transfer Reinforcement Learning24 May 2025
50 newest of 231 All 231 (JSON)
- 2020Keep Doing What Worked: Behavioral Modelling Priors for Offline Reinforcement Learningcited by 213
- 2020Language Conditioned Imitation Learning over Unstructured Datacited by 208
- 2022Pre-Training for Robots: Offline RL Enables Learning New Tasks from a Handful of Trialscited by 78
- 2022Offline Q-Learning on Diverse Multi-Task Data Both Scales And Generalizescited by 72
- 2021Conservative Data Sharing for Multi-Task Offline Reinforcement Learningcited by 64
- 2022Multi-Task Imitation Learning for Linear Dynamical Systemscited by 22
- 2024Language-Conditioned Offline RL for Multi-Robot Navigationcited by 9
- 2024Active Fine-Tuning of Multi-Task Policiescited by 8
- 2022SKILL-IL: Disentangling Skill and Knowledge in Multitask Imitation Learningcited by 4
- 2022Data augmentation for efficient learning from parametric expertscited by 4
- 2024Skills Regularized Task Decomposition for Multi-task Offline Reinforcement Learningcited by 3
- 2025Residual Policy Gradient: A Reward View of KL-regularized Objectivecited by 2
12 most cited of 15 All 15 (JSON)
- 2026Decoupled Behavioral Cloning for Scalable Inductive Generalization in RL from Specifications30 May 2026
- 2025Soft Conflict-Resolution Decision Transformer for Offline Multi-Task Reinforcement Learning17 Nov 2025
- 2025Residual Policy Gradient: A Reward View of KL-regularized Objective14 Mar 2025
- 2024Active Fine-Tuning of Multi-Task Policies7 Oct 2024
- 2024Skills Regularized Task Decomposition for Multi-task Offline Reinforcement Learning28 Aug 2024
- 2024Language-Conditioned Offline RL for Multi-Robot Navigation29 Jul 2024
- 2023Learning Vision-based Robotic Manipulation Tasks Sequentially in Offline Reinforcement Learning Settings31 Jan 2023
- 2022Multi-Task Imitation Learning for Linear Dynamical Systems1 Dec 2022
- 2022Offline Q-Learning on Diverse Multi-Task Data Both Scales And Generalizes28 Nov 2022
- 2022Pre-Training for Robots: Offline RL Enables Learning New Tasks from a Handful of Trials11 Oct 2022
- 2022Data augmentation for efficient learning from parametric experts23 May 2022
- 2022SKILL-IL: Disentangling Skill and Knowledge in Multitask Imitation Learning6 May 2022
12 newest of 15 All 15 (JSON)
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 232
- 2023Confronting Reward Model Overoptimization with Constrained RLHFcited by 96
- 2024The Power of Active Multi-Task Learning in Reinforcement Learning from Human Feedbackcited by 1
- 2026Reward-Conditioned Reinforcement Learningcited by 1
- 2025GRACE: A Language Model Framework for Explainable Inverse Reinforcement Learningcited by 0
- 2025Conservative Bias in Multi-Teacher Learning: Why Agents Prefer Low-Reward Advisorscited by 0
- 2026Reward-Conditioned Reinforcement Learning5 Mar 2026
- 2025Conservative Bias in Multi-Teacher Learning: Why Agents Prefer Low-Reward Advisors19 Dec 2025
- 2025GRACE: A Language Model Framework for Explainable Inverse Reinforcement Learning2 Oct 2025
- 2024The Power of Active Multi-Task Learning in Reinforcement Learning from Human Feedback18 May 2024
- 2023Confronting Reward Model Overoptimization with Constrained RLHF6 Oct 2023
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewards7 Jun 2023
- 2026Power Distribution Bridges Sampling, Self-Reward RL, and Self-Distillationcited by 3
- 2026Your Language Model is Its Own Critic: Reinforcement Learning with Value Estimation from Actor's Internal Statescited by 2
- 2026LANTERN: LLM-Augmented Neurosymbolic Transfer with Experience-Gated Reasoning Networkscited by 0
- 2026Rethinking Transfer in Continual Learning: A Replay-Based Realisationcited by 0
- 2026PAC: Progress-Augmented Advantage Curriculum for Multi-Task Reinforcement Learning of LLMscited by 0
- 2026TelecomGPT-R1: Unified Post-Training for Reasoning Across Heterogeneous Telecom Taskscited by 0
- 2026TelecomGPT-R1: Unified Post-Training for Reasoning Across Heterogeneous Telecom Tasks21 Sep 2026
- 2026PAC: Progress-Augmented Advantage Curriculum for Multi-Task Reinforcement Learning of LLMs31 Aug 2026
- 2026Rethinking Transfer in Continual Learning: A Replay-Based Realisation17 Jul 2026
- 2026Your Language Model is Its Own Critic: Reinforcement Learning with Value Estimation from Actor's Internal States8 May 2026
- 2026LANTERN: LLM-Augmented Neurosymbolic Transfer with Experience-Gated Reasoning Networks6 May 2026
- 2026Power Distribution Bridges Sampling, Self-Reward RL, and Self-Distillation6 May 2026
- 2023HarmonyDream: Task Harmonization Inside World Modelscited by 22
- 2024Scaling Offline Model-Based RL via Jointly-Optimized World-Action Model Pretrainingcited by 13
- 2021Fractional Transfer Learning for Deep Model-Based Reinforcement Learningcited by 6
- 2025Driving Beyond Privilege: Distilling Dense-Reward Knowledge into Sparse-Reward Policiescited by 2
- 2025TD-MPC-Opt: Distilling Model-Based Multi-Task Reinforcement Learning Agentscited by 1
- 2025Driving Beyond Privilege: Distilling Dense-Reward Knowledge into Sparse-Reward Policies3 Dec 2025
- 2025TD-MPC-Opt: Distilling Model-Based Multi-Task Reinforcement Learning Agents2 Jul 2025
- 2024Scaling Offline Model-Based RL via Jointly-Optimized World-Action Model Pretraining1 Oct 2024
- 2023HarmonyDream: Task Harmonization Inside World Models30 Sep 2023
- 2021Fractional Transfer Learning for Deep Model-Based Reinforcement Learning14 Aug 2021
- 2022Retrieval-Augmented Reinforcement Learningcited by 53
- 2022Retrieval-Augmented Reinforcement Learning17 Feb 2022
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
- 2022The Primacy Bias in Deep Reinforcement Learningcited by 242
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scalecited by 226
- 2023RVT: Robotic View Transformer for 3D Object Manipulationcited by 224
- 2021Open-Ended Learning Leads to Generally Capable Agentscited by 177
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021Goal-Conditioned Reinforcement Learning with Imagined Subgoalscited by 129
- 2022Understanding and Preventing Capacity Loss in Reinforcement Learningcited by 124
- 2023Learning Agile Soccer Skills for a Bipedal Robot with Deep Reinforcement Learningcited by 111
- 2021Why Generalization in RL is Difficult: Epistemic POMDPs and Implicit Partial Observabilitycited by 102
- 2023UniDexGrasp++: Improving Dexterous Grasping Policy Learning via Geometry-aware Curriculum and Iterative Generalist-Specialist Learningcited by 101
12 most cited of 191 All 191 (JSON)
- 2026Task Specialization Fine-Tuning for Contextual Reinforcement Learning17 Aug 2026
- 2026Multi-Task Learning for Heterogeneous Prediction from Video Game State with Transfer Learning23 Jul 2026
- 2026Building2Building: A Large Scale Benchmark for Generalizable Real-World Reinforcement Learning17 Jul 2026
- 2026Regularized Reward-Punishment Reinforcement Learning26 Jun 2026
- 2026Representation Learning Enables Scalable Multitask Deep Reinforcement Learning4 Jun 2026
- 2026Task diversity produces systematic transfer but inhibits continual reinforcement learning30 May 2026
- 2026FedQHD: Closed-Form Function-Space Federated Reinforcement Learning27 May 2026
- 2026Market-Alignment Risk in Pricing Agents: Trace Diagnostics and Trace-Prior RL under Hidden Competitor State7 May 2026
- 2026Schedule-and-Calibrate: Utility-Guided Multi-Task Reinforcement Learning for Code LLMs7 May 2026
- 2026Contextual Multi-Task Reinforcement Learning for Autonomous Reef Monitoring14 Apr 2026
- 2026Provable Multi-Task Reinforcement Learning: A Representation Learning Framework with Low Rank Rewards4 Apr 2026
- 2026Neural Operators for Multi-Task Control and Adaptation3 Apr 2026
12 newest of 191 All 191 (JSON)