Modular Multitask Reinforcement Learning with Policy Sketches
Not on the map. Why?
Abstract (arXiv)
We describe a framework for multitask deep reinforcement learning guided by policy sketches. Sketches annotate tasks with sequences of named subtasks, providing information about high-level structural relationships among tasks but not how to implement them---specifically not providing the detailed guidance used by much previous work on learning policy abstractions for RL (e.g. intermediate rewards, subtask completion signals, or intrinsic motivations). To learn from sketches, we present a model that associates every subtask with a modular subpolicy, and jointly maximizes reward over full task-specific policies by tying parameters across shared subpolicies. Optimization is accomplished via a decoupled actor--critic training objective that facilitates learning common behaviors from multiple dissimilar reward functions. We evaluate the effectiveness of our approach in three environments featuring both discrete and continuous control, and with sparse rewards that can be obtained only after completing a number of high-level subgoals. Experiments show that using our approach to learn policies guided by sketches gives better performance than existing techniques for learning task-specific or shared policies, while naturally inducing a library of interpretable primitive behaviors that can be recombined to rapidly adapt to new tasks.
Key facts. Cited by 200 papers in the map (data through 7 Oct 2026). First posted on arXiv 6 Nov 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 1; 2020: 0; 2021: 42; 2022: 45; 2023: 32; 2024: 33; 2025: 27; 2026: 19.
200 papers.
Sort:
By area:
- 2022Do As I Can, Not As I Say: Grounding Language in Robotic Affordancescited by 2,302
- 2023VoxPoser: Composable 3D Value Maps for Robotic Manipulation with Language Modelscited by 733
- 2022MineDojo: Building Open-Ended Embodied Agents with Internet-Scale Knowledgecited by 469
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2023Training Language Models with Language Feedback at Scalecited by 125
- 2021Learning Language-Conditioned Robot Behavior from Offline Data and Crowd-Sourced Annotationcited by 117
- 2024Survey on Large Language Model-Enhanced Reinforcement Learning: Concept, Taxonomy, and Methodscited by 114
- 2021Skill Induction and Planning with Latent Languagecited by 99
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysiscited by 99
- 2023Modular Deep Learningcited by 82
- 2021LTL2Action: Generalizing LTL Instructions for Multi-Task RLcited by 72
38 more
- 2024HiRT: Enhancing Robotic Control with Hierarchical Robot Transformerscited by 72
- 2021A Survey on Interpretable Reinforcement Learningcited by 70
- 2023Learning to Model the World with Languagecited by 66
- 2021Compositional Reinforcement Learning from Logical Specificationscited by 65
- 2021Learning to Synthesize Programs as Interpretable and Generalizable Policiescited by 59
- 2022TASKOGRAPHY: Evaluating robot task planning over large 3D scene graphscited by 59
- 2022Training Language Models with Language Feedbackcited by 48
- 2021Grounding Language to Entities and Dynamics for Generalization in Reinforcement Learningcited by 45
- 2021Adversarial Skill Chaining for Long-Horizon Robot Manipulation via Terminal State Regularizationcited by 44
- 2021ELLA: Exploration through Learned Language Abstractioncited by 42
- 2024RT-Sketch: Goal-Conditioned Imitation Learning from Hand-Drawn Sketchescited by 41
- 2022PaCo: Parameter-Compositional Multi-Task Reinforcement Learningcited by 40
- 2022The Neural Race Reduction: Dynamics of Abstraction in Gated Networkscited by 37
- 2023Learning adaptive planning representations with natural language guidancecited by 36
- 2023Plan, Eliminate, and Track -- Language Models are Good Teachers for Embodied Agentscited by 32
- 2022Disentangling Transfer in Continual Reinforcement Learningcited by 31
- 2024Unsupervised Zero-Shot Reinforcement Learning via Functional Reward Encodingscited by 31
- 2023Learning to Modulate pre-trained Models in RLcited by 28
- 2021Block Contextual MDPs for Continual Learningcited by 25
- 2021The Logical Options Frameworkcited by 24
- 2021Compositional Transformers for Scene Generationcited by 24
- 2021Avoiding Catastrophe: Active Dendrites Enable Multi-Task Learning in Dynamic Environmentscited by 24
- 2023SPRINT: Scalable Policy Pre-Training via Language Instruction Relabelingcited by 24
- 2024MaestroMotif: Skill Design from Artificial Intelligence Feedbackcited by 24
- 2022EAGER: Asking and Answering Questions for Automatic Reward Shaping in Language-guided RLcited by 23
- 2022Leveraging Approximate Symbolic Models for Reinforcement Learning via Skill Diversitycited by 22
- 2022Learning Options via Compressioncited by 22
- 2024DeepLTL: Learning to Efficiently Satisfy Complex LTL Specifications for Multi-Task RLcited by 20
- 2021Solving Compositional Reinforcement Learning Problems via Task Reductioncited by 19
- 2024Compositional Automata Embeddings for Goal-Conditioned Reinforcement Learningcited by 18
- 2021Computational Benefits of Intermediate Rewards for Goal-Reaching Policy Learningcited by 17
- 2021In a Nutshell, the Human Asked for This: Latent Goals for Following Temporal Specificationscited by 17
- 2022Using Both Demonstrations and Language Instructions to Efficiently Learn Robotic Taskscited by 17
- 2023Sample Efficient Model-free Reinforcement Learning from LTL Specifications with Optimality Guaranteescited by 17
- 2022Understanding the Complexity Gains of Single-Task RL with a Curriculumcited by 16
- 2021Learning Task Decomposition with Ordered Memory Policy Networkcited by 15
- 2022Transformers are Adaptable Task Plannerscited by 15
- 2023Multi-task Hierarchical Adversarial Inverse Reinforcement Learningcited by 15
50 most cited of 200 All 200 (JSON)
- 2026Reward as Observation: Learning Reward-Based Policies for Rapid Adaptation30 Sep 2026
- 2026Jaxolotl: A Unified High-Performance Benchmark Suite for LTL-Based Multi-Task RL29 Sep 2026
- 2026State Trace Rationale As Auxiliary Task in Reinforcement Learning29 Sep 2026
- 2026Curriculum-Based Adversarial Heterogeneous Agent Reinforcement Learning for Autonomous Quad-Copter Landing in Maritime Settings11 Sep 2026
- 2026Policy-Invariant Reward Shaping from LLM Feedback: A Framework for Hybrid RL Agents18 Aug 2026
- 2026tinyDSM: A Framework for Skill Modeling and Development for Resource-Constrained Millirobots18 Aug 2026
- 2026Training Language Models to Cooperate with Inference-Time Controllers26 Jul 2026
- 2026MEMENTO: Memory-Guided Memetic Code-as-Policy Evolution24 Jul 2026
- 2026Anything2Skill: Compiling External Knowledge into Reusable Skills for Agents8 Jun 2026
- 2026A Goal-Set Characterization of Task Composition in the Boolean Task Algebra2 Jun 2026
- 2026Task diversity produces systematic transfer but inhibits continual reinforcement learning30 May 2026
- 2026Structure-Induced Information for Rerooting Levin Tree Search28 May 2026
38 more
- 2026Goal-Conditioned Agents that Learn Everything All at Once22 May 2026
- 2026LANTERN: LLM-Augmented Neurosymbolic Transfer with Experience-Gated Reasoning Networks6 May 2026
- 2026Active Reward Machine Inference From Raw State Trajectories8 Apr 2026
- 2026Unified Policy Value Decomposition for Rapid Adaptation18 Mar 2026
- 2026NS-VLA: Towards Neuro-Symbolic Vision-Language-Action Models10 Mar 2026
- 2026Phase-Aware Mixture of Experts for Agentic Reinforcement Learning19 Feb 2026
- 2026Grounding LTL Tasks in Sub-Symbolic RL Environments for Zero-Shot Generalization10 Feb 2026
- 2025Flexible Multitask Learning with Factorized Diffusion Policy26 Dec 2025
- 2025Iterative Compositional Data Generation for Robot Control11 Dec 2025
- 2025Learning When to Switch: Adaptive Policy Selection via Reinforcement Learning6 Dec 2025
- 2025Realizable Abstractions: Near-Optimal Hierarchical Reinforcement Learning4 Dec 2025
- 2025Beyond Fixed Tasks: Unsupervised Environment Design for Task-Level Pairs16 Nov 2025
- 2025DAIL: Beyond Task Ambiguity for Language-Conditioned Reinforcement Learning22 Oct 2025
- 2025Heterogeneous Adversarial Play in Interactive Environments21 Oct 2025
- 2025RLAF: Reinforcement Learning from Automaton Feedback17 Oct 2025
- 2025Reinforcement Learning with Stochastic Reward Machines16 Oct 2025
- 2025Multi-Task Reinforcement Learning with Language-Encoded Gated Policy Networks7 Oct 2025
- 2025Data-Efficient Multitask DAgger29 Sep 2025
- 2025Amico: An Event-Driven Modular Framework for Persistent and Embedded Autonomy19 Jul 2025
- 2025Ground-Compose-Reinforce: Grounding Language in Agentic Behaviours using Limited Data14 Jul 2025
- 2025CyberRAG: An Agentic RAG cyber attack classification and reporting tool3 Jul 2025
- 2025Evolutionary Developmental Biology Can Serve as the Conceptual Foundation for a New Design Paradigm in Artificial Intelligence15 Jun 2025
- 2025Subgoal-Guided Policy Heuristic Search with Learned Subgoals8 Jun 2025
- 2025The challenge of hidden gifts in multi-agent reinforcement learning26 May 2025
- 2025Modular Diffusion Policy Training: Decoupling and Recombining Guidance and Diffusion for Offline RL19 May 2025
- 2025On the Parallels Between Evolutionary Theory and the State of AI13 May 2025
- 2025Humanoid Agent via Embodied Chain-of-Action Reasoning with Multimodal Foundation Models for Zero-Shot Loco-Manipulation13 Apr 2025
- 2025HypRL: Reinforcement Learning of Control Policies for Hyperproperties7 Apr 2025
- 2025Causally Aligned Curriculum Learning21 Mar 2025
- 2025Towards Automated Semantic Interpretability in Reinforcement Learning via Vision-Language Models20 Mar 2025
- 2025A Generalist Hanabi Agent17 Mar 2025
- 2025Shared Autonomy for Proximal Teaching27 Feb 2025
- 2025Automating Curriculum Learning for Reinforcement Learning using a Skill-Based Bayesian Network21 Feb 2025
- 2025Foundations of a Developmental Design Paradigm for Integrated Continual Learning, Deliberative Behavior, and Comprehensibility19 Feb 2025
- 2024A Research Agenda for Usability and Generalisation in Reinforcement Learning22 Dec 2024
- 2024MaestroMotif: Skill Design from Artificial Intelligence Feedback11 Dec 2024
- 2024CAREL: Instruction-guided reinforcement learning with cross-modal auxiliary objectives29 Nov 2024
- 2024Efficient, Low-Regret, Online Reinforcement Learning for Linear MDPs16 Nov 2024
50 newest of 200 All 200 (JSON)
- 2023Plan, Eliminate, and Track -- Language Models are Good Teachers for Embodied Agentscited by 32
- 2023LLM Augmented Hierarchical Agentscited by 9
- 2025Ground-Compose-Reinforce: Grounding Language in Agentic Behaviours using Limited Datacited by 5
- 2026Phase-Aware Mixture of Experts for Agentic Reinforcement Learningcited by 4
- 2025Humanoid Agent via Embodied Chain-of-Action Reasoning with Multimodal Foundation Models for Zero-Shot Loco-Manipulationcited by 3
- 2024Words as Beacons: Guiding RL Agents with High-Level Language Promptscited by 1
- 2026Policy-Invariant Reward Shaping from LLM Feedback: A Framework for Hybrid RL Agentscited by 0
- 2026Policy-Invariant Reward Shaping from LLM Feedback: A Framework for Hybrid RL Agents18 Aug 2026
- 2026Phase-Aware Mixture of Experts for Agentic Reinforcement Learning19 Feb 2026
- 2025Ground-Compose-Reinforce: Grounding Language in Agentic Behaviours using Limited Data14 Jul 2025
- 2025Humanoid Agent via Embodied Chain-of-Action Reasoning with Multimodal Foundation Models for Zero-Shot Loco-Manipulation13 Apr 2025
- 2024Words as Beacons: Guiding RL Agents with High-Level Language Prompts11 Oct 2024
- 2023LLM Augmented Hierarchical Agents9 Nov 2023
- 2023Plan, Eliminate, and Track -- Language Models are Good Teachers for Embodied Agents3 May 2023
- 2024RT-Sketch: Goal-Conditioned Imitation Learning from Hand-Drawn Sketchescited by 41
- 2023Multi-task Hierarchical Adversarial Inverse Reinforcement Learningcited by 15
- 2018CompILE: Compositional Imitation Learning and Executioncited by 7
- 2019Integration of Imitation Learning using GAIL and Reinforcement Learning using Task-achievement Rewards via Probabilistic Graphical Modelcited by 7
- 2022SKILL-IL: Disentangling Skill and Knowledge in Multitask Imitation Learningcited by 4
- 2024Skills Regularized Task Decomposition for Multi-task Offline Reinforcement Learningcited by 3
- 2024Skills Regularized Task Decomposition for Multi-task Offline Reinforcement Learning28 Aug 2024
- 2024RT-Sketch: Goal-Conditioned Imitation Learning from Hand-Drawn Sketches5 Mar 2024
- 2023Multi-task Hierarchical Adversarial Inverse Reinforcement Learning22 May 2023
- 2022SKILL-IL: Disentangling Skill and Knowledge in Multitask Imitation Learning6 May 2022
- 2019Integration of Imitation Learning using GAIL and Reinforcement Learning using Task-achievement Rewards via Probabilistic Graphical Model3 Jul 2019
- 2018CompILE: Compositional Imitation Learning and Execution4 Dec 2018
- 2023Training Language Models with Language Feedback at Scalecited by 125
- 2022Training Language Models with Language Feedbackcited by 48
- 2025RLAF: Reinforcement Learning from Automaton Feedbackcited by 0
- 2025RLAF: Reinforcement Learning from Automaton Feedback17 Oct 2025
- 2023Training Language Models with Language Feedback at Scale28 Mar 2023
- 2022Training Language Models with Language Feedback29 Apr 2022
- 2026NS-VLA: Towards Neuro-Symbolic Vision-Language-Action Models10 Mar 2026
- 2022Do As I Can, Not As I Say: Grounding Language in Robotic Affordancescited by 2,302
- 2023VoxPoser: Composable 3D Value Maps for Robotic Manipulation with Language Modelscited by 733
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2021Multi-Task Reinforcement Learning with Context-based Representationscited by 169
- 2021Learning Language-Conditioned Robot Behavior from Offline Data and Crowd-Sourced Annotationcited by 117
- 2024Survey on Large Language Model-Enhanced Reinforcement Learning: Concept, Taxonomy, and Methodscited by 114
- 2021Skill Induction and Planning with Latent Languagecited by 99
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysiscited by 99
- 2023Modular Deep Learningcited by 82
- 2021LTL2Action: Generalizing LTL Instructions for Multi-Task RLcited by 72
- 2021A Survey on Interpretable Reinforcement Learningcited by 70
- 2023Learning to Model the World with Languagecited by 66
12 most cited of 171 All 171 (JSON)
- 2026Reward as Observation: Learning Reward-Based Policies for Rapid Adaptation30 Sep 2026
- 2026Jaxolotl: A Unified High-Performance Benchmark Suite for LTL-Based Multi-Task RL29 Sep 2026
- 2026State Trace Rationale As Auxiliary Task in Reinforcement Learning29 Sep 2026
- 2026Curriculum-Based Adversarial Heterogeneous Agent Reinforcement Learning for Autonomous Quad-Copter Landing in Maritime Settings11 Sep 2026
- 2026tinyDSM: A Framework for Skill Modeling and Development for Resource-Constrained Millirobots18 Aug 2026
- 2026Training Language Models to Cooperate with Inference-Time Controllers26 Jul 2026
- 2026MEMENTO: Memory-Guided Memetic Code-as-Policy Evolution24 Jul 2026
- 2026A Goal-Set Characterization of Task Composition in the Boolean Task Algebra2 Jun 2026
- 2026Task diversity produces systematic transfer but inhibits continual reinforcement learning30 May 2026
- 2026Structure-Induced Information for Rerooting Levin Tree Search28 May 2026
- 2026Goal-Conditioned Agents that Learn Everything All at Once22 May 2026
- 2026Active Reward Machine Inference From Raw State Trajectories8 Apr 2026
12 newest of 171 All 171 (JSON)