Imitating Latent Policies from Observation
On the map: Offline RL and imitation learning from data → Imitation learning methods
Abstract (arXiv)
In this paper, we describe a novel approach to imitation learning that infers latent policies directly from state observations. We introduce a method that characterizes the causal effects of latent actions on observations while simultaneously predicting their likelihood. We then outline an action alignment procedure that leverages a small amount of environment interactions to determine a mapping between the latent and real-world actions. We show that this corrected labeling can be used for imitating the observed behavior, even though no expert actions are given. We evaluate our approach within classic control environments and a platform game and demonstrate that it performs better than standard approaches. Code for this work is available at https://github.com/ashedwards/ILPO.
Key facts. Cited by 101 papers in the map (data through 14 Oct 2026). Placed in Imitation learning methods, one of 13 directions in Offline RL and imitation learning from data. First posted on arXiv 21 May 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 4; 2020: 5; 2021: 10; 2022: 19; 2023: 8; 2024: 16; 2025: 16; 2026: 23.
101 papers.
Sort:
By area:
- 2024Genie: Generative Interactive Environmentscited by 747
- 2022MineDojo: Building Open-Ended Embodied Agents with Internet-Scale Knowledgecited by 478
- 2022Video PreTraining (VPT): Learning to Act by Watching Unlabeled Online Videoscited by 345
- 2024Latent Action Pretraining from Videoscited by 321
- 2025Motus: A Unified Latent Action World Modelcited by 251
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challengescited by 148
- 2020Primal Wasserstein Imitation Learningcited by 121
- 2023Learning to Act without Actionscited by 121
- 2026VLA-JEPA: Enhancing Vision-Language-Action Model with Latent World Modelcited by 118
- 2025AdaWorld: Learning Adaptable World Models with Latent Actionscited by 114
- 2022Reinforcement Learning with Action-Free Pre-Training from Videoscited by 108
- 2021XIRL: Cross-embodiment Inverse Reinforcement Learningcited by 107
38 more
- 2019Recent Advances in Imitation Learning from Observationcited by 93
- 2020State-Only Imitation Learning for Dexterous Manipulationcited by 91
- 2024Video as the New Language for Real-World Decision Makingcited by 87
- 2019Provably Efficient Imitation Learning from Observation Alonecited by 78
- 2019State Alignment-based Imitation Learningcited by 73
- 2021Off-Policy Imitation Learning from Observationscited by 70
- 2021Imitation Learning: Progress, Taxonomies and Challengescited by 58
- 2026Learning Latent Action World Models In The Wildcited by 47
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2024DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Controlcited by 44
- 2025Latent Action Learning Requires Supervision in the Presence of Distractorscited by 44
- 2020Provable Representation Learning for Imitation Learning via Bi-level Optimizationcited by 41
- 2024Towards Generalist Robot Learning from Internet Video: A Surveycited by 40
- 2025What Do Latent Action Models Actually Learn?cited by 32
- 2022Deep Interactive Motion Prediction and Planning: Playing Games with Motion Prediction Modelscited by 27
- 2026NitroGen: An Open Foundation Model for Generalist Gaming Agentscited by 27
- 2025Latent Diffusion Planning for Imitation Learningcited by 26
- 2019Imitation Learning from Video by Leveraging Proprioceptioncited by 23
- 2021Self-Supervised Disentangled Representation Learning for Third-Person Imitation Learningcited by 23
- 2023Efficient Preference-Based Reinforcement Learning Using Learned Dynamics Modelscited by 21
- 2022MUGEN: A Playground for Video-Audio-Text Multimodal Understanding and GENerationcited by 20
- 2022A Ranking Game for Imitation Learningcited by 18
- 2022Discriminator-Guided Model-Based Offline Imitation Learningcited by 17
- 2024From Words to Actions: Unveiling the Theoretical Underpinnings of LLM-Driven Autonomous Systemscited by 17
- 2022Dungeons and Data: A Large-Scale NetHack Datasetcited by 16
- 2024A Survey of Imitation Learning Methods, Environments and Metricscited by 16
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modelingcited by 15
- 2023Learning from Visual Observation via Offline Pretrained State-to-Go Transformercited by 14
- 2025PlaySlot: Learning Inverse Latent Dynamics for Controllable Object-Centric Video Prediction and Planningcited by 14
- 2021Imitation by Predicting Observationscited by 13
- 2021Recent Advances in Leveraging Human Guidance for Sequential Decision-Making Taskscited by 13
- 2022Imitation Learning from Observations under Transition Model Disparitycited by 11
- 2023Domain Adaptive Imitation Learning with Visual Observationcited by 11
- 2025Closing the Train-Test Gap in World Models for Gradient-Based Planningcited by 11
- 2022Adversarial Imitation Learning from Video using a State Observercited by 10
- 2023Imitation Learning from Observation with Automatic Discount Schedulingcited by 10
- 2024AD3: Implicit Action is the Key for World Models to Distinguish the Diverse Visual Distractorscited by 10
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasetscited by 10
50 most cited of 101 All 101 (JSON)
- 2026Markovian Dynamics Enforcer: Feasibility Preserving Correction on Learned Dynamics Manifolds30 Sep 2026
- 2026What Matters for Latent Actions in Robot Learning20 Aug 2026
- 2026Latent-Action-Guided Video-Language Feature Learning for Surgical Instrument-Tissue Interaction Recognition22 Jul 2026
- 2026Causally Debiased Latent Action Model for Embodied Action Conditioned World Models10 Jul 2026
- 2026Geometry-Aware Motion Latents for Learning Robust Manipulation Policies6 Jul 2026
- 2026Flowing With Purpose: Latent Action Guided Flow Matching Policies For Robotic Manipulation22 Jun 2026
- 2026FlexLAM: Resolving the Bottleneck Trade-off in Latent Action Learning17 Jun 2026
- 2026Learning Red Agent Policy from Observations for Neurosymbolic Autonomous Cyber Agents16 Jun 2026
- 20263DVLA: Enhancing Vision-Language-Action Models via 3D Spatial and Instance Understanding28 May 2026
- 2026ASH: Agents that Self-Hone in Long-Horizon Worlds14 May 2026
- 2026From Imagined Futures to Executable Actions: Mixture of Latent Actions for Robot Manipulation12 May 2026
- 2026Identifying Latent Actions and Dynamics from Offline Data via Demonstrator Diversity18 Mar 2026
38 more
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modeling4 Mar 2026
- 2026UniLACT: Depth-Aware RGB Latent Action Learning for Vision-Language-Action Models23 Feb 2026
- 2026Factored Latent Action World Models18 Feb 2026
- 2026Olaf-World: Orienting Latent Actions for Video World Modeling10 Feb 2026
- 2026VLA-JEPA: Enhancing Vision-Language-Action Model with Latent World Model10 Feb 2026
- 2026On the Sample Efficiency of Inverse Dynamics Models for Semi-Supervised Imitation Learning2 Feb 2026
- 2026Segment to Focus: Guiding Latent Action Models in the Presence of Distractors2 Feb 2026
- 2026Grounding Generated Videos in Feasible Plans via World Models2 Feb 2026
- 2026Vision-Language Models Unlock Task-Centric Latent Actions30 Jan 2026
- 2026Learning Latent Action World Models In The Wild8 Jan 2026
- 2026NitroGen: An Open Foundation Model for Generalist Gaming Agents4 Jan 2026
- 2025Videos are Sample-Efficient Supervisions: Behavior Cloning from Videos via Latent Representations25 Dec 2025
- 2025Motus: A Unified Latent Action World Model15 Dec 2025
- 2025Closing the Train-Test Gap in World Models for Gradient-Based Planning10 Dec 2025
- 2025Improving Robustness to Out-of-Distribution States in Imitation Learning via Deep Koopman-Boosted Diffusion Policy1 Nov 2025
- 2025TimeRewarder: Learning Dense Reward from Passive Videos via Frame-wise Temporal Distance30 Sep 2025
- 2025Learning from Observation: A Survey of Recent Advances20 Sep 2025
- 2025VITA: Zero-Shot Value Functions via Test-Time Adaptation of Vision-Language Models11 Jun 2025
- 2025What Do Latent Action Models Actually Learn?27 May 2025
- 2025Latent Diffusion Planning for Imitation Learning23 Apr 2025
- 2025Controlling Large Language Model with Latent Actions27 Mar 2025
- 2025Offline Action-Free Learning of Ex-BMDPs by Comparing Diverse Datasets26 Mar 2025
- 2025AdaWorld: Learning Adaptable World Models with Latent Actions24 Mar 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2025Object-Centric Latent Action Learning13 Feb 2025
- 2025PlaySlot: Learning Inverse Latent Dynamics for Controllable Object-Centric Video Prediction and Planning11 Feb 2025
- 2025Latent Action Learning Requires Supervision in the Presence of Distractors1 Feb 2025
- 2024Sample-efficient Unsupervised Policy Cloning from Ensemble Self-supervised Labeled Videos14 Dec 2024
- 2024Pre-trained Visual Dynamics Representations for Efficient Policy Learning5 Nov 2024
- 2024Latent Action Pretraining from Videos15 Oct 2024
- 2024Imitation Learning with Limited Actions via Diffusion Planners and Deep Koopman Controllers10 Oct 2024
- 2024DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Control18 Sep 2024
- 2024Goal-Reaching Policy Learning from Non-Expert Observations via Effective Subgoal Guidance6 Sep 2024
- 2024NOLO: Navigate Only Look Once2 Aug 2024
- 2024Explorative Imitation Learning: A Path Signature Approach for Continuous Environments5 Jul 2024
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasets13 Jun 2024
- 2024From Words to Actions: Unveiling the Theoretical Underpinnings of LLM-Driven Autonomous Systems30 May 2024
- 2024Towards Generalist Robot Learning from Internet Video: A Survey30 Apr 2024
50 newest of 101 All 101 (JSON)
- 2022Video PreTraining (VPT): Learning to Act by Watching Unlabeled Online Videoscited by 345
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challengescited by 148
- 2020Primal Wasserstein Imitation Learningcited by 121
- 2021XIRL: Cross-embodiment Inverse Reinforcement Learningcited by 107
- 2019Recent Advances in Imitation Learning from Observationcited by 93
- 2020State-Only Imitation Learning for Dexterous Manipulationcited by 91
- 2019Provably Efficient Imitation Learning from Observation Alonecited by 78
- 2019State Alignment-based Imitation Learningcited by 73
- 2021Off-Policy Imitation Learning from Observationscited by 70
- 2021Imitation Learning: Progress, Taxonomies and Challengescited by 58
- 2021MobILE: Model-Based Imitation Learning From Observation Alonecited by 44
- 2024DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Controlcited by 44
12 most cited of 36 All 36 (JSON)
- 2026On the Sample Efficiency of Inverse Dynamics Models for Semi-Supervised Imitation Learning2 Feb 2026
- 2025Learning from Observation: A Survey of Recent Advances20 Sep 2025
- 2025IL-SOAR : Imitation Learning with Soft Optimistic Actor cRitic27 Feb 2025
- 2024Imitation Learning with Limited Actions via Diffusion Planners and Deep Koopman Controllers10 Oct 2024
- 2024DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor Control18 Sep 2024
- 2024Explorative Imitation Learning: A Path Signature Approach for Continuous Environments5 Jul 2024
- 2024A Dual Approach to Imitation Learning from Observations with Offline Datasets13 Jun 2024
- 2024A Survey of Imitation Learning Methods, Environments and Metrics30 Apr 2024
- 2023Domain Adaptive Imitation Learning with Visual Observation1 Dec 2023
- 2023Imitation Learning from Observation with Automatic Discount Scheduling11 Oct 2023
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challenges5 Sep 2023
- 2022Dungeons and Data: A Large-Scale NetHack Dataset1 Nov 2022
12 newest of 36 All 36 (JSON)
- 2024Genie: Generative Interactive Environmentscited by 747
- 2022MineDojo: Building Open-Ended Embodied Agents with Internet-Scale Knowledgecited by 478
- 2025Motus: A Unified Latent Action World Modelcited by 251
- 2023Learning to Act without Actionscited by 121
- 2026VLA-JEPA: Enhancing Vision-Language-Action Model with Latent World Modelcited by 118
- 2025AdaWorld: Learning Adaptable World Models with Latent Actionscited by 114
- 2026Learning Latent Action World Models In The Wildcited by 47
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modelingcited by 15
- 2025Closing the Train-Test Gap in World Models for Gradient-Based Planningcited by 11
- 2024AD3: Implicit Action is the Key for World Models to Distinguish the Diverse Visual Distractorscited by 10
- 2026Olaf-World: Orienting Latent Actions for Video World Modelingcited by 10
- 2026Factored Latent Action World Modelscited by 7
12 most cited of 15 All 15 (JSON)
- 2026Causally Debiased Latent Action Model for Embodied Action Conditioned World Models10 Jul 2026
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modeling4 Mar 2026
- 2026Factored Latent Action World Models18 Feb 2026
- 2026Olaf-World: Orienting Latent Actions for Video World Modeling10 Feb 2026
- 2026VLA-JEPA: Enhancing Vision-Language-Action Model with Latent World Model10 Feb 2026
- 2026Grounding Generated Videos in Feasible Plans via World Models2 Feb 2026
- 2026Learning Latent Action World Models In The Wild8 Jan 2026
- 2025Motus: A Unified Latent Action World Model15 Dec 2025
- 2025Closing the Train-Test Gap in World Models for Gradient-Based Planning10 Dec 2025
- 2025AdaWorld: Learning Adaptable World Models with Latent Actions24 Mar 2025
- 2024AD3: Implicit Action is the Key for World Models to Distinguish the Diverse Visual Distractors15 Mar 2024
- 2024Genie: Generative Interactive Environments23 Feb 2024
12 newest of 15 All 15 (JSON)
- 2024Pre-trained Visual Dynamics Representations for Efficient Policy Learningcited by 8
- 2025TimeRewarder: Learning Dense Reward from Passive Videos via Frame-wise Temporal Distancecited by 6
- 2024A Model-Based Approach for Improving Reinforcement Learning Efficiency Leveraging Expert Observationscited by 0
- 2026Learning Red Agent Policy from Observations for Neurosymbolic Autonomous Cyber Agentscited by 0
- 2026Learning Red Agent Policy from Observations for Neurosymbolic Autonomous Cyber Agents16 Jun 2026
- 2025TimeRewarder: Learning Dense Reward from Passive Videos via Frame-wise Temporal Distance30 Sep 2025
- 2024Pre-trained Visual Dynamics Representations for Efficient Policy Learning5 Nov 2024
- 2024A Model-Based Approach for Improving Reinforcement Learning Efficiency Leveraging Expert Observations29 Feb 2024
- 2026Vision-Language Models Unlock Task-Centric Latent Actions30 Jan 2026
- 2022Reinforcement Learning with Action-Free Pre-Training from Videoscited by 108
- 2024Video as the New Language for Real-World Decision Makingcited by 87
- 2025Latent Action Learning Requires Supervision in the Presence of Distractorscited by 44
- 2024Towards Generalist Robot Learning from Internet Video: A Surveycited by 40
- 2025What Do Latent Action Models Actually Learn?cited by 32
- 2022Deep Interactive Motion Prediction and Planning: Playing Games with Motion Prediction Modelscited by 27
- 2023Efficient Preference-Based Reinforcement Learning Using Learned Dynamics Modelscited by 21
- 2022MUGEN: A Playground for Video-Audio-Text Multimodal Understanding and GENerationcited by 20
- 2024From Words to Actions: Unveiling the Theoretical Underpinnings of LLM-Driven Autonomous Systemscited by 17
- 2023Learning from Visual Observation via Offline Pretrained State-to-Go Transformercited by 14
- 2025PlaySlot: Learning Inverse Latent Dynamics for Controllable Object-Centric Video Prediction and Planningcited by 14
- 2021Recent Advances in Leveraging Human Guidance for Sequential Decision-Making Taskscited by 13
12 most cited of 37 All 37 (JSON)
- 2026Markovian Dynamics Enforcer: Feasibility Preserving Correction on Learned Dynamics Manifolds30 Sep 2026
- 2026What Matters for Latent Actions in Robot Learning20 Aug 2026
- 2026Latent-Action-Guided Video-Language Feature Learning for Surgical Instrument-Tissue Interaction Recognition22 Jul 2026
- 2026Geometry-Aware Motion Latents for Learning Robust Manipulation Policies6 Jul 2026
- 2026FlexLAM: Resolving the Bottleneck Trade-off in Latent Action Learning17 Jun 2026
- 2026ASH: Agents that Self-Hone in Long-Horizon Worlds14 May 2026
- 2026From Imagined Futures to Executable Actions: Mixture of Latent Actions for Robot Manipulation12 May 2026
- 2026Identifying Latent Actions and Dynamics from Offline Data via Demonstrator Diversity18 Mar 2026
- 2026Segment to Focus: Guiding Latent Action Models in the Presence of Distractors2 Feb 2026
- 2025Videos are Sample-Efficient Supervisions: Behavior Cloning from Videos via Latent Representations25 Dec 2025
- 2025What Do Latent Action Models Actually Learn?27 May 2025
- 2025Controlling Large Language Model with Latent Actions27 Mar 2025
12 newest of 37 All 37 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2020Imitating Unknown Policies via Exploration
- 2026When Does On-Policy Interaction Help? Representational Tradeoffs in Value-Based Imitation Learning
- 2022Causal Imitation Learning under Temporally Correlated Noise
- 2022Information-Theoretic Policy Learning from Partial Observations with Fully Informed Decision Makers
- 2023Invariant Causal Imitation Learning for Generalizable Policies
- 2018Fast Policy Learning through Imitation and Reinforcement
- 2022Deconfounding Imitation Learning with Variational Inference
- 2019Causal Confusion in Imitation Learning