imitation: Clean Imitation Learning Implementations
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
imitation provides open-source implementations of imitation and reward learning algorithms in PyTorch. We include three inverse reinforcement learning (IRL) algorithms, three imitation learning algorithms and a preference comparison algorithm. The implementations have been benchmarked against previous results, and automated tests cover 98% of the code. Moreover, the algorithms are implemented in a modular fashion, making it simple to develop novel algorithms in the framework. Our source code, including documentation and examples, is available at https://github.com/HumanCompatibleAI/imitation
Key facts. Cited by 34 papers in the map (data through 7 Oct 2026). Placed in Imitation learning methods, one of 13 directions in Offline RL and imitation learning from data. Builds on 6 of its 15 references that are in the map. First posted on arXiv 22 Nov 2022.
Builds on
6 of 15 references.
Closest ancestors · 2
Same direction, newest first
- 2021What Matters for Adversarial Imitation Learning?1 Jun
- 2018Discriminator-Actor-Critic: Addressing Sample Inefficiency and Reward Bias in Adversarial Imitation Learning9 Sep
Other references · 4
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2019Dota 2 with Large Scale Deep Reinforcement Learningcited by 1,084
- 2021APReL: A Library for Active Preference-based Reward Learning Algorithmscited by 14
Built on by
Citing papers per year: 2022: 1; 2023: 7; 2024: 10; 2025: 13; 2026: 3.
34 papers.
Sort:
By area:
- 2023A Survey of Reinforcement Learning from Human Feedbackcited by 259
- 2024Human-compatible driving partners through data-regularized self-play reinforcement learningcited by 17
- 2023Towards Solving Fuzzy Tasks with Human Feedback: A Retrospective of the MineRL BASALT 2022 Competitioncited by 15
- 2023RLHF-Blender: A Configurable Interactive Interface for Learning from Diverse Human Feedbackcited by 15
- 2024Diffusion Imitation from Observationcited by 12
- 2025CurricuVLM: Towards Safe Autonomous Driving via Personalized Safety-Critical Curriculum Learning with Vision-Language Modelscited by 11
- 2023Residual Q-Learning: Offline and Online Policy Customization without Valuecited by 9
- 2025X-IL: Exploring the Design Space of Imitation Learning Policiescited by 9
- 2025Reward Learning from Multiple Feedback Typescited by 9
- 2023BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Taskscited by 8
- 2024RICE: Breaking Through the Training Bottlenecks of Reinforcement Learning with Explanationcited by 7
- 2025SPACeR: Self-Play Anchoring with Centralized Reference Modelscited by 7
22 more
- 2025WOFOSTGym: A Crop Simulator for Learning Annual and Perennial Crop Management Strategiescited by 4
- 2025Interactive Groupwise Comparison for Reinforcement Learning from Human Feedbackcited by 4
- 2022Safe Imitation Learning of Nonlinear Model Predictive Control for Flexible Robotscited by 3
- 2024C-GAIL: Stabilizing Generative Adversarial Imitation Learning with Control Theorycited by 3
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shapingcited by 3
- 2025Leveraging Genetic Algorithms for Efficient Demonstration Generation in Real-World Reinforcement Learning Environmentscited by 3
- 2024"Give Me an Example Like This": Episodic Active Reinforcement Learning from Demonstrationscited by 2
- 2024Synergistic Reinforcement and Imitation Learning for Vision-driven Autonomous Flight of UAV Along Rivercited by 1
- 2024Preserving the Privacy of Reward Functions in MDPs through Deceptioncited by 1
- 2024A Graph-based Adversarial Imitation Learning Framework for Reliable & Realtime Fleet Scheduling in Urban Air Mobilitycited by 1
- 2025On Discovering Algorithms for Adversarial Imitation Learningcited by 1
- 2023Synthetically Generating Human-like Data for Sequential Decision Making Tasks via Reward-Shaped Imitation Learningcited by 0
- 2023Dynamic value alignment through preference aggregation of multiple objectivescited by 0
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Rewardcited by 0
- 2025Supervised Reward Inferencecited by 0
- 2025Whenever, Wherever: Towards Orchestrating Crowd Simulations with Spatio-Temporal Spawn Dynamicscited by 0
- 2025Learning Dolly-In Filming From Demonstration Using a Ground-Based Robotcited by 0
- 2025Buzz, Choose, Forget: A Meta-Bandit Framework for Bee-Like Decision Makingcited by 0
- 2025An Introduction to Deep Reinforcement and Imitation Learningcited by 0
- 2026Generative adversarial imitation learning for robot swarms: Learning from human demonstrations and trained policiescited by 0
- 2026Latent Wasserstein Adversarial Imitation Learningcited by 0
- 2026Hierarchical Apprenticeship Learning from Imperfect Demonstrations with Evolving Rewardscited by 0
- 2026Hierarchical Apprenticeship Learning from Imperfect Demonstrations with Evolving Rewards31 Mar 2026
- 2026Latent Wasserstein Adversarial Imitation Learning5 Mar 2026
- 2026Generative adversarial imitation learning for robot swarms: Learning from human demonstrations and trained policies3 Mar 2026
- 2025An Introduction to Deep Reinforcement and Imitation Learning8 Dec 2025
- 2025SPACeR: Self-Play Anchoring with Centralized Reference Models20 Oct 2025
- 2025Buzz, Choose, Forget: A Meta-Bandit Framework for Bee-Like Decision Making18 Oct 2025
- 2025On Discovering Algorithms for Adversarial Imitation Learning1 Oct 2025
- 2025Learning Dolly-In Filming From Demonstration Using a Ground-Based Robot30 Aug 2025
- 2025Interactive Groupwise Comparison for Reinforcement Learning from Human Feedback6 Jul 2025
- 2025Leveraging Genetic Algorithms for Efficient Demonstration Generation in Real-World Reinforcement Learning Environments1 Jul 2025
- 2025Whenever, Wherever: Towards Orchestrating Crowd Simulations with Spatio-Temporal Spawn Dynamics20 Mar 2025
- 2025Reward Learning from Multiple Feedback Types28 Feb 2025
22 more
- 2025WOFOSTGym: A Crop Simulator for Learning Annual and Perennial Crop Management Strategies26 Feb 2025
- 2025Supervised Reward Inference25 Feb 2025
- 2025CurricuVLM: Towards Safe Autonomous Driving via Personalized Safety-Critical Curriculum Learning with Vision-Language Models21 Feb 2025
- 2025X-IL: Exploring the Design Space of Imitation Learning Policies17 Feb 2025
- 2024Diffusion Imitation from Observation7 Oct 2024
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shaping4 Oct 2024
- 2024A Graph-based Adversarial Imitation Learning Framework for Reliable & Realtime Fleet Scheduling in Urban Air Mobility16 Jul 2024
- 2024Preserving the Privacy of Reward Functions in MDPs through Deception13 Jul 2024
- 2024"Give Me an Example Like This": Episodic Active Reinforcement Learning from Demonstrations5 Jun 2024
- 2024RICE: Breaking Through the Training Bottlenecks of Reinforcement Learning with Explanation5 May 2024
- 2024Human-compatible driving partners through data-regularized self-play reinforcement learning28 Mar 2024
- 2024C-GAIL: Stabilizing Generative Adversarial Imitation Learning with Control Theory26 Feb 2024
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Reward17 Feb 2024
- 2024Synergistic Reinforcement and Imitation Learning for Vision-driven Autonomous Flight of UAV Along River17 Jan 2024
- 2023A Survey of Reinforcement Learning from Human Feedback22 Dec 2023
- 2023BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks5 Dec 2023
- 2023Dynamic value alignment through preference aggregation of multiple objectives9 Oct 2023
- 2023RLHF-Blender: A Configurable Interactive Interface for Learning from Diverse Human Feedback8 Aug 2023
- 2023Residual Q-Learning: Offline and Online Policy Customization without Value15 Jun 2023
- 2023Synthetically Generating Human-like Data for Sequential Decision Making Tasks via Reward-Shaped Imitation Learning14 Apr 2023
- 2023Towards Solving Fuzzy Tasks with Human Feedback: A Retrospective of the MineRL BASALT 2022 Competition23 Mar 2023
- 2022Safe Imitation Learning of Nonlinear Model Predictive Control for Flexible Robots6 Dec 2022
- 2023Residual Q-Learning: Offline and Online Policy Customization without Valuecited by 9
- 2025X-IL: Exploring the Design Space of Imitation Learning Policiescited by 9
- 2022Safe Imitation Learning of Nonlinear Model Predictive Control for Flexible Robotscited by 3
- 2024C-GAIL: Stabilizing Generative Adversarial Imitation Learning with Control Theorycited by 3
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shapingcited by 3
- 2024"Give Me an Example Like This": Episodic Active Reinforcement Learning from Demonstrationscited by 2
- 2024Synergistic Reinforcement and Imitation Learning for Vision-driven Autonomous Flight of UAV Along Rivercited by 1
- 2024A Graph-based Adversarial Imitation Learning Framework for Reliable & Realtime Fleet Scheduling in Urban Air Mobilitycited by 1
- 2025On Discovering Algorithms for Adversarial Imitation Learningcited by 1
- 2023Synthetically Generating Human-like Data for Sequential Decision Making Tasks via Reward-Shaped Imitation Learningcited by 0
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Rewardcited by 0
- 2025Learning Dolly-In Filming From Demonstration Using a Ground-Based Robotcited by 0
12 most cited of 16 All 16 (JSON)
- 2026Latent Wasserstein Adversarial Imitation Learning5 Mar 2026
- 2026Generative adversarial imitation learning for robot swarms: Learning from human demonstrations and trained policies3 Mar 2026
- 2025An Introduction to Deep Reinforcement and Imitation Learning8 Dec 2025
- 2025Buzz, Choose, Forget: A Meta-Bandit Framework for Bee-Like Decision Making18 Oct 2025
- 2025On Discovering Algorithms for Adversarial Imitation Learning1 Oct 2025
- 2025Learning Dolly-In Filming From Demonstration Using a Ground-Based Robot30 Aug 2025
- 2025X-IL: Exploring the Design Space of Imitation Learning Policies17 Feb 2025
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shaping4 Oct 2024
- 2024A Graph-based Adversarial Imitation Learning Framework for Reliable & Realtime Fleet Scheduling in Urban Air Mobility16 Jul 2024
- 2024"Give Me an Example Like This": Episodic Active Reinforcement Learning from Demonstrations5 Jun 2024
- 2024C-GAIL: Stabilizing Generative Adversarial Imitation Learning with Control Theory26 Feb 2024
- 2024Multi Task Inverse Reinforcement Learning for Common Sense Reward17 Feb 2024
12 newest of 16 All 16 (JSON)
- 2023A Survey of Reinforcement Learning from Human Feedbackcited by 259
- 2023Towards Solving Fuzzy Tasks with Human Feedback: A Retrospective of the MineRL BASALT 2022 Competitioncited by 15
- 2023RLHF-Blender: A Configurable Interactive Interface for Learning from Diverse Human Feedbackcited by 15
- 2025Interactive Groupwise Comparison for Reinforcement Learning from Human Feedbackcited by 4
- 2025Interactive Groupwise Comparison for Reinforcement Learning from Human Feedback6 Jul 2025
- 2023A Survey of Reinforcement Learning from Human Feedback22 Dec 2023
- 2023RLHF-Blender: A Configurable Interactive Interface for Learning from Diverse Human Feedback8 Aug 2023
- 2023Towards Solving Fuzzy Tasks with Human Feedback: A Retrospective of the MineRL BASALT 2022 Competition23 Mar 2023
- 2025Reward Learning from Multiple Feedback Typescited by 9
- 2025Reward Learning from Multiple Feedback Types28 Feb 2025
- 2024Diffusion Imitation from Observationcited by 12
- 2024Diffusion Imitation from Observation7 Oct 2024
- 2023BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Taskscited by 8
- 2024RICE: Breaking Through the Training Bottlenecks of Reinforcement Learning with Explanationcited by 7
- 2025SPACeR: Self-Play Anchoring with Centralized Reference Modelscited by 7
- 2025WOFOSTGym: A Crop Simulator for Learning Annual and Perennial Crop Management Strategiescited by 4
- 2025Leveraging Genetic Algorithms for Efficient Demonstration Generation in Real-World Reinforcement Learning Environmentscited by 3
- 2024Preserving the Privacy of Reward Functions in MDPs through Deceptioncited by 1
- 2023Dynamic value alignment through preference aggregation of multiple objectivescited by 0
- 2025Supervised Reward Inferencecited by 0
- 2025Whenever, Wherever: Towards Orchestrating Crowd Simulations with Spatio-Temporal Spawn Dynamicscited by 0
- 2026Hierarchical Apprenticeship Learning from Imperfect Demonstrations with Evolving Rewardscited by 0
- 2026Hierarchical Apprenticeship Learning from Imperfect Demonstrations with Evolving Rewards31 Mar 2026
- 2025SPACeR: Self-Play Anchoring with Centralized Reference Models20 Oct 2025
- 2025Leveraging Genetic Algorithms for Efficient Demonstration Generation in Real-World Reinforcement Learning Environments1 Jul 2025
- 2025Whenever, Wherever: Towards Orchestrating Crowd Simulations with Spatio-Temporal Spawn Dynamics20 Mar 2025
- 2025WOFOSTGym: A Crop Simulator for Learning Annual and Perennial Crop Management Strategies26 Feb 2025
- 2025Supervised Reward Inference25 Feb 2025
- 2024Preserving the Privacy of Reward Functions in MDPs through Deception13 Jul 2024
- 2024RICE: Breaking Through the Training Bottlenecks of Reinforcement Learning with Explanation5 May 2024
- 2023BEDD: The MineRL BASALT Evaluation and Demonstrations Dataset for Training and Benchmarking Agents that Solve Fuzzy Tasks5 Dec 2023
- 2023Dynamic value alignment through preference aggregation of multiple objectives9 Oct 2023
Similar, not cited
Close in wording, no citation either way
- 2024Rethinking Inverse Reinforcement Learning: from Data Alignment to Task Alignment
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challenges
- 2022A Ranking Game for Imitation Learning
- 2019A Divergence Minimization Perspective on Imitation Learning Methods
- 2023Inverse Reinforcement Learning without Reinforcement Learning
- 2019Imitation Learning via Off-Policy Distribution Matching
- 2024EvIL: Evolution Strategies for Generalisable Imitation Learning
- 2024Inverse Reinforcement Learning by Estimating Expertise of Demonstrators