Multi-Task Reinforcement Learning with Soft Modularization
Not on the map. Why?
Abstract (arXiv)
Multi-task learning is a very challenging problem in reinforcement learning. While training multiple tasks jointly allow the policies to share parameters across different tasks, the optimization problem becomes non-trivial: It remains unclear what parameters in the network should be reused across tasks, and how the gradients from different tasks may interfere with each other. Thus, instead of naively sharing parameters across tasks, we introduce an explicit modularization technique on policy representation to alleviate this optimization issue. Given a base policy network, we design a routing network which estimates different routing strategies to reconfigure the base network for each task. Instead of directly selecting routes for each task, our task-specific policy uses a method called soft modularization to softly combine all the possible routes, which makes it suitable for sequential tasks. We experiment with various robotics manipulation tasks in simulation and show our method improves both sample efficiency and performance over strong baselines by a large margin.
Key facts. Cited by 138 papers in the map (data through 7 Oct 2026). First posted on arXiv 30 Mar 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 17; 2022: 24; 2023: 29; 2024: 27; 2025: 25; 2026: 16.
138 papers.
Sort:
By area:
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scalecited by 226
- 2022Multi-Task Learning as a Bargaining Gamecited by 209
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2023Diffusion Model is an Effective Planner and Data Synthesizer for Multi-Task Reinforcement Learningcited by 114
- 2023GNFactor: Multi-Task Real Robot Learning with Generalizable Neural Feature Fieldscited by 99
- 2021Reset-Free Reinforcement Learning via Multi-Task Learning: Learning Dexterous Manipulation Behaviors without Human Interventioncited by 90
- 2023FAMO: Fast Adaptive Multitask Optimizationcited by 74
- 2023Independent Component Alignment for Multi-Task Learningcited by 69
- 2021Conservative Data Sharing for Multi-Task Offline Reinforcement Learningcited by 64
- 2023Recon: Reducing Conflicting Gradients from the Root for Multi-Task Learningcited by 50
38 more
- 2023Multi-Task Reinforcement Learning with Mixture of Orthogonal Expertscited by 45
- 2022PaCo: Parameter-Compositional Multi-Task Reinforcement Learningcited by 40
- 2022Modular Lifelong Reinforcement Learning via Neural Compositioncited by 39
- 2022The Neural Race Reduction: Dynamics of Abstraction in Gated Networkscited by 37
- 2025Meta-World+: An Improved, Standardized, RL Benchmarkcited by 37
- 2021Lifelong Robotic Reinforcement Learning by Retaining Experiencescited by 34
- 2023Direction-oriented Multi-objective Learning: Simple and Provable Stochastic Algorithmscited by 33
- 2024LLaMAR: Long-Horizon Planning for Multi-Agent Robots in Partially Observable Environmentscited by 32
- 2022Don't Start From Scratch: Leveraging Prior Data to Automate Robotic Reinforcement Learningcited by 28
- 2024LLM-Empowered State Representation for Reinforcement Learningcited by 28
- 2022How to Reuse and Compose Knowledge for a Lifetime of Tasks: A Survey on Continual Learning and Functional Compositioncited by 27
- 2024Learning an Actionable Discrete Diffusion Policy via Large-Scale Actionless Video Pre-Trainingcited by 26
- 2021Variational Automatic Curriculum Learning for Sparse-Reward Cooperative Multi-Agent Problemscited by 25
- 2021Avoiding Catastrophe: Active Dendrites Enable Multi-Task Learning in Dynamic Environmentscited by 24
- 2024DNAct: Diffusion Guided Multi-Task 3D Policy Learningcited by 22
- 2022CompoSuite: A Compositional Reinforcement Learning Benchmarkcited by 21
- 2023Universal Morphology Control via Contextual Modulationcited by 21
- 2021The Functional Correspondence Problemcited by 20
- 2024Jack of All Trades, Master of Some, a Multi-Purpose Transformer Agentcited by 20
- 2023Large Sequence Models for Sequential Decision-Making: A Surveycited by 19
- 2022On the Effectiveness of Fine-tuning Versus Meta-reinforcement Learningcited by 18
- 2022Using Both Demonstrations and Language Instructions to Efficiently Learn Robotic Taskscited by 17
- 2022Robust Policy Learning over Multiple Uncertainty Setscited by 16
- 2023Structure in Deep Reinforcement Learning: A Survey and Open Problemscited by 16
- 2024Contrastive Imitation Learning for Language-guided Multi-Task Robotic Manipulationcited by 15
- 2025Mastering Massive Multi-Task Reinforcement Learning via Mixture-of-Expert Decision Transformercited by 15
- 2022Multi-Agent Policy Transfer via Task Relationship Modelingcited by 14
- 2022Mitigating Gradient Bias in Multi-objective Learning: A Provably Convergent Stochastic Approachcited by 14
- 2024HarmoDT: Harmony Multi-Task Decision Transformer for Offline Reinforcement Learningcited by 14
- 2024Scaling Offline Model-Based RL via Jointly-Optimized World-Action Model Pretrainingcited by 13
- 2025Skill Expansion and Composition in Parameter Spacecited by 13
- 2022A System for Morphology-Task Generalization via Unified Representation and Behavior Distillationcited by 12
- 2024Make-An-Agent: A Generalizable Policy Network Generator with Behavior-Prompted Diffusioncited by 12
- 2024Multi-Task Reinforcement Learning for Quadrotorscited by 12
- 2022Beyond Supervised Continual Learning: a Reviewcited by 11
- 2022Task-Agnostic Continual Reinforcement Learning: Gaining Insights and Overcoming Challengescited by 10
- 2022Transporters with Visual Foresight for Solving Unseen Rearrangement Taskscited by 9
- 2023Improvable Gap Balancing for Multi-Task Learningcited by 9
50 most cited of 138 All 138 (JSON)
- 2026T3S: Improving Multi-Task Reinforcement Learning with Task-Specific Feature Selector and Scheduler31 Aug 2026
- 2026CDCP: Conditional Diffusion Model with Contextual Prompts for Multi-task Offline Safe Reinforcement Learning4 Jul 2026
- 2026Towards Generalization-Oriented Models for Vehicle Routing Problems with Mixture-of-Experts26 May 2026
- 2026Learning in Low-Dimensional Subspaces: Orthogonal Bottlenecks for Reinforcement Learning25 May 2026
- 2026Goal-Conditioned Agents that Learn Everything All at Once22 May 2026
- 2026Spatially Prompted Visual Trajectory Prediction for Egocentric Manipulation19 May 2026
- 2026DyGRO-VLA: Cross-Task Scaling of Vision-Language-Action Models via Dynamic Grouped Residual Optimization17 May 2026
- 2026Distributionally Robust Multi-Task Reinforcement Learning via Adaptive Task Sampling14 May 2026
- 2026TOPPO: Rethinking PPO for Multi-Task Reinforcement Learning with Critic Balancing12 May 2026
- 2026Schedule-and-Calibrate: Utility-Guided Multi-Task Reinforcement Learning for Code LLMs7 May 2026
- 2026Detecting is Easy, Adapting is Hard: Local Expert Growth for Visual Model-Based Reinforcement Learning under Distribution Shift30 Apr 2026
- 2026Scaling Tasks, Not Samples: Mastering Humanoid Control through Multi-Task Model-Based Reinforcement Learning2 Mar 2026
38 more
- 2026Interactive LLM-assisted Curriculum Learning for Multi-Task Evolutionary Policy Search11 Feb 2026
- 2026Mixture-of-World Models: Scaling Multi-Task Reinforcement Learning with Modular Latent Dynamics1 Feb 2026
- 2026Continual Policy Consolidation for Lifelong Robot Learning30 Jan 2026
- 2026Learning Transferable Skills in Action RPGs via Directed Skill Graphs and Selective Adaptation25 Jan 2026
- 2025Iterative Compositional Data Generation for Robot Control11 Dec 2025
- 2025Knowledge Diversion for Efficient Morphology Control and Policy Transfer10 Dec 2025
- 2025Complex Instruction Following with Diverse Style Policies in Football Games25 Nov 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
- 2025One Model for All Tasks: Leveraging Efficient World Models in Multi-Task Planning9 Sep 2025
- 2025Sparsity-Driven Plasticity in Multi-Task Reinforcement Learning9 Aug 2025
- 2025FAST: Similarity-based Knowledge Transfer for Efficient Policy Learning27 Jul 2025
- 2025Goal-Oriented Skill Abstraction for Offline Multi-Task Reinforcement Learning9 Jul 2025
- 2025Efficient Multi-Task Reinforcement Learning with Cross-Task Policy Guidance9 Jul 2025
- 2025Evolutionary Developmental Biology Can Serve as the Conceptual Foundation for a New Design Paradigm in Artificial Intelligence15 Jun 2025
- 2025Gradient Similarity Surgery in Multi-Task Deep Learning6 Jun 2025
- 2025Mastering Massive Multi-Task Reinforcement Learning via Mixture-of-Expert Decision Transformer30 May 2025
- 2025Pessimism Principle Can Be Effective: Towards a Framework for Zero-Shot Transfer Reinforcement Learning24 May 2025
- 2025Meta-World+: An Improved, Standardized, RL Benchmark16 May 2025
- 2025On the Parallels Between Evolutionary Theory and the State of AI13 May 2025
- 2025Next-Future: Sample-Efficient Policy Learning for Robotic-Arm Tasks15 Apr 2025
- 2025Neuron-level Balance between Stability and Plasticity in Deep Reinforcement Learning9 Apr 2025
- 2025Harmonia: Enhancing Data Placement and Migration in Hybrid Storage Systems via Multi-Agent Reinforcement Learning26 Mar 2025
- 2025Multi-Task Reinforcement Learning Enables Parameter Scaling7 Mar 2025
- 2025Learning Policy Committees for Effective Personalization in MDPs with Diverse Tasks26 Feb 2025
- 2025Foundations of a Developmental Design Paradigm for Integrated Continual Learning, Deliberative Behavior, and Comprehensibility19 Feb 2025
- 2025Heterogeneous Resource Allocation with Multi-task Learning for Wireless Networks14 Feb 2025
- 2025Skill Expansion and Composition in Parameter Space9 Feb 2025
- 2024Cross-Modal Mapping: Mitigating the Modality Gap for Few-Shot Image Classification28 Dec 2024
- 2024Multi-Task Reinforcement Learning for Quadrotors17 Dec 2024
- 2024Task-Aware Harmony Multi-Task Decision Transformer for Offline Reinforcement Learning2 Nov 2024
- 2024TapWeight: Reweighting Pretraining Objectives for Task-Adaptive Pretraining13 Oct 2024
- 2024Scaling Offline Model-Based RL via Jointly-Optimized World-Action Model Pretraining1 Oct 2024
- 2024Breaking Neural Network Scaling Laws with Modularity9 Sep 2024
- 2024Non-stationary and Sparsely-correlated Multi-output Gaussian Process with Spike-and-Slab Prior5 Sep 2024
- 2024Skills Regularized Task Decomposition for Multi-task Offline Reinforcement Learning28 Aug 2024
- 2024Centralized Reward Agent for Knowledge Sharing and Transfer in Multi-Task Reinforcement Learning20 Aug 2024
50 newest of 138 All 138 (JSON)
- 2021Conservative Data Sharing for Multi-Task Offline Reinforcement Learningcited by 64
- 2024Contrastive Imitation Learning for Language-guided Multi-Task Robotic Manipulationcited by 15
- 2024HarmoDT: Harmony Multi-Task Decision Transformer for Offline Reinforcement Learningcited by 14
- 2022Transporters with Visual Foresight for Solving Unseen Rearrangement Taskscited by 9
- 2024Skills Regularized Task Decomposition for Multi-task Offline Reinforcement Learningcited by 3
- 2022Modular Adaptive Policy Selection for Multi-Task Imitation Learning through Task Divisioncited by 2
- 2025Soft Conflict-Resolution Decision Transformer for Offline Multi-Task Reinforcement Learningcited by 2
- 2024Task-Aware Harmony Multi-Task Decision Transformer for Offline Reinforcement Learningcited by 1
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Rewardcited by 0
- 2025Soft Conflict-Resolution Decision Transformer for Offline Multi-Task Reinforcement Learning17 Nov 2025
- 2024Task-Aware Harmony Multi-Task Decision Transformer for Offline Reinforcement Learning2 Nov 2024
- 2024Skills Regularized Task Decomposition for Multi-task Offline Reinforcement Learning28 Aug 2024
- 2024Contrastive Imitation Learning for Language-guided Multi-Task Robotic Manipulation14 Jun 2024
- 2024HarmoDT: Harmony Multi-Task Decision Transformer for Offline Reinforcement Learning28 May 2024
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Reward17 Feb 2024
- 2022Modular Adaptive Policy Selection for Multi-Task Imitation Learning through Task Division28 Mar 2022
- 2022Transporters with Visual Foresight for Solving Unseen Rearrangement Tasks22 Feb 2022
- 2021Conservative Data Sharing for Multi-Task Offline Reinforcement Learning16 Sep 2021
- 2024Learning an Actionable Discrete Diffusion Policy via Large-Scale Actionless Video Pre-Trainingcited by 26
- 2024Scaling Offline Model-Based RL via Jointly-Optimized World-Action Model Pretrainingcited by 13
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learningcited by 5
- 2024LIMT: Language-Informed Multi-Task Visual World Modelscited by 4
- 2025One Model for All Tasks: Leveraging Efficient World Models in Multi-Task Planningcited by 3
- 2026Mixture-of-World Models: Scaling Multi-Task Reinforcement Learning with Modular Latent Dynamicscited by 1
- 2026Mixture-of-World Models: Scaling Multi-Task Reinforcement Learning with Modular Latent Dynamics1 Feb 2026
- 2025One Model for All Tasks: Leveraging Efficient World Models in Multi-Task Planning9 Sep 2025
- 2024Scaling Offline Model-Based RL via Jointly-Optimized World-Action Model Pretraining1 Oct 2024
- 2024LIMT: Language-Informed Multi-Task Visual World Models18 Jul 2024
- 2024Learning an Actionable Discrete Diffusion Policy via Large-Scale Actionless Video Pre-Training22 Feb 2024
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learning9 Mar 2023
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scalecited by 226
- 2022Multi-Task Learning as a Bargaining Gamecited by 209
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2023GNFactor: Multi-Task Real Robot Learning with Generalizable Neural Feature Fieldscited by 99
- 2021Reset-Free Reinforcement Learning via Multi-Task Learning: Learning Dexterous Manipulation Behaviors without Human Interventioncited by 90
- 2023FAMO: Fast Adaptive Multitask Optimizationcited by 74
- 2023Independent Component Alignment for Multi-Task Learningcited by 69
- 2023Recon: Reducing Conflicting Gradients from the Root for Multi-Task Learningcited by 50
- 2023Multi-Task Reinforcement Learning with Mixture of Orthogonal Expertscited by 45
- 2022PaCo: Parameter-Compositional Multi-Task Reinforcement Learningcited by 40
12 most cited of 117 All 117 (JSON)
- 2026T3S: Improving Multi-Task Reinforcement Learning with Task-Specific Feature Selector and Scheduler31 Aug 2026
- 2026Learning in Low-Dimensional Subspaces: Orthogonal Bottlenecks for Reinforcement Learning25 May 2026
- 2026Goal-Conditioned Agents that Learn Everything All at Once22 May 2026
- 2026Spatially Prompted Visual Trajectory Prediction for Egocentric Manipulation19 May 2026
- 2026Distributionally Robust Multi-Task Reinforcement Learning via Adaptive Task Sampling14 May 2026
- 2026TOPPO: Rethinking PPO for Multi-Task Reinforcement Learning with Critic Balancing12 May 2026
- 2026Schedule-and-Calibrate: Utility-Guided Multi-Task Reinforcement Learning for Code LLMs7 May 2026
- 2026Detecting is Easy, Adapting is Hard: Local Expert Growth for Visual Model-Based Reinforcement Learning under Distribution Shift30 Apr 2026
- 2026Scaling Tasks, Not Samples: Mastering Humanoid Control through Multi-Task Model-Based Reinforcement Learning2 Mar 2026
- 2026Interactive LLM-assisted Curriculum Learning for Multi-Task Evolutionary Policy Search11 Feb 2026
- 2026Continual Policy Consolidation for Lifelong Robot Learning30 Jan 2026
- 2026Learning Transferable Skills in Action RPGs via Directed Skill Graphs and Selective Adaptation25 Jan 2026
12 newest of 117 All 117 (JSON)