One-Shot Hierarchical Imitation Learning of Compound Visuomotor Tasks
On the map: Offline RL and imitation learning from data → Robot manipulation imitation learning
Abstract (arXiv)
We consider the problem of learning multi-stage vision-based tasks on a real robot from a single video of a human performing the task, while leveraging demonstration data of subtasks with other objects. This problem presents a number of major challenges. Video demonstrations without teleoperation are easy for humans to provide, but do not provide any direct supervision. Learning policies from raw pixels enables full generality but calls for large function approximators with many parameters to be learned. Finally, compound tasks can require impractical amounts of demonstration data, when treated as a monolithic skill. To address these challenges, we propose a method that learns both how to learn primitive behaviors from video demonstrations and how to dynamically compose these behaviors to perform multi-stage tasks by "watching" a human demonstrator. Our results on a simulated Sawyer robot and real PR2 robot illustrate our method for learning a variety of order fulfillment and kitchen serving tasks with novel objects and raw pixel inputs.
Key facts. Cited by 33 papers in the map (data through 14 Oct 2026). Placed in Robot manipulation imitation learning, one of 13 directions in Offline RL and imitation learning from data. Builds on 17 of its 38 references that are in the map. First posted on arXiv 25 Oct 2018.
Builds on
17 of 38 references.
References · 17
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2016Layer Normalizationcited by 5,669
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
- 2017Time-Contrastive Networks: Self-Supervised Learning from Videocited by 459
- 2017Deep Imitation Learning for Complex Manipulation Tasks from Virtual Reality Teleoperationcited by 289
- 2017One-Shot Imitation Learningcited by 233
- 2017Searching for Activation Functionscited by 169
- 2017Imitation from Observation: Learning to Imitate Behaviors from Raw Video via Context Translationcited by 167
- 2018One-Shot Imitation from Observing Humans via Domain-Adaptive Meta-Learningcited by 148
- 2017Third-Person Imitation Learningcited by 103
- 2017Vision-Based Multi-Task Manipulation for Inexpensive Robots Using End-To-End Learning from Demonstrationcited by 100
5 more
- 2017Neural Task Programming: Learning to Generalize Across Hierarchical Taskscited by 76
- 2017Multi-Modal Imitation Learning from Unstructured Demonstrations using Generative Adversarial Netscited by 64
- 2018TACO: Learning Task Decomposition via Temporal Alignment for Controlcited by 62
- 2018Neural Task Graphs: Generalizing to Unseen Tasks from a Single Video Demonstrationcited by 61
- 2017Translating Videos to Commands for Robotic Manipulation with Deep Recurrent Neural Networkscited by 21
Built on by
Citing papers per year: 2019: 3; 2020: 3; 2021: 12; 2022: 4; 2023: 4; 2024: 4; 2025: 2; 2026: 1.
33 papers.
Sort:
By area:
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scalecited by 229
- 2021Bottom-Up Skill Discovery from Unsegmented Demonstrations for Long-Horizon Robot Manipulationcited by 83
- 2023XSkill: Cross Embodiment Skill Discoverycited by 75
- 2023LOTUS: Continual Imitation Learning for Robot Manipulation Through Unsupervised Skill Discoverycited by 67
- 2023Behavior Retrieval: Few-Shot Imitation Learning by Querying Unlabeled Datasetscited by 58
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2020Deep Imitation Learning for Bimanual Robotic Manipulationcited by 50
- 2021SORNet: Spatial Object-Centric Representations for Sequential Manipulationcited by 48
- 2019Federated Imitation Learning: A Novel Framework for Cloud Robotic Systems with Heterogeneous Sensor Datacited by 24
- 2021CRIL: Continual Robot Imitation Learning via Generative and Prediction Modelcited by 24
- 2021Causal Navigation by Continuous-time Neural Networkscited by 23
- 2021Learning Multi-Stage Tasks with One Demonstration via Self-Replaycited by 21
21 more
- 2021Adversarial Option-Aware Hierarchical Imitation Learningcited by 19
- 2021Guided Imitation of Task and Motion Planningcited by 18
- 2022Cross-Domain Transfer via Semantic Skill Imitationcited by 18
- 2023Multi-task Hierarchical Adversarial Inverse Reinforcement Learningcited by 15
- 2022An Independently Learnable Hierarchical Model for Bilateral Control-Based Imitation Learning Applicationscited by 14
- 2025Think Small, Act Big: Primitive Prompt Learning for Lifelong Robot Manipulationcited by 14
- 2021Example-Driven Model-Based Reinforcement Learning for Solving Long-Horizon Visuomotor Taskscited by 13
- 2019To Follow or not to Follow: Selective Imitation Learning from Observationscited by 10
- 2021Towards Robust One-shot Task Execution using Knowledge Graph Embeddingscited by 8
- 2022Transfering Hierarchical Structure with Dual Meta Imitation Learningcited by 7
- 2024One-shot Imitation in a Non-Stationary Environment via Multi-Modal Skillcited by 7
- 2022Are All Vision Models Created Equal? A Study of the Open-Loop to Closed-Loop Causality Gapcited by 5
- 2026Robot Learning from Human Videos: A Surveycited by 5
- 2020Imitation Learning for High Precision Peg-in-Hole Taskscited by 3
- 2021Online Baum-Welch algorithm for Hierarchical Imitation Learningcited by 3
- 2025Trajectory Conditioned Cross-embodiment Skill Transfercited by 3
- 2021Off-policy Imitation Learning from Visual Inputscited by 2
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learningcited by 2
- 2020Multi-Instance Aware Localization for End-to-End Imitation Learningcited by 0
- 2024On the benefits of pixel-based hierarchical policies for task generalizationcited by 0
- 2024One-Shot Robust Imitation Learning for Long-Horizon Visuomotor Tasks from Unsegmented Demonstrationscited by 0
- 2026Robot Learning from Human Videos: A Survey30 Apr 2026
- 2025Trajectory Conditioned Cross-embodiment Skill Transfer9 Oct 2025
- 2025Think Small, Act Big: Primitive Prompt Learning for Lifelong Robot Manipulation1 Apr 2025
- 2024One-Shot Robust Imitation Learning for Long-Horizon Visuomotor Tasks from Unsegmented Demonstrations2 Oct 2024
- 2024On the benefits of pixel-based hierarchical policies for task generalization27 Jul 2024
- 2024One-shot Imitation in a Non-Stationary Environment via Multi-Modal Skill13 Feb 2024
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learning1 Feb 2024
- 2023LOTUS: Continual Imitation Learning for Robot Manipulation Through Unsupervised Skill Discovery3 Nov 2023
- 2023XSkill: Cross Embodiment Skill Discovery19 Jul 2023
- 2023Multi-task Hierarchical Adversarial Inverse Reinforcement Learning22 May 2023
- 2023Behavior Retrieval: Few-Shot Imitation Learning by Querying Unlabeled Datasets18 Apr 2023
- 2022Cross-Domain Transfer via Semantic Skill Imitation14 Dec 2022
21 more
- 2022Are All Vision Models Created Equal? A Study of the Open-Loop to Closed-Loop Causality Gap9 Oct 2022
- 2022An Independently Learnable Hierarchical Model for Bilateral Control-Based Imitation Learning Applications16 Mar 2022
- 2022Transfering Hierarchical Structure with Dual Meta Imitation Learning28 Jan 2022
- 2021Guided Imitation of Task and Motion Planning6 Dec 2021
- 2021Learning Multi-Stage Tasks with One Demonstration via Self-Replay14 Nov 2021
- 2021Off-policy Imitation Learning from Visual Inputs8 Nov 2021
- 2021Bottom-Up Skill Discovery from Unsegmented Demonstrations for Long-Horizon Robot Manipulation28 Sep 2021
- 2021Example-Driven Model-Based Reinforcement Learning for Solving Long-Horizon Visuomotor Tasks21 Sep 2021
- 2021SORNet: Spatial Object-Centric Representations for Sequential Manipulation8 Sep 2021
- 2021CRIL: Continual Robot Imitation Learning via Generative and Prediction Model17 Jun 2021
- 2021Causal Navigation by Continuous-time Neural Networks15 Jun 2021
- 2021Adversarial Option-Aware Hierarchical Imitation Learning10 Jun 2021
- 2021Towards Robust One-shot Task Execution using Knowledge Graph Embeddings10 May 2021
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scale16 Apr 2021
- 2021Online Baum-Welch algorithm for Hierarchical Imitation Learning22 Mar 2021
- 2020Multi-Instance Aware Localization for End-to-End Imitation Learning26 Dec 2020
- 2020Imitation Learning for High Precision Peg-in-Hole Tasks26 Dec 2020
- 2020Deep Imitation Learning for Bimanual Robotic Manipulation11 Oct 2020
- 2019Federated Imitation Learning: A Novel Framework for Cloud Robotic Systems with Heterogeneous Sensor Data24 Dec 2019
- 2019To Follow or not to Follow: Selective Imitation Learning from Observations16 Dec 2019
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variables20 Sep 2019
- 2021Bottom-Up Skill Discovery from Unsegmented Demonstrations for Long-Horizon Robot Manipulationcited by 83
- 2023XSkill: Cross Embodiment Skill Discoverycited by 75
- 2023LOTUS: Continual Imitation Learning for Robot Manipulation Through Unsupervised Skill Discoverycited by 67
- 2023Behavior Retrieval: Few-Shot Imitation Learning by Querying Unlabeled Datasetscited by 58
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2020Deep Imitation Learning for Bimanual Robotic Manipulationcited by 50
- 2019Federated Imitation Learning: A Novel Framework for Cloud Robotic Systems with Heterogeneous Sensor Datacited by 24
- 2021CRIL: Continual Robot Imitation Learning via Generative and Prediction Modelcited by 24
- 2021Causal Navigation by Continuous-time Neural Networkscited by 23
- 2021Learning Multi-Stage Tasks with One Demonstration via Self-Replaycited by 21
- 2021Adversarial Option-Aware Hierarchical Imitation Learningcited by 19
- 2023Multi-task Hierarchical Adversarial Inverse Reinforcement Learningcited by 15
12 most cited of 21 All 21 (JSON)
- 2024One-Shot Robust Imitation Learning for Long-Horizon Visuomotor Tasks from Unsegmented Demonstrations2 Oct 2024
- 2024Expert Proximity as Surrogate Rewards for Single Demonstration Imitation Learning1 Feb 2024
- 2023LOTUS: Continual Imitation Learning for Robot Manipulation Through Unsupervised Skill Discovery3 Nov 2023
- 2023XSkill: Cross Embodiment Skill Discovery19 Jul 2023
- 2023Multi-task Hierarchical Adversarial Inverse Reinforcement Learning22 May 2023
- 2023Behavior Retrieval: Few-Shot Imitation Learning by Querying Unlabeled Datasets18 Apr 2023
- 2022An Independently Learnable Hierarchical Model for Bilateral Control-Based Imitation Learning Applications16 Mar 2022
- 2022Transfering Hierarchical Structure with Dual Meta Imitation Learning28 Jan 2022
- 2021Learning Multi-Stage Tasks with One Demonstration via Self-Replay14 Nov 2021
- 2021Off-policy Imitation Learning from Visual Inputs8 Nov 2021
- 2021Bottom-Up Skill Discovery from Unsegmented Demonstrations for Long-Horizon Robot Manipulation28 Sep 2021
- 2021CRIL: Continual Robot Imitation Learning via Generative and Prediction Model17 Jun 2021
12 newest of 21 All 21 (JSON)
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scalecited by 229
- 2021SORNet: Spatial Object-Centric Representations for Sequential Manipulationcited by 48
- 2021Guided Imitation of Task and Motion Planningcited by 18
- 2022Cross-Domain Transfer via Semantic Skill Imitationcited by 18
- 2025Think Small, Act Big: Primitive Prompt Learning for Lifelong Robot Manipulationcited by 14
- 2021Example-Driven Model-Based Reinforcement Learning for Solving Long-Horizon Visuomotor Taskscited by 13
- 2021Towards Robust One-shot Task Execution using Knowledge Graph Embeddingscited by 8
- 2024One-shot Imitation in a Non-Stationary Environment via Multi-Modal Skillcited by 7
- 2022Are All Vision Models Created Equal? A Study of the Open-Loop to Closed-Loop Causality Gapcited by 5
- 2026Robot Learning from Human Videos: A Surveycited by 5
- 2025Trajectory Conditioned Cross-embodiment Skill Transfercited by 3
- 2026Robot Learning from Human Videos: A Survey30 Apr 2026
- 2025Trajectory Conditioned Cross-embodiment Skill Transfer9 Oct 2025
- 2025Think Small, Act Big: Primitive Prompt Learning for Lifelong Robot Manipulation1 Apr 2025
- 2024One-shot Imitation in a Non-Stationary Environment via Multi-Modal Skill13 Feb 2024
- 2022Cross-Domain Transfer via Semantic Skill Imitation14 Dec 2022
- 2022Are All Vision Models Created Equal? A Study of the Open-Loop to Closed-Loop Causality Gap9 Oct 2022
- 2021Guided Imitation of Task and Motion Planning6 Dec 2021
- 2021Example-Driven Model-Based Reinforcement Learning for Solving Long-Horizon Visuomotor Tasks21 Sep 2021
- 2021SORNet: Spatial Object-Centric Representations for Sequential Manipulation8 Sep 2021
- 2021Towards Robust One-shot Task Execution using Knowledge Graph Embeddings10 May 2021
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scale16 Apr 2021
Similar, not cited
Close in wording, no citation either way
- 2021Coarse-to-Fine Imitation Learning: Robot Manipulation from a Single Demonstration
- 2026Human-to-Robot Interaction: Learning from Video Demonstration for Robot Imitation
- 2019Third-Person Visual Imitation Learning via Decoupled Hierarchical Controller
- 2021Towards More Generalizable One-shot Visual Imitation Learning
- 2018Reinforcement and Imitation Learning for Diverse Visuomotor Skills
- 2022Learning Sensorimotor Primitives of Sequential Manipulation Tasks from Visual Demonstrations
- 2020Goal-Conditioned End-to-End Visuomotor Control for Versatile Skill Primitives
- 2021Seeing All the Angles: Learning Multiview Manipulation Policies for Contact-Rich Tasks from Demonstrations