Robust Imitation Learning against Variations in Environment Dynamics
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
In this paper, we propose a robust imitation learning (IL) framework that improves the robustness of IL when environment dynamics are perturbed. The existing IL framework trained in a single environment can catastrophically fail with perturbations in environment dynamics because it does not capture the situation that underlying environment dynamics can be changed. Our framework effectively deals with environments with varying dynamics by imitating multiple experts in sampled environment dynamics to enhance the robustness in general variations in environment dynamics. In order to robustly imitate the multiple sample experts, we minimize the risk with respect to the Jensen-Shannon divergence between the agent's policy and each of the sample experts. Numerical results show that our algorithm significantly improves robustness against dynamics perturbations compared to conventional IL baselines.
Key facts. Cited by 26 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 39 of its 52 references that are in the map. First posted on arXiv 19 Jun 2022.
Builds on
39 of 52 references.
Closest ancestors · 12
Same direction, newest first
- 2021Cross-domain Imitation from Observations20 May
- 2020The MAGICAL Benchmark for Robust Imitation1 Nov
- 2020Robust Imitation Learning from Noisy Demonstrations20 Oct
- 2020ADAIL: Adaptive Adversarial Imitation Learning23 Aug
- 2020State-only Imitation with Transition Dynamics Mismatch27 Feb
- 2020Multi-Agent Interactions Modeling with Correlated Policies4 Jan
- 2019State Alignment-based Imitation Learning21 Nov
- 2019Watch, Try, Learn: Meta-Learning from Demonstrations and Reward7 Jun
- 2019Imitation Learning from Imperfect Demonstration27 Jan
- 2018Discriminator-Actor-Critic: Addressing Sample Inefficiency and Reward Bias in Adversarial Imitation Learning9 Sep
- 2018Generative Adversarial Imitation from Observation17 Jul
- 2018Behavioral Cloning from Observation4 May
Other references · 27
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2015Continuous control with deep reinforcement learningcited by 3,279
- 2015Trust Region Policy Optimizationcited by 2,503
- 2018Addressing Function Approximation Error in Actor-Critic Methodscited by 2,302
- 2017Improved Training of Wasserstein GANscited by 2,223
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 1,916
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2017Learning Robust Rewards with Adversarial Inverse Reinforcement Learningcited by 485
- 2017Robust Adversarial Reinforcement Learningcited by 439
- 2016Guided Cost Learning: Deep Inverse Optimal Control via Policy Optimizationcited by 400
- 2017One-Shot Imitation Learningcited by 232
15 more
- 2017One-Shot Visual Imitation Learning via Meta-Learningcited by 225
- 2016Coupled Generative Adversarial Networkscited by 216
- 2017DART: Noise Injection for Robust Imitation Learningcited by 187
- 2019Action Robust Reinforcement Learning and Applications in Continuous Controlcited by 167
- 2017Robust Imitation of Diverse Behaviorscited by 85
- 2019Robust Reinforcement Learning for Continuous Control with Model Misspecificationcited by 76
- 2018Task-Embedded Control Networks for Few-Shot Imitation Learningcited by 65
- 2020Robust Reinforcement Learning using Adversarial Populationscited by 64
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2020Experience Replay with Likelihood-free Importance Weightscited by 42
- 2018Soft-Robust Actor-Critic Policy-Gradientcited by 37
- 2018Learning a Prior over Intent via Meta-Inverse Reinforcement Learningcited by 36
- 2020Robust Inverse Reinforcement Learning under Transition Dynamics Mismatchcited by 33
- 2018Multi-task Maximum Entropy Inverse Reinforcement Learningcited by 20
- 2018Learning Robust Optionscited by 17
Built on by
Citing papers per year: 2022: 2; 2023: 4; 2024: 4; 2025: 8; 2026: 8.
26 papers.
Sort:
By area:
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challengescited by 145
- 2024Contrastive Representation for Data Filtering in Cross-Domain Offline Reinforcement Learningcited by 18
- 2023Policy Gradient Algorithms for Robust MDPs with Non-Rectangular Uncertainty Setscited by 15
- 2023Domain Adaptive Imitation Learning with Visual Observationcited by 11
- 2023Robust Visual Imitation Learning with Inverse Dynamics Representationscited by 8
- 2024Exclusively Penalized Q-learning for Offline Reinforcement Learningcited by 8
- 2025Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuningcited by 8
- 2025Wolfpack Adversarial Attack for Robust Multi-Agent Reinforcement Learningcited by 6
- 2026Flow Actor-Critic for Offline Reinforcement Learningcited by 6
- 2025Self-Improving Skill Learning for Robust Skill-based Meta-Reinforcement Learningcited by 5
- 2022Environment Design for Inverse Reinforcement Learningcited by 4
- 2025Task-Aware Virtual Training: Enhancing Generalization in Meta-Reinforcement Learning for Out-of-Distribution Taskscited by 3
14 more
- 2024DIDA: Denoised Imitation Learning based on Domain Adaptationcited by 2
- 2025Efficient Cross-Domain Offline Reinforcement Learning with Dynamics- and Value-Aligned Data Filteringcited by 2
- 2025Inversely Learning Transferable Rewards via Abstracted Statescited by 1
- 2025Balance Equation-based Distributionally Robust Offline Imitation Learningcited by 1
- 2025Distributionally Robust Imitation Learning: Layered Control Architecture for Certifiable Autonomycited by 1
- 2026Interaction-Breaking Adversarial Learning Framework for Robust Multi-Agent Reinforcement Learningcited by 1
- 2022Cross apprenticeship learning framework: Properties and solution approachescited by 0
- 2024DyPNIPP: Predicting Environment Dynamics for RL-based Robust Informative Path Planningcited by 0
- 2026Bridging Domain Gaps with Target-Aligned Generation for Offline Reinforcement Learningcited by 0
- 2026When Dynamics Shift, Robust Task Inference Wins: Offline Imitation Learning with Behavior Foundation Models Revisitedcited by 0
- 2026ConTraIRL: Factorized Contrastive Abstractions for Transferable IRLcited by 0
- 2026Distributionally Robust and Safe Imitation Learningcited by 0
- 2026Scalable Causal Imitation Learningcited 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
- 2026Scalable Causal Imitation Learning18 Jul 2026
- 2026Distributionally Robust and Safe Imitation Learning15 Jul 2026
- 2026ConTraIRL: Factorized Contrastive Abstractions for Transferable IRL2 Jun 2026
- 2026Interaction-Breaking Adversarial Learning Framework for Robust Multi-Agent Reinforcement Learning18 May 2026
- 2026When Dynamics Shift, Robust Task Inference Wins: Offline Imitation Learning with Behavior Foundation Models Revisited16 May 2026
- 2026Bridging Domain Gaps with Target-Aligned Generation for Offline Reinforcement Learning13 May 2026
- 2026Flow Actor-Critic for Offline Reinforcement Learning20 Feb 2026
- 2025Distributionally Robust Imitation Learning: Layered Control Architecture for Certifiable Autonomy19 Dec 2025
- 2025Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuning18 Dec 2025
- 2025Efficient Cross-Domain Offline Reinforcement Learning with Dynamics- and Value-Aligned Data Filtering2 Dec 2025
- 2025Balance Equation-based Distributionally Robust Offline Imitation Learning11 Nov 2025
14 more
- 2025Self-Improving Skill Learning for Robust Skill-based Meta-Reinforcement Learning6 Feb 2025
- 2025Wolfpack Adversarial Attack for Robust Multi-Agent Reinforcement Learning5 Feb 2025
- 2025Task-Aware Virtual Training: Enhancing Generalization in Meta-Reinforcement Learning for Out-of-Distribution Tasks5 Feb 2025
- 2025Inversely Learning Transferable Rewards via Abstracted States3 Jan 2025
- 2024DyPNIPP: Predicting Environment Dynamics for RL-based Robust Informative Path Planning22 Oct 2024
- 2024Exclusively Penalized Q-learning for Offline Reinforcement Learning23 May 2024
- 2024Contrastive Representation for Data Filtering in Cross-Domain Offline Reinforcement Learning10 May 2024
- 2024DIDA: Denoised Imitation Learning based on Domain Adaptation4 Apr 2024
- 2023Domain Adaptive Imitation Learning with Visual Observation1 Dec 2023
- 2023Robust Visual Imitation Learning with Inverse Dynamics Representations22 Oct 2023
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challenges5 Sep 2023
- 2023Policy Gradient Algorithms for Robust MDPs with Non-Rectangular Uncertainty Sets30 May 2023
- 2022Environment Design for Inverse Reinforcement Learning26 Oct 2022
- 2022Cross apprenticeship learning framework: Properties and solution approaches6 Sep 2022
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challengescited by 145
- 2024Contrastive Representation for Data Filtering in Cross-Domain Offline Reinforcement Learningcited by 18
- 2023Domain Adaptive Imitation Learning with Visual Observationcited by 11
- 2023Robust Visual Imitation Learning with Inverse Dynamics Representationscited by 8
- 2024Exclusively Penalized Q-learning for Offline Reinforcement Learningcited by 8
- 2025Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuningcited by 8
- 2026Flow Actor-Critic for Offline Reinforcement Learningcited by 6
- 2022Environment Design for Inverse Reinforcement Learningcited by 4
- 2024DIDA: Denoised Imitation Learning based on Domain Adaptationcited by 2
- 2025Efficient Cross-Domain Offline Reinforcement Learning with Dynamics- and Value-Aligned Data Filteringcited by 2
- 2025Inversely Learning Transferable Rewards via Abstracted Statescited by 1
- 2025Balance Equation-based Distributionally Robust Offline Imitation Learningcited by 1
12 most cited of 19 All 19 (JSON)
- 2026Generalize and Guide: Decomposing Rewards for Few-Shot Inverse Reinforcement Learning20 Jul 2026
- 2026Scalable Causal Imitation Learning18 Jul 2026
- 2026Distributionally Robust and Safe Imitation Learning15 Jul 2026
- 2026ConTraIRL: Factorized Contrastive Abstractions for Transferable IRL2 Jun 2026
- 2026When Dynamics Shift, Robust Task Inference Wins: Offline Imitation Learning with Behavior Foundation Models Revisited16 May 2026
- 2026Bridging Domain Gaps with Target-Aligned Generation for Offline Reinforcement Learning13 May 2026
- 2026Flow Actor-Critic for Offline Reinforcement Learning20 Feb 2026
- 2025Distributionally Robust Imitation Learning: Layered Control Architecture for Certifiable Autonomy19 Dec 2025
- 2025Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuning18 Dec 2025
- 2025Efficient Cross-Domain Offline Reinforcement Learning with Dynamics- and Value-Aligned Data Filtering2 Dec 2025
- 2025Balance Equation-based Distributionally Robust Offline Imitation Learning11 Nov 2025
- 2025Inversely Learning Transferable Rewards via Abstracted States3 Jan 2025
12 newest of 19 All 19 (JSON)
- 2023Policy Gradient Algorithms for Robust MDPs with Non-Rectangular Uncertainty Setscited by 15
- 2025Wolfpack Adversarial Attack for Robust Multi-Agent Reinforcement Learningcited by 6
- 2025Self-Improving Skill Learning for Robust Skill-based Meta-Reinforcement Learningcited by 5
- 2025Task-Aware Virtual Training: Enhancing Generalization in Meta-Reinforcement Learning for Out-of-Distribution Taskscited by 3
- 2026Interaction-Breaking Adversarial Learning Framework for Robust Multi-Agent Reinforcement Learningcited by 1
- 2022Cross apprenticeship learning framework: Properties and solution approachescited by 0
- 2024DyPNIPP: Predicting Environment Dynamics for RL-based Robust Informative Path Planningcited by 0
- 2026Interaction-Breaking Adversarial Learning Framework for Robust Multi-Agent Reinforcement Learning18 May 2026
- 2025Self-Improving Skill Learning for Robust Skill-based Meta-Reinforcement Learning6 Feb 2025
- 2025Wolfpack Adversarial Attack for Robust Multi-Agent Reinforcement Learning5 Feb 2025
- 2025Task-Aware Virtual Training: Enhancing Generalization in Meta-Reinforcement Learning for Out-of-Distribution Tasks5 Feb 2025
- 2024DyPNIPP: Predicting Environment Dynamics for RL-based Robust Informative Path Planning22 Oct 2024
- 2023Policy Gradient Algorithms for Robust MDPs with Non-Rectangular Uncertainty Sets30 May 2023
- 2022Cross apprenticeship learning framework: Properties and solution approaches6 Sep 2022
Similar, not cited
Close in wording, no citation either way
- 2022Robust Learning from Observation with Model Misspecification
- 2022Robust Imitation of a Few Demonstrations with a Backwards Model
- 2025RISE: Robust Imitation through Stochastic Encoding
- 2023Provably Efficient Adversarial Imitation Learning with Unknown Transitions
- 2024Reward-free World Models for Online Imitation Learning
- 2024Efficient Imitation Learning with Conservative World Models
- 2025A Model-Based Approach to Imitation Learning through Multi-Step Predictions
- 2022Imitation Learning by State-Only Distribution Matching