Model-Based Inverse Reinforcement Learning from Visual Demonstrations
On the map: Offline RL and imitation learning from data → Real-world robot offline RL
Abstract (arXiv)
Scaling model-based inverse reinforcement learning (IRL) to real robotic manipulation tasks with unknown dynamics remains an open problem. The key challenges lie in learning good dynamics models, developing algorithms that scale to high-dimensional state-spaces and being able to learn from both visual and proprioceptive demonstrations. In this work, we present a gradient-based inverse reinforcement learning framework that utilizes a pre-trained visual dynamics model to learn cost functions when given only visual human demonstrations. The learned cost functions are then used to reproduce the demonstrated behavior via visual model predictive control. We evaluate our framework on hardware on two basic object manipulation tasks.
Key facts. Cited by 52 papers in the map (data through 7 Oct 2026). Placed in Real-world robot offline RL, one of 13 directions in Offline RL and imitation learning from data. Builds on 16 of its 32 references that are in the map. First posted on arXiv 18 Oct 2020.
Builds on
16 of 32 references.
References · 16
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2018An Algorithmic Perspective on Imitation Learningcited by 455
- 2016Guided Cost Learning: Deep Inverse Optimal Control via Policy Optimizationcited by 400
- 2016Deep Visual Foresight for Planning Robot Motioncited by 276
- 2015Embed to Control: A Locally Linear Latent Dynamics Model for Control from Raw Imagescited by 266
- 2018Visual Foresight: Model-Based Deep Reinforcement Learning for Vision-Based Robotic Controlcited by 220
- 2017Imitation from Observation: Learning to Imitate Behaviors from Raw Video via Context Translationcited by 167
- 2020DIGIT: A Novel Design for a Low-Cost Compact High-Resolution Tactile Sensor with Application to In-Hand Manipulationcited by 164
- 2019kPAM: KeyPoint Affordances for Category-Level Robotic Manipulationcited by 152
- 2018Dense Object Nets: Learning Dense Visual Object Descriptors By and For Robotic Manipulationcited by 140
- 2019Unsupervised Learning of Object Keypoints for Perception and Controlcited by 109
- 2019Unsupervised Learning of Object Structure and Dynamics from Videoscited by 92
4 more
- 2019Generalized Inner Loop Meta-Learningcited by 92
- 2016Unsupervised Perceptual Rewards for Imitation Learningcited by 64
- 2019Meta-Learning via Learned Losscited by 60
- 2020Approximate Inverse Reinforcement Learning from Vision-based Imitation Learningcited by 6
Built on by
Citing papers per year: 2021: 9; 2022: 8; 2023: 11; 2024: 15; 2025: 8; 2026: 1.
52 papers.
Sort:
By area:
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 738
- 2022R3M: A Universal Visual Representation for Robot Manipulationcited by 678
- 2024Mobile ALOHA: Learning Bimanual Mobile Manipulation with Low-Cost Whole-Body Teleoperationcited by 377
- 2023MimicPlay: Long-Horizon Imitation Learning by Watching Human Playcited by 214
- 2022Masked World Models for Visual Controlcited by 169
- 2021Learning Generalizable Robotic Reward Functions from "In-The-Wild" Human Videoscited by 116
- 2022VideoDex: Learning Dexterity from Internet Videoscited by 101
- 2023RoboTAP: Tracking Arbitrary Points for Few-Shot Visual Imitationcited by 88
- 2025Humanoid Policy ~ Human Policycited by 65
- 2021Imitation Learning: Progress, Taxonomies and Challengescited by 56
- 2023Towards A Unified Agent with Foundation Modelscited by 46
- 2021Visual Adversarial Imitation Learning using Variational Modelscited by 42
38 more
- 2024Surgical Robot Transformer (SRT): Imitation Learning for Surgical Taskscited by 38
- 2024Comprehensive Overview of Reward Engineering and Shaping in Advancing Reinforcement Learning Applicationscited by 34
- 2021VOILA: Visual-Observation-Only Imitation Learning for Autonomous Navigationcited by 27
- 2023One-Shot Imitation Learning: A Pose Estimation Perspectivecited by 21
- 2024AdaptAgent: Adapting Multimodal Web Agents with Few-Shot Learning from Human Demonstrationscited by 19
- 2025Dexterous Manipulation through Imitation Learning: A Surveycited by 18
- 2022Cross-Domain Transfer via Semantic Skill Imitationcited by 17
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignmentcited by 17
- 2023Learning Video-Conditioned Policies for Unseen Manipulation Taskscited by 15
- 2022Learning Reward Functions for Robotic Manipulation by Observing Humanscited by 14
- 2023A Survey of Demonstration Learningcited by 12
- 2022Continual Learning from Demonstration of Robotics Skillscited by 11
- 2023Human-oriented Representation Learning for Robotic Manipulationcited by 10
- 2023A Bayesian Approach to Robust Inverse Reinforcement Learningcited by 9
- 2024Adapt2Reward: Adapting Video-Language Models to Generalizable Robotic Rewards via Failure Promptscited by 9
- 2024VLMimic: Vision Language Models are Visual Imitation Learner for Fine-grained Actionscited by 8
- 2025H2R-Grounder: A Paired-Data-Free Paradigm for Translating Human Interaction Videos into Physically Grounded Robot Videoscited by 8
- 2023BC-IRL: Learning Generalizable Reward Functions from Demonstrationscited by 7
- 2024Reward-free World Models for Online Imitation Learningcited by 7
- 2023Giving Robots a Hand: Learning Generalizable Manipulation with Eye-in-Hand Human Video Demonstrationscited by 6
- 2023Scalable and Efficient Continual Learning from Demonstration via a Hypernetwork-generated Stable Dynamics Modelcited by 6
- 2025Chain-of-Modality: Learning Manipulation Programs from Multimodal Human Videos with Vision-Language-Modelscited by 6
- 2021Vision-based system identification and 3D keypoint discovery using dynamics constraintscited by 5
- 2022Learning Object Manipulation Skills from Video via Approximate Differentiable Physicscited by 5
- 2024Multi-task real-robot data with gaze attention for dual-arm fine manipulationcited by 5
- 2024TieBot: Learning to Knot a Tie from Visual Demonstration through a Real-to-Sim-to-Real Approachcited by 5
- 2021Back to Reality for Imitation Learningcited by 3
- 2024CANDERE-COACH: Reinforcement Learning from Noisy Feedbackcited by 3
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shapingcited by 3
- 2024Language-Model-Assisted Bi-Level Programming for Reward Learning from Internet Videoscited by 3
- 2021A Differentiable Recipe for Learning Visual Non-Prehensile Planar Manipulationcited by 2
- 2025Model Predictive Adversarial Imitation Learning for Planning from Observationcited by 2
- 2021Imitation Learning via Simultaneous Optimization of Policies and Auxiliary Trajectoriescited by 1
- 2022Modeling Human Behavior Part I -- Learning and Belief Approachescited by 1
- 2024Visual IRL for Human-Like Robotic Manipulationcited by 1
- 2021Learning Time-Invariant Reward Functions through Model-Based Inverse Reinforcement Learningcited by 0
- 2024Semi-Supervised One-Shot Imitation Learningcited by 0
- 2025Inverse RL Scene Dynamics Learning for Nonlinear Predictive Control in Autonomous Vehiclescited by 0
50 most cited of 52 All 52 (JSON)
- 2026Q-based Variational Inverse Reinforcement Learning17 Aug 2026
- 2025H2R-Grounder: A Paired-Data-Free Paradigm for Translating Human Interaction Videos into Physically Grounded Robot Videos10 Dec 2025
- 2025Model Predictive Adversarial Imitation Learning for Planning from Observation29 Jul 2025
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planning11 Jun 2025
- 2025Coupled Distributional Random Expert Distillation for World Model Online Imitation Learning4 May 2025
- 2025Chain-of-Modality: Learning Manipulation Programs from Multimodal Human Videos with Vision-Language-Models17 Apr 2025
- 2025Dexterous Manipulation through Imitation Learning: A Survey4 Apr 2025
- 2025Inverse RL Scene Dynamics Learning for Nonlinear Predictive Control in Autonomous Vehicles2 Apr 2025
- 2025Humanoid Policy ~ Human Policy17 Mar 2025
- 2024Visual IRL for Human-Like Robotic Manipulation16 Dec 2024
- 2024AdaptAgent: Adapting Multimodal Web Agents with Few-Shot Learning from Human Demonstrations20 Nov 2024
- 2024VLMimic: Vision Language Models are Visual Imitation Learner for Fine-grained Actions28 Oct 2024
38 more
- 2024Reward-free World Models for Online Imitation Learning17 Oct 2024
- 2024Language-Model-Assisted Bi-Level Programming for Reward Learning from Internet Videos11 Oct 2024
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shaping4 Oct 2024
- 2024CANDERE-COACH: Reinforcement Learning from Noisy Feedback23 Sep 2024
- 2024Semi-Supervised One-Shot Imitation Learning9 Aug 2024
- 2024Comprehensive Overview of Reward Engineering and Shaping in Advancing Reinforcement Learning Applications22 Jul 2024
- 2024Adapt2Reward: Adapting Video-Language Models to Generalizable Robotic Rewards via Failure Prompts20 Jul 2024
- 2024Surgical Robot Transformer (SRT): Imitation Learning for Surgical Tasks17 Jul 2024
- 2024TieBot: Learning to Knot a Tie from Visual Demonstration through a Real-to-Sim-to-Real Approach3 Jul 2024
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignment4 Jun 2024
- 2024Multi-task real-robot data with gaze attention for dual-arm fine manipulation15 Jan 2024
- 2024Mobile ALOHA: Learning Bimanual Mobile Manipulation with Low-Cost Whole-Body Teleoperation4 Jan 2024
- 2023Scalable and Efficient Continual Learning from Demonstration via a Hypernetwork-generated Stable Dynamics Model6 Nov 2023
- 2023One-Shot Imitation Learning: A Pose Estimation Perspective18 Oct 2023
- 2023Human-oriented Representation Learning for Robotic Manipulation4 Oct 2023
- 2023A Bayesian Approach to Robust Inverse Reinforcement Learning15 Sep 2023
- 2023RoboTAP: Tracking Arbitrary Points for Few-Shot Visual Imitation30 Aug 2023
- 2023Towards A Unified Agent with Foundation Models18 Jul 2023
- 2023Giving Robots a Hand: Learning Generalizable Manipulation with Eye-in-Hand Human Video Demonstrations12 Jul 2023
- 2023Learning Video-Conditioned Policies for Unseen Manipulation Tasks10 May 2023
- 2023BC-IRL: Learning Generalizable Reward Functions from Demonstrations28 Mar 2023
- 2023A Survey of Demonstration Learning20 Mar 2023
- 2023MimicPlay: Long-Horizon Imitation Learning by Watching Human Play24 Feb 2023
- 2022Cross-Domain Transfer via Semantic Skill Imitation14 Dec 2022
- 2022VideoDex: Learning Dexterity from Internet Videos8 Dec 2022
- 2022Learning Reward Functions for Robotic Manipulation by Observing Humans16 Nov 2022
- 2022Learning Object Manipulation Skills from Video via Approximate Differentiable Physics3 Aug 2022
- 2022Masked World Models for Visual Control28 Jun 2022
- 2022Modeling Human Behavior Part I -- Learning and Belief Approaches13 May 2022
- 2022R3M: A Universal Visual Representation for Robot Manipulation23 Mar 2022
- 2022Continual Learning from Demonstration of Robotics Skills14 Feb 2022
- 2021Back to Reality for Imitation Learning25 Nov 2021
- 2021A Differentiable Recipe for Learning Visual Non-Prehensile Planar Manipulation9 Nov 2021
- 2021Vision-based system identification and 3D keypoint discovery using dynamics constraints13 Sep 2021
- 2021Visual Adversarial Imitation Learning using Variational Models16 Jul 2021
- 2021Learning Time-Invariant Reward Functions through Model-Based Inverse Reinforcement Learning7 Jul 2021
- 2021Imitation Learning: Progress, Taxonomies and Challenges23 Jun 2021
- 2021VOILA: Visual-Observation-Only Imitation Learning for Autonomous Navigation19 May 2021
50 newest of 52 All 52 (JSON)
- 2024Mobile ALOHA: Learning Bimanual Mobile Manipulation with Low-Cost Whole-Body Teleoperationcited by 377
- 2023MimicPlay: Long-Horizon Imitation Learning by Watching Human Playcited by 214
- 2021Imitation Learning: Progress, Taxonomies and Challengescited by 56
- 2021Visual Adversarial Imitation Learning using Variational Modelscited by 42
- 2024Surgical Robot Transformer (SRT): Imitation Learning for Surgical Taskscited by 38
- 2021VOILA: Visual-Observation-Only Imitation Learning for Autonomous Navigationcited by 27
- 2023One-Shot Imitation Learning: A Pose Estimation Perspectivecited by 21
- 2025Dexterous Manipulation through Imitation Learning: A Surveycited by 18
- 2023A Survey of Demonstration Learningcited by 12
- 2022Continual Learning from Demonstration of Robotics Skillscited by 11
- 2023A Bayesian Approach to Robust Inverse Reinforcement Learningcited by 9
- 2023BC-IRL: Learning Generalizable Reward Functions from Demonstrationscited by 7
12 most cited of 23 All 23 (JSON)
- 2026Q-based Variational Inverse Reinforcement Learning17 Aug 2026
- 2025Model Predictive Adversarial Imitation Learning for Planning from Observation29 Jul 2025
- 2025Dexterous Manipulation through Imitation Learning: A Survey4 Apr 2025
- 2024Language-Model-Assisted Bi-Level Programming for Reward Learning from Internet Videos11 Oct 2024
- 2024Enhancing Inverse Reinforcement Learning through Encoding Dynamic Information in Reward Shaping4 Oct 2024
- 2024CANDERE-COACH: Reinforcement Learning from Noisy Feedback23 Sep 2024
- 2024Semi-Supervised One-Shot Imitation Learning9 Aug 2024
- 2024Surgical Robot Transformer (SRT): Imitation Learning for Surgical Tasks17 Jul 2024
- 2024Multi-task real-robot data with gaze attention for dual-arm fine manipulation15 Jan 2024
- 2024Mobile ALOHA: Learning Bimanual Mobile Manipulation with Low-Cost Whole-Body Teleoperation4 Jan 2024
- 2023Scalable and Efficient Continual Learning from Demonstration via a Hypernetwork-generated Stable Dynamics Model6 Nov 2023
- 2023One-Shot Imitation Learning: A Pose Estimation Perspective18 Oct 2023
12 newest of 23 All 23 (JSON)
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 738
- 2022Masked World Models for Visual Controlcited by 169
- 2024Reward-free World Models for Online Imitation Learningcited by 7
- 2025Coupled Distributional Random Expert Distillation for World Model Online Imitation Learningcited by 0
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planning11 Jun 2025
- 2025Coupled Distributional Random Expert Distillation for World Model Online Imitation Learning4 May 2025
- 2024Reward-free World Models for Online Imitation Learning17 Oct 2024
- 2022Masked World Models for Visual Control28 Jun 2022
- 2022R3M: A Universal Visual Representation for Robot Manipulationcited by 678
- 2021Learning Generalizable Robotic Reward Functions from "In-The-Wild" Human Videoscited by 116
- 2022VideoDex: Learning Dexterity from Internet Videoscited by 101
- 2023RoboTAP: Tracking Arbitrary Points for Few-Shot Visual Imitationcited by 88
- 2025Humanoid Policy ~ Human Policycited by 65
- 2024Comprehensive Overview of Reward Engineering and Shaping in Advancing Reinforcement Learning Applicationscited by 34
- 2022Cross-Domain Transfer via Semantic Skill Imitationcited by 17
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignmentcited by 17
- 2023Learning Video-Conditioned Policies for Unseen Manipulation Taskscited by 15
- 2022Learning Reward Functions for Robotic Manipulation by Observing Humanscited by 14
- 2023Human-oriented Representation Learning for Robotic Manipulationcited by 10
- 2024Adapt2Reward: Adapting Video-Language Models to Generalizable Robotic Rewards via Failure Promptscited by 9
12 most cited of 21 All 21 (JSON)
- 2025H2R-Grounder: A Paired-Data-Free Paradigm for Translating Human Interaction Videos into Physically Grounded Robot Videos10 Dec 2025
- 2025Chain-of-Modality: Learning Manipulation Programs from Multimodal Human Videos with Vision-Language-Models17 Apr 2025
- 2025Humanoid Policy ~ Human Policy17 Mar 2025
- 2024Visual IRL for Human-Like Robotic Manipulation16 Dec 2024
- 2024Comprehensive Overview of Reward Engineering and Shaping in Advancing Reinforcement Learning Applications22 Jul 2024
- 2024Adapt2Reward: Adapting Video-Language Models to Generalizable Robotic Rewards via Failure Prompts20 Jul 2024
- 2024TieBot: Learning to Knot a Tie from Visual Demonstration through a Real-to-Sim-to-Real Approach3 Jul 2024
- 2024Cross-Embodiment Robot Manipulation Skill Transfer using Latent Space Alignment4 Jun 2024
- 2023Human-oriented Representation Learning for Robotic Manipulation4 Oct 2023
- 2023RoboTAP: Tracking Arbitrary Points for Few-Shot Visual Imitation30 Aug 2023
- 2023Giving Robots a Hand: Learning Generalizable Manipulation with Eye-in-Hand Human Video Demonstrations12 Jul 2023
- 2023Learning Video-Conditioned Policies for Unseen Manipulation Tasks10 May 2023
12 newest of 21 All 21 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024Robotic Arm Manipulation with Inverse Reinforcement Learning & TD-MPC
- 2024Predictive Inverse Dynamics Models are Scalable Learners for Robotic Manipulation
- 2026Online World Modeling Enables Real-World Inverse Reinforcement Learning from Observation
- 2020State-Only Imitation Learning for Dexterous Manipulation
- 2026Coherent Off-Policy Improvement of Large Behavior Models with Learned Rewards
- 2024Precise and Dexterous Robotic Manipulation via Human-in-the-Loop Reinforcement Learning
- 2019RIDM: Reinforced Inverse Dynamics Modeling for Learning from a Single Observed Demonstration
- 2021Residual Reinforcement Learning from Demonstrations