Multi-task Maximum Entropy Inverse Reinforcement Learning
On the map: Offline RL and imitation learning from data → Inverse reinforcement learning
Abstract (arXiv)
Multi-task Inverse Reinforcement Learning (IRL) is the problem of inferring multiple reward functions from expert demonstrations. Prior work, built on Bayesian IRL, is unable to scale to complex environments due to computational constraints. This paper contributes a formulation of multi-task IRL in the more computationally efficient Maximum Causal Entropy (MCE) IRL framework. Experiments show our approach can perform one-shot imitation learning in a gridworld environment that single-task IRL algorithms need hundreds of demonstrations to solve. We outline preliminary work using meta-learning to extend our method to the function approximator setting of modern MCE IRL algorithms. Evaluating on multi-task variants of common simulated robotics benchmarks, we discover serious limitations of these IRL algorithms, and conclude with suggestions for further work.
Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). Placed in Inverse reinforcement learning, one of 13 directions in Offline RL and imitation learning from data. Builds on 10 of its 25 references that are in the map. First posted on arXiv 22 May 2018.
Builds on
10 of 25 references.
References · 10
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2018On First-Order Meta-Learning Algorithmscited by 1,003
- 2016Guided Cost Learning: Deep Inverse Optimal Control via Policy Optimizationcited by 400
- 2017One-Shot Imitation Learningcited by 232
- 2016Unrolled Generative Adversarial Networkscited by 208
- 2017VEEGAN: Reducing Mode Collapse in GANs using Implicit Variational Learningcited by 199
- 2017Robust Imitation of Diverse Behaviorscited by 85
- 2017Multi-Modal Imitation Learning from Unstructured Demonstrations using Generative Adversarial Netscited by 64
- 2017Repeated Inverse Reinforcement Learningcited by 29
Built on by
Citing papers per year: 2019: 1; 2020: 0; 2021: 6; 2022: 2; 2023: 6; 2024: 3; 2025: 0; 2026: 2.
20 papers.
Sort:
By area:
- 2021Generalized Decision Transformer for Offline Hindsight Information Matchingcited by 105
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2022Robust Imitation Learning against Variations in Environment Dynamicscited by 26
- 2021MORAL: Aligning AI with Human Norms through Multi-Objective Reinforced Active Learningcited by 24
- 2023Learning Shared Safety Constraints from Multi-task Demonstrationscited by 15
- 2023SIRL: Similarity-based Implicit Representation Learningcited by 13
- 2023Aligning Robot and Human Representationscited by 11
- 2023Multi-intention Inverse Q-learning for Interpretable Behavior Representationcited by 9
- 2021Combining Reward Information from Multiple Sourcescited by 7
- 2023BC-IRL: Learning Generalizable Reward Functions from Demonstrationscited by 7
- 2021Meta-Adversarial Inverse Reinforcement Learning for Decision-making Taskscited by 6
- 2022Basis for Intentions: Efficient Inverse Reinforcement Learning using Past Experiencecited by 4
8 more
- 2024Few-Shot Task Learning through Inverse Generative Modelingcited by 4
- 2021LiMIIRL: Lightweight Multiple-Intent Inverse Reinforcement Learningcited by 2
- 2026Can Optimal Transport Improve Federated Inverse Reinforcement Learning?cited by 1
- 2021Quickest change detection with unknown parameters: Constant complexity and near optimalitycited by 0
- 2023Kernel Density Bayesian Inverse Reinforcement Learningcited by 0
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Rewardcited by 0
- 2024Inverse Reinforcement Learning with Multiple Planning Horizonscited by 0
- 2026Generalize and Guide: Decomposing Rewards for Few-Shot Inverse Reinforcement Learningcited by 0
- 2026Generalize and Guide: Decomposing Rewards for Few-Shot Inverse Reinforcement Learning20 Jul 2026
- 2026Can Optimal Transport Improve Federated Inverse Reinforcement Learning?1 Jan 2026
- 2024Few-Shot Task Learning through Inverse Generative Modeling7 Nov 2024
- 2024Inverse Reinforcement Learning with Multiple Planning Horizons26 Sep 2024
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Reward17 Feb 2024
- 2023Multi-intention Inverse Q-learning for Interpretable Behavior Representation23 Nov 2023
- 2023Learning Shared Safety Constraints from Multi-task Demonstrations1 Sep 2023
- 2023BC-IRL: Learning Generalizable Reward Functions from Demonstrations28 Mar 2023
- 2023Kernel Density Bayesian Inverse Reinforcement Learning13 Mar 2023
- 2023Aligning Robot and Human Representations3 Feb 2023
- 2023SIRL: Similarity-based Implicit Representation Learning2 Jan 2023
- 2022Basis for Intentions: Efficient Inverse Reinforcement Learning using Past Experience9 Aug 2022
8 more
- 2022Robust Imitation Learning against Variations in Environment Dynamics19 Jun 2022
- 2021MORAL: Aligning AI with Human Norms through Multi-Objective Reinforced Active Learning30 Dec 2021
- 2021Generalized Decision Transformer for Offline Hindsight Information Matching19 Nov 2021
- 2021Quickest change detection with unknown parameters: Constant complexity and near optimality9 Jun 2021
- 2021LiMIIRL: Lightweight Multiple-Intent Inverse Reinforcement Learning3 Jun 2021
- 2021Meta-Adversarial Inverse Reinforcement Learning for Decision-making Tasks23 Mar 2021
- 2021Combining Reward Information from Multiple Sources22 Mar 2021
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variables20 Sep 2019
- 2021Generalized Decision Transformer for Offline Hindsight Information Matchingcited by 105
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2022Robust Imitation Learning against Variations in Environment Dynamicscited by 26
- 2023Multi-intention Inverse Q-learning for Interpretable Behavior Representationcited by 9
- 2023BC-IRL: Learning Generalizable Reward Functions from Demonstrationscited by 7
- 2021Meta-Adversarial Inverse Reinforcement Learning for Decision-making Taskscited by 6
- 2022Basis for Intentions: Efficient Inverse Reinforcement Learning using Past Experiencecited by 4
- 2021LiMIIRL: Lightweight Multiple-Intent Inverse Reinforcement Learningcited by 2
- 2026Can Optimal Transport Improve Federated Inverse Reinforcement Learning?cited by 1
- 2023Kernel Density Bayesian Inverse Reinforcement Learningcited by 0
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Rewardcited by 0
- 2024Inverse Reinforcement Learning with Multiple Planning Horizonscited by 0
12 most cited of 13 All 13 (JSON)
- 2026Generalize and Guide: Decomposing Rewards for Few-Shot Inverse Reinforcement Learning20 Jul 2026
- 2026Can Optimal Transport Improve Federated Inverse Reinforcement Learning?1 Jan 2026
- 2024Inverse Reinforcement Learning with Multiple Planning Horizons26 Sep 2024
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Reward17 Feb 2024
- 2023Multi-intention Inverse Q-learning for Interpretable Behavior Representation23 Nov 2023
- 2023BC-IRL: Learning Generalizable Reward Functions from Demonstrations28 Mar 2023
- 2023Kernel Density Bayesian Inverse Reinforcement Learning13 Mar 2023
- 2022Basis for Intentions: Efficient Inverse Reinforcement Learning using Past Experience9 Aug 2022
- 2022Robust Imitation Learning against Variations in Environment Dynamics19 Jun 2022
- 2021Generalized Decision Transformer for Offline Hindsight Information Matching19 Nov 2021
- 2021LiMIIRL: Lightweight Multiple-Intent Inverse Reinforcement Learning3 Jun 2021
- 2021Meta-Adversarial Inverse Reinforcement Learning for Decision-making Tasks23 Mar 2021
12 newest of 13 All 13 (JSON)
- 2023Learning Shared Safety Constraints from Multi-task Demonstrationscited by 15
- 2023SIRL: Similarity-based Implicit Representation Learningcited by 13
- 2023Aligning Robot and Human Representationscited by 11
- 2021Combining Reward Information from Multiple Sourcescited by 7
- 2024Few-Shot Task Learning through Inverse Generative Modelingcited by 4
- 2021Quickest change detection with unknown parameters: Constant complexity and near optimalitycited by 0
- 2024Few-Shot Task Learning through Inverse Generative Modeling7 Nov 2024
- 2023Learning Shared Safety Constraints from Multi-task Demonstrations1 Sep 2023
- 2023Aligning Robot and Human Representations3 Feb 2023
- 2023SIRL: Similarity-based Implicit Representation Learning2 Jan 2023
- 2021Quickest change detection with unknown parameters: Constant complexity and near optimality9 Jun 2021
- 2021Combining Reward Information from Multiple Sources22 Mar 2021
Similar, not cited
Close in wording, no citation either way
- 2022A Primer on Maximum Causal Entropy Inverse Reinforcement Learning
- 2019Generalized Maximum Causal Entropy for Inverse Reinforcement Learning
- 2022Weighted Maximum Entropy Inverse Reinforcement Learning
- 2020Revisiting Maximum Entropy Inverse Reinforcement Learning: New Perspectives and Algorithms
- 2022Task-Guided IRL in POMDPs that Scales
- 2021Task-Guided Inverse Reinforcement Learning Under Partial Information
- 2019Maximum Likelihood Constraint Inference for Inverse Reinforcement Learning
- 2020Efficient Exploration of Reward Functions in Inverse Reinforcement Learning via Bayesian Optimization