State Alignment-based Imitation Learning
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
Consider an imitation learning problem that the imitator and the expert have different dynamics models. Most of the current imitation learning methods fail because they focus on imitating actions. We propose a novel state alignment-based imitation learning method to train the imitator to follow the state sequences in expert demonstrations as much as possible. The state alignment comes from both local and global perspectives and we combine them into a reinforcement learning framework by a regularized policy update objective. We show the superiority of our method on standard imitation learning settings and imitation learning settings where the expert and imitator have different dynamics models.
Key facts. Cited by 73 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 26 of its 46 references that are in the map. First posted on arXiv 21 Nov 2019.
Builds on
26 of 46 references.
Closest ancestors · 7
Same direction, newest first
- 2019Wasserstein Adversarial Imitation Learning19 Jun
- 2019Imitation Learning as f-Divergence Minimization30 May
- 2019Provably Efficient Imitation Learning from Observation Alone27 May
- 2018Sample-Efficient Imitation Learning via Generative Adversarial Nets6 Sep
- 2018Generative Adversarial Imitation from Observation17 Jul
- 2018Imitating Latent Policies from Observation21 May
- 2018Behavioral Cloning from Observation4 May
Other references · 19
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2013Auto-Encoding Variational Bayescited by 3,980
- 2015Trust Region Policy Optimizationcited by 2,503
- 2017Improved Training of Wasserstein GANscited by 2,223
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2017Wasserstein GANcited by 635
- 2016f-GAN: Training Generative Neural Samplers using Variational Divergence Minimizationcited by 504
- 2017Learning Robust Rewards with Adversarial Inverse Reinforcement Learningcited by 485
- 2018Data-Efficient Hierarchical Reinforcement Learningcited by 436
- 2019Efficient Exploration via State Marginal Matchingcited by 186
- 2017Imitation from Observation: Learning to Imitate Behaviors from Raw Video via Context Translationcited by 167
7 more
- 2018Zero-Shot Visual Imitationcited by 114
- 2018Near-Optimal Representation Learning for Hierarchical Reinforcement Learningcited by 107
- 2018Playing hard exploration games by watching YouTubecited by 98
- 2017Coordinated Multi-Agent Imitation Learningcited by 70
- 2018Observe and Look Further: Achieving Consistent Performance on Ataricited by 42
- 2018Adversarial Imitation via Variational Inverse Reinforcement Learningcited by 36
- 2018Optimal Transport on Discrete Domainscited by 16
Built on by
Citing papers per year: 2020: 7; 2021: 11; 2022: 21; 2023: 11; 2024: 13; 2025: 9; 2026: 1.
73 papers.
Sort:
By area:
- 2021DexMV: Imitation Learning for Dexterous Manipulation from Human Videoscited by 194
- 2021ManiSkill: Generalizable Manipulation Skill Benchmark with Large-Scale Demonstrationscited by 193
- 2020Primal Wasserstein Imitation Learningcited by 120
- 2021XIRL: Cross-embodiment Inverse Reinforcement Learningcited by 101
- 2023UniDexGrasp++: Improving Dexterous Grasping Policy Learning via Geometry-aware Curriculum and Iterative Generalist-Specialist Learningcited by 101
- 2020State-Only Imitation Learning for Dexterous Manipulationcited by 90
- 2021Off-Policy Imitation Learning from Observationscited by 70
- 2020f-IRL: Inverse Reinforcement Learning via State Marginal Matchingcited by 67
- 2020Strictly Batch Imitation Learning by Energy-based Distribution Matchingcited by 54
- 2021Cross-Domain Imitation Learning via Optimal Transportcited by 49
- 2023Zero-Shot Robot Manipulation from Passive Human Videoscited by 49
- 2022Learning Generalizable Dexterous Manipulation from Human Grasp Affordancecited by 46
38 more
- 2022Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matchingcited by 43
- 2022Maximum-Likelihood Inverse Reinforcement Learning with Finite-Time Guaranteescited by 42
- 2020Energy-Based Imitation Learningcited by 39
- 2025CLAM: Continuous Latent Action Models for Robot Learning from Unlabeled Demonstrationscited by 39
- 2020Robust Inverse Reinforcement Learning under Transition Dynamics Mismatchcited by 33
- 2023Cross-Domain Policy Adaptation via Value-Guided Data Filteringcited by 30
- 2021Learning from Imperfect Demonstrations from Agents with Varying Dynamicscited by 26
- 2022Robust Imitation Learning against Variations in Environment Dynamicscited by 26
- 2023State Regularized Policy Optimization on Data with Dynamics Shiftcited by 23
- 2022LobsDICE: Offline Learning from Observation via Stationary Distribution Correction Estimationcited by 22
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directionscited by 20
- 2022A Ranking Game for Imitation Learningcited by 18
- 2023MAHALO: Unifying Offline Reinforcement Learning and Imitation Learning from Observationscited by 18
- 2024PhysReaction: Physically Plausible Real-Time Humanoid Reaction Synthesis via Forward Dynamics Guided 4D Imitationcited by 18
- 2024A Comprehensive Survey of Cross-Domain Policy Transfer for Embodied Agentscited by 16
- 2021Unsupervised Domain Adaptation with Dynamics-Aware Rewards in Reinforcement Learningcited by 15
- 2022Learn what matters: cross-domain imitation learning with task-relevant embeddingscited by 14
- 2022REvolveR: Continuous Evolutionary Models for Robot-to-robot Policy Transfercited by 12
- 2022Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimizationcited by 12
- 2022Structural Estimation of Markov Decision Processes in High-Dimensional State Space with Finite-Time Guaranteescited by 12
- 2023CEIL: Generalized Contextual Imitation Learningcited by 12
- 2022Imitation Learning from Observations under Transition Model Disparitycited by 11
- 2023Domain Adaptive Imitation Learning with Visual Observationcited by 11
- 2021Learning Feasibility to Imitate Demonstrators with Different Dynamicscited by 10
- 2022Robust Learning from Observation with Model Misspecificationcited by 9
- 2024DrS: Learning Reusable Dense Rewards for Multi-Stage Taskscited by 9
- 2024xTED: Cross-Domain Adaptation via Diffusion-Based Trajectory Editingcited by 9
- 2022HERD: Continuous Human-to-Robot Evolution for Learning from Human Demonstrationcited by 8
- 2022Masked Imitation Learning: Discovering Environment-Invariant Modalities in Multimodal Demonstrationscited by 6
- 2021Skeletal Feature Compensation for Imitation Learning with Embodiment Mismatchcited by 5
- 2022Transferable Reward Learning by Dynamics-Agnostic Discriminator Ensemblecited by 5
- 2023Offline Imitation from Observation via Primal Wasserstein State Occupancy Matchingcited by 5
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRiticcited by 5
- 2022Reinforcement learning with Demonstrations from Mismatched Task under Sparse Rewardcited by 4
- 2021Off-Dynamics Inverse Reinforcement Learning from Hetero-Domaincited by 3
- 2022Co-Imitation: Learning Design and Behaviour by Imitationcited by 3
- 2022Out-of-Dynamics Imitation Learning from Multimodal Demonstrationscited by 3
- 2024Meta-Evolve: Continuous Robot Evolution for One-to-many Policy Transfercited by 3
50 most cited of 73 All 73 (JSON)
- 2026Semi-Supervised Cross-Domain Imitation Learning11 Feb 2026
- 2025When a Robot is More Capable than a Human: Learning from Constrained Demonstrators10 Oct 2025
- 2025Learning from Observation: A Survey of Recent Advances20 Sep 2025
- 2025Action-Constrained Imitation Learning20 Aug 2025
- 2025Mixture of Autoencoder Experts Guidance using Unlabeled and Incomplete Data for Exploration in Reinforcement Learning21 Jul 2025
- 2025Value from Observations: Towards Large-Scale Imitation Learning via Self-Improvement9 Jul 2025
- 2025CLAM: Continuous Latent Action Models for Robot Learning from Unlabeled Demonstrations8 May 2025
- 2025Understanding Inverse Reinforcement Learning under Overparameterization: Non-Asymptotic Analysis and Global Optimality22 Mar 2025
- 2025Policy Regularization on Globally Accessible States in Cross-Dynamics Reinforcement Learning10 Mar 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2024On Reward Transferability in Adversarial Inverse Reinforcement Learning: Insights from Random Matrix Theory10 Oct 2024
- 2024xTED: Cross-Domain Adaptation via Diffusion-Based Trajectory Editing13 Sep 2024
38 more
- 2024Mitigating Distribution Shift in Model-based Offline RL via Shifts-aware Reward Learning23 Aug 2024
- 2024Cross-Domain Policy Transfer by Representation Alignment via Multi-Domain Behavioral Cloning24 Jul 2024
- 2024Adaptive Teaching in Heterogeneous Agents: Balancing Surprise in Sparse Reward Scenarios23 May 2024
- 2024Meta-Evolve: Continuous Robot Evolution for One-to-many Policy Transfer6 May 2024
- 2024DrS: Learning Reusable Dense Rewards for Multi-Stage Tasks25 Apr 2024
- 2024AdaDemo: Data-Efficient Demonstration Expansion for Generalist Robotic Agent11 Apr 2024
- 2024DIDA: Denoised Imitation Learning based on Domain Adaptation4 Apr 2024
- 2024PhysReaction: Physically Plausible Real-Time Humanoid Reaction Synthesis via Forward Dynamics Guided 4D Imitation1 Apr 2024
- 2024Rethinking Adversarial Inverse Reinforcement Learning: Policy Imitation, Transferable Reward Recovery and Algebraic Equilibrium Proof21 Mar 2024
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directions21 Feb 2024
- 2024A Comprehensive Survey of Cross-Domain Policy Transfer for Embodied Agents7 Feb 2024
- 2023Policy Optimization with Smooth Guidance Learned from State-Only Demonstrations30 Dec 2023
- 2023Domain Adaptive Imitation Learning with Visual Observation1 Dec 2023
- 2023Offline Imitation from Observation via Primal Wasserstein State Occupancy Matching2 Nov 2023
- 2023Correspondence learning between morphologically different robots via task demonstrations20 Oct 2023
- 2023CEIL: Generalized Contextual Imitation Learning26 Jun 2023
- 2023State Regularized Policy Optimization on Data with Dynamics Shift6 Jun 2023
- 2023GAN-MPC: Training Model Predictive Controllers with Parameterized Cost Functions using Demonstrations from Non-identical Experts30 May 2023
- 2023Cross-Domain Policy Adaptation via Value-Guided Data Filtering28 May 2023
- 2023UniDexGrasp++: Improving Dexterous Grasping Policy Learning via Geometry-aware Curriculum and Iterative Generalist-Specialist Learning2 Apr 2023
- 2023MAHALO: Unifying Offline Reinforcement Learning and Imitation Learning from Observations30 Mar 2023
- 2023Zero-Shot Robot Manipulation from Passive Human Videos3 Feb 2023
- 2022HERD: Continuous Human-to-Robot Evolution for Learning from Human Demonstration8 Dec 2022
- 2022Reinforcement learning with Demonstrations from Mismatched Task under Sparse Reward3 Dec 2022
- 2022Transfer RL via the Undo Maps Formalism26 Nov 2022
- 2022Out-of-Dynamics Imitation Learning from Multimodal Demonstrations13 Nov 2022
- 2022Abstract-to-Executable Trajectory Translation for One-Shot Task Generalization14 Oct 2022
- 2022Maximum-Likelihood Inverse Reinforcement Learning with Finite-Time Guarantees4 Oct 2022
- 2022Structural Estimation of Markov Decision Processes in High-Dimensional State Space with Finite-Time Guarantees4 Oct 2022
- 2022Learn what matters: cross-domain imitation learning with task-relevant embeddings24 Sep 2022
- 2022Masked Imitation Learning: Discovering Environment-Invariant Modalities in Multimodal Demonstrations16 Sep 2022
- 2022Co-Imitation: Learning Design and Behaviour by Imitation2 Sep 2022
- 2022Robust Imitation Learning against Variations in Environment Dynamics19 Jun 2022
- 2022Model-based Offline Imitation Learning with Non-expert Data11 Jun 2022
- 2022Transferable Reward Learning by Dynamics-Agnostic Discriminator Ensemble1 Jun 2022
- 2022Imitation Learning from Observations under Transition Model Disparity25 Apr 2022
- 2022Learning Generalizable Dexterous Manipulation from Human Grasp Affordance5 Apr 2022
- 2022Plan Your Target and Learn Your Skills: Transferable State-Only Imitation Learning via Decoupled Policy Optimization4 Mar 2022
50 newest of 73 All 73 (JSON)
- 2021DexMV: Imitation Learning for Dexterous Manipulation from Human Videoscited by 194
- 2020Primal Wasserstein Imitation Learningcited by 120
- 2021XIRL: Cross-embodiment Inverse Reinforcement Learningcited by 101
- 2020State-Only Imitation Learning for Dexterous Manipulationcited by 90
- 2021Off-Policy Imitation Learning from Observationscited by 70
- 2020f-IRL: Inverse Reinforcement Learning via State Marginal Matchingcited by 67
- 2020Strictly Batch Imitation Learning by Energy-based Distribution Matchingcited by 54
- 2021Cross-Domain Imitation Learning via Optimal Transportcited by 49
- 2022Learning Generalizable Dexterous Manipulation from Human Grasp Affordancecited by 46
- 2022Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matchingcited by 43
- 2022Maximum-Likelihood Inverse Reinforcement Learning with Finite-Time Guaranteescited by 42
- 2020Energy-Based Imitation Learningcited by 39
12 most cited of 48 All 48 (JSON)
- 2026Semi-Supervised Cross-Domain Imitation Learning11 Feb 2026
- 2025When a Robot is More Capable than a Human: Learning from Constrained Demonstrators10 Oct 2025
- 2025Learning from Observation: A Survey of Recent Advances20 Sep 2025
- 2025Action-Constrained Imitation Learning20 Aug 2025
- 2025Value from Observations: Towards Large-Scale Imitation Learning via Self-Improvement9 Jul 2025
- 2025Understanding Inverse Reinforcement Learning under Overparameterization: Non-Asymptotic Analysis and Global Optimality22 Mar 2025
- 2025Policy Regularization on Globally Accessible States in Cross-Dynamics Reinforcement Learning10 Mar 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2024On Reward Transferability in Adversarial Inverse Reinforcement Learning: Insights from Random Matrix Theory10 Oct 2024
- 2024Mitigating Distribution Shift in Model-based Offline RL via Shifts-aware Reward Learning23 Aug 2024
- 2024Cross-Domain Policy Transfer by Representation Alignment via Multi-Domain Behavioral Cloning24 Jul 2024
- 2024Adaptive Teaching in Heterogeneous Agents: Balancing Surprise in Sparse Reward Scenarios23 May 2024
12 newest of 48 All 48 (JSON)
- 2021ManiSkill: Generalizable Manipulation Skill Benchmark with Large-Scale Demonstrationscited by 193
- 2023UniDexGrasp++: Improving Dexterous Grasping Policy Learning via Geometry-aware Curriculum and Iterative Generalist-Specialist Learningcited by 101
- 2023Zero-Shot Robot Manipulation from Passive Human Videoscited by 49
- 2025CLAM: Continuous Latent Action Models for Robot Learning from Unlabeled Demonstrationscited by 39
- 2023Cross-Domain Policy Adaptation via Value-Guided Data Filteringcited by 30
- 2023State Regularized Policy Optimization on Data with Dynamics Shiftcited by 23
- 2022LobsDICE: Offline Learning from Observation via Stationary Distribution Correction Estimationcited by 22
- 2024PhysReaction: Physically Plausible Real-Time Humanoid Reaction Synthesis via Forward Dynamics Guided 4D Imitationcited by 18
- 2024A Comprehensive Survey of Cross-Domain Policy Transfer for Embodied Agentscited by 16
- 2021Unsupervised Domain Adaptation with Dynamics-Aware Rewards in Reinforcement Learningcited by 15
- 2022REvolveR: Continuous Evolutionary Models for Robot-to-robot Policy Transfercited by 12
- 2024DrS: Learning Reusable Dense Rewards for Multi-Stage Taskscited by 9
12 most cited of 23 All 23 (JSON)
- 2025Mixture of Autoencoder Experts Guidance using Unlabeled and Incomplete Data for Exploration in Reinforcement Learning21 Jul 2025
- 2025CLAM: Continuous Latent Action Models for Robot Learning from Unlabeled Demonstrations8 May 2025
- 2024Meta-Evolve: Continuous Robot Evolution for One-to-many Policy Transfer6 May 2024
- 2024DrS: Learning Reusable Dense Rewards for Multi-Stage Tasks25 Apr 2024
- 2024PhysReaction: Physically Plausible Real-Time Humanoid Reaction Synthesis via Forward Dynamics Guided 4D Imitation1 Apr 2024
- 2024A Comprehensive Survey of Cross-Domain Policy Transfer for Embodied Agents7 Feb 2024
- 2023Policy Optimization with Smooth Guidance Learned from State-Only Demonstrations30 Dec 2023
- 2023Offline Imitation from Observation via Primal Wasserstein State Occupancy Matching2 Nov 2023
- 2023Correspondence learning between morphologically different robots via task demonstrations20 Oct 2023
- 2023State Regularized Policy Optimization on Data with Dynamics Shift6 Jun 2023
- 2023GAN-MPC: Training Model Predictive Controllers with Parameterized Cost Functions using Demonstrations from Non-identical Experts30 May 2023
- 2023Cross-Domain Policy Adaptation via Value-Guided Data Filtering28 May 2023
12 newest of 23 All 23 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2020State-only Imitation with Transition Dynamics Mismatch
- 2025Imitation Learning in the Deep Learning Era: A Novel Taxonomy and Recent Advances
- 2025A Model-Based Approach to Imitation Learning through Multi-Step Predictions
- 2021Cross-domain Imitation from Observations
- 2024Rethinking Inverse Reinforcement Learning: from Data Alignment to Task Alignment
- 2023Robust Visual Imitation Learning with Inverse Dynamics Representations
- 2024Efficient Imitation Learning with Conservative World Models
- 2020Interactive Imitation Learning in State-Space