Learning to Generalize Across Long-Horizon Tasks from Human Demonstrations
On the map: Offline RL and imitation learning from data → Robot manipulation imitation learning
Abstract (arXiv)
Imitation learning is an effective and safe technique to train robot policies in the real world because it does not depend on an expensive random exploration process. However, due to the lack of exploration, learning policies that generalize beyond the demonstrated behaviors is still an open challenge. We present a novel imitation learning framework to enable robots to 1) learn complex real world manipulation tasks efficiently from a small number of human demonstrations, and 2) synthesize new behaviors not contained in the collected demonstrations. Our key insight is that multi-task domains often present a latent structure, where demonstrated trajectories for different tasks intersect at common regions of the state space. We present Generalization Through Imitation (GTI), a two-stage offline imitation learning algorithm that exploits this intersecting structure to train goal-directed policies that generalize to unseen start and goal state combinations. In the first stage of GTI, we train a stochastic policy that leverages trajectory intersections to have the capacity to compose behaviors from different demonstration trajectories together. In the second stage of GTI, we collect a small set of rollouts from the unconditioned stochastic policy of the first stage, and train a goal-directed agent to generalize to novel start and goal configurations. We validate GTI in both simulated domains and a challenging long-horizon robotic manipulation domain in the real world. Additional results and videos are available at https://sites.google.com/view/gti2020/ .
Key facts. Cited by 114 papers in the map (data through 7 Oct 2026). Placed in Robot manipulation imitation learning, one of 13 directions in Offline RL and imitation learning from data. First posted on arXiv 13 Mar 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 4; 2021: 8; 2022: 21; 2023: 21; 2024: 28; 2025: 20; 2026: 12.
114 papers.
Sort:
By area:
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 699
- 2024RoboCasa: Large-Scale Simulation of Everyday Tasks for Generalist Robotscited by 391
- 2023MimicGen: A Data Generation System for Scalable Robot Learning using Human Demonstrationscited by 308
- 2023Goal-Conditioned Imitation Learning using Score-based Diffusion Policiescited by 229
- 2023MimicPlay: Long-Horizon Imitation Learning by Watching Human Playcited by 214
- 2024EgoMimic: Scaling Imitation Learning via Egocentric Videocited by 160
- 2022VIOLA: Imitation Learning for Vision-Based Manipulation with Object Proposal Priorscited by 135
- 2025RoboTwin: Dual-Arm Robot Benchmark with Generative Digital Twinscited by 117
- 2023Policy Representation via Diffusion Probability Model for Reinforcement Learningcited by 105
- 2024DexMimicGen: Automated Data Generation for Bimanual Dexterous Manipulation via Imitation Learningcited by 104
- 2020Human-in-the-Loop Imitation Learning using Remote Teleoperationcited by 103
38 more
- 2022From Play to Policy: Conditional Behavior Generation from Uncurated Robot Datacited by 103
- 2022Robot Learning on the Job: Human-in-the-Loop Autonomy and Learning During Deploymentcited by 101
- 2021Learning by Watching: Physical Imitation of Manipulation Skills from Human Videoscited by 87
- 2020COG: Connecting New Skills to Past Experience with Offline Reinforcement Learningcited by 83
- 2021Bottom-Up Skill Discovery from Unsegmented Demonstrations for Long-Horizon Robot Manipulationcited by 82
- 2022Latent Plans for Task-Agnostic Offline Reinforcement Learningcited by 81
- 2022BITS: Bi-level Imitation for Traffic Simulationcited by 80
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
- 2022Learning and Retrieval from Prior Data for Skill-based Imitation Learningcited by 77
- 2023XSkill: Cross Embodiment Skill Discoverycited by 74
- 2022Can Foundation Models Perform Zero-Shot Task Specification For Robot Manipulation?cited by 70
- 2021Understanding Domain Randomization for Sim-to-real Transfercited by 69
- 2021A Workflow for Offline Model-Free Robotic Reinforcement Learningcited by 68
- 2024RoboTwin: Dual-Arm Robot Benchmark with Generative Digital Twins (early version)cited by 68
- 2023Imitating Task and Motion Planning with Visuomotor Transformerscited by 63
- 2023LOTUS: Continual Imitation Learning for Robot Manipulation Through Unsupervised Skill Discoverycited by 63
- 2024OKAMI: Teaching Humanoid Robots Manipulation Skills through Single Video Imitationcited by 62
- 2023Learning Generalizable Manipulation Policies with Object-Centric 3D Representationscited by 60
- 2021Error-Aware Imitation Learning from Teleoperation Data for Mobile Manipulationcited by 59
- 2022Learning Neuro-Symbolic Skills for Bilevel Planningcited by 55
- 2022End-to-End Affordance Learning for Robotic Manipulationcited by 54
- 2022Skill-based Model-based Reinforcement Learningcited by 50
- 2022Hierarchical Planning Through Goal-Conditioned Offline Reinforcement Learningcited by 48
- 2024SkillMimicGen: Automated Demonstration Generation for Efficient Skill Learning and Deploymentcited by 48
- 2024Vision-based Manipulation from Single Human Video with Open-World Object Graphscited by 46
- 2022MoCoDA: Model-based Counterfactual Data Augmentationcited by 44
- 2025Sim-and-Real Co-Training: A Simple Recipe for Vision-Based Robotic Manipulationcited by 42
- 2024CleanDiffuser: An Easy-to-use Modularized Library for Diffusion Models in Decision Makingcited by 38
- 2023Query-Dependent Prompt Evaluation and Optimization with Offline Inverse RLcited by 37
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrationscited by 35
- 2023Human-in-the-Loop Task and Motion Planning for Imitation Learningcited by 28
- 2021Generalization Through Hand-Eye Coordination: An Action Space for Learning Spatially-Invariant Visuomotor Controlcited by 24
- 2024IntervenGen: Interventional Data Generation for Robust and Data-Efficient Robot Imitation Learningcited by 24
- 2023Universal Visual Decomposer: Long-Horizon Manipulation Made Easycited by 23
- 2022Goal-conditioned dual-action imitation learning for dexterous dual-arm robot manipulationcited by 21
- 2022PATO: Policy Assisted TeleOperation for Scalable Robot Data Collectioncited by 19
- 2025Dexterous Manipulation through Imitation Learning: A Surveycited by 18
- 2023Transfer Learning in Robotics: An Upcoming Breakthrough? A Review of Promises and Challengescited by 17
50 most cited of 114 All 114 (JSON)
- 2026Same Scene, Different Task: Skill Alignment for Compositional Generalization in VLAs30 Sep 2026
- 2026Vision-Language Grounded Task-Context-Aware Imitation Learning for Robotic Disassembly15 Sep 2026
- 2026MA-VLA: Multi-Arm Vision-Language-Action Model for Collaboration and Compositional Generalization26 Aug 2026
- 2026Iterative Grasp Pose Refinement: A Deep Reinforcement Learning Approach for 2D Vision18 Aug 2026
- 2026hint2: Hierarchical World Models for Inference-Time Temporal Logic Guidance13 Aug 2026
- 2026AutoIntervene: Calibrated Intervention for Action-Chunking Imitation Learning Policies7 Aug 2026
- 2026Language-Critique Imitation Learning from Suboptimal Demonstrations1 Jul 2026
- 2026Offline Reinforcement Learning for Fluid Controls: Data-based Multi-observational Policy Extraction30 Jun 2026
- 2026Imagine to Ensure Safety in Hierarchical Reinforcement Learning21 Jun 2026
- 2026GHOST: Hierarchical Sub-Goal Policies for Generalizing Robot Manipulation8 Jun 2026
- 2026When Life Gives You BC, Make Q-functions: Extracting Q-values from Behavior Cloning for On-Robot Reinforcement Learning6 May 2026
- 2026Optimizing Neurorobot Policy under Limited Demonstration Data through Preference Regret4 Apr 2026
38 more
- 2025From Power to Precision: Learning Fine-grained Dexterity for Multi-fingered Robotic Hands17 Nov 2025
- 2025PADiff: Predictive and Adaptive Diffusion Policies for Ad Hoc Teamwork10 Nov 2025
- 2025Imitation Learning in the Deep Learning Era: A Novel Taxonomy and Recent Advances5 Nov 2025
- 2025GauDP: Reinventing Multi-Agent Collaboration through Gaussian-Image Synergy in Diffusion Policies2 Nov 2025
- 2025Data-Efficient Multitask DAgger29 Sep 2025
- 2025GLUE: Global-Local Unified Encoding for Imitation Learning via Key-Patch Tracking27 Sep 2025
- 2025Learning to Evaluate Autonomous Behaviour in Human-Robot Interaction8 Jul 2025
- 2025PartInstruct: Part-level Instruction Following for Fine-grained Robot Manipulation27 May 2025
- 2025RoboTwin: Dual-Arm Robot Benchmark with Generative Digital Twins17 Apr 2025
- 2025Dexterous Manipulation through Imitation Learning: A Survey4 Apr 2025
- 2025Sim-and-Real Co-Training: A Simple Recipe for Vision-Based Robotic Manipulation31 Mar 2025
- 2025GraspCorrect: Robotic Grasp Correction via Vision-Language Model-Guided Feedback19 Mar 2025
- 2025CoinRobot: Generalized End-to-end Robotic Learning for Physical Intelligence7 Mar 2025
- 2025Adapt3R: Adaptive 3D Scene Representation for Domain Transfer in Imitation Learning6 Mar 2025
- 2025Robotic Compliant Object Prying Using Diffusion Policy Guided by Vision and Force Observations6 Mar 2025
- 2025A2Perf: Real-World Autonomous Agents Benchmark4 Mar 2025
- 2025Generative Models in Decision Making: A Survey24 Feb 2025
- 2025IMLE Policy: Fast and Sample Efficient Visuomotor Policy Learning via Implicit Maximum Likelihood Estimation17 Feb 2025
- 2025Temporal Representation Alignment: Successor Features Enable Emergent Compositionality in Robot Instruction Following8 Feb 2025
- 2025Horizon Generalization in Reinforcement Learning6 Jan 2025
- 2024Contractive Dynamical Imitation Policies for Efficient Out-of-Sample Recovery10 Dec 2024
- 2024P3-PO: Prescriptive Point Priors for Visuo-Spatial Generalization of Robot Policies9 Dec 2024
- 2024RoboMD: Uncovering Robot Vulnerabilities through Semantic Potential Fields3 Dec 2024
- 2024EgoMimic: Scaling Imitation Learning via Egocentric Video31 Oct 2024
- 2024DexMimicGen: Automated Data Generation for Bimanual Dexterous Manipulation via Imitation Learning31 Oct 2024
- 2024VLMimic: Vision Language Models are Visual Imitation Learner for Fine-grained Actions28 Oct 2024
- 2024SkillMimicGen: Automated Demonstration Generation for Efficient Skill Learning and Deployment24 Oct 2024
- 2024Learning from 10 Demos: Generalisable and Sample-Efficient Policy Learning with Oriented Affordance Frames15 Oct 2024
- 2024OKAMI: Teaching Humanoid Robots Manipulation Skills through Single Video Imitation15 Oct 2024
- 2024DeformPAM: Data-Efficient Learning for Long-horizon Deformable Object Manipulation via Preference-based Action Alignment15 Oct 2024
- 2024One-Shot Robust Imitation Learning for Long-Horizon Visuomotor Tasks from Unsegmented Demonstrations2 Oct 2024
- 2024Development and Validation of Heparin Dosing Policies Using an Offline Reinforcement Learning Algorithm24 Sep 2024
- 2024Generalized Robot Learning Framework18 Sep 2024
- 2024LEMMo-Plan: LLM-Enhanced Learning from Multi-Modal Demonstration for Planning Sequential Contact-Rich Manipulation Tasks18 Sep 2024
- 2024RoboTwin: Dual-Arm Robot Benchmark with Generative Digital Twins (early version)4 Sep 2024
- 2024Automating Deformable Gasket Assembly22 Aug 2024
- 2024Deep Generative Models in Robotics: A Survey on Learning from Multimodal Demonstrations8 Aug 2024
- 2024VITAL: Interactive Few-Shot Imitation Learning via Visual Human-in-the-Loop Corrections30 Jul 2024
50 newest of 114 All 114 (JSON)
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 699
- 2023MimicGen: A Data Generation System for Scalable Robot Learning using Human Demonstrationscited by 308
- 2023MimicPlay: Long-Horizon Imitation Learning by Watching Human Playcited by 214
- 2024EgoMimic: Scaling Imitation Learning via Egocentric Videocited by 160
- 2022VIOLA: Imitation Learning for Vision-Based Manipulation with Object Proposal Priorscited by 135
- 2024DexMimicGen: Automated Data Generation for Bimanual Dexterous Manipulation via Imitation Learningcited by 104
- 2020Human-in-the-Loop Imitation Learning using Remote Teleoperationcited by 103
- 2020COG: Connecting New Skills to Past Experience with Offline Reinforcement Learningcited by 83
- 2021Bottom-Up Skill Discovery from Unsegmented Demonstrations for Long-Horizon Robot Manipulationcited by 82
- 2022Latent Plans for Task-Agnostic Offline Reinforcement Learningcited by 81
- 2022Learning and Retrieval from Prior Data for Skill-based Imitation Learningcited by 77
12 most cited of 56 All 56 (JSON)
- 2026Vision-Language Grounded Task-Context-Aware Imitation Learning for Robotic Disassembly15 Sep 2026
- 2026AutoIntervene: Calibrated Intervention for Action-Chunking Imitation Learning Policies7 Aug 2026
- 2026Language-Critique Imitation Learning from Suboptimal Demonstrations1 Jul 2026
- 2026Offline Reinforcement Learning for Fluid Controls: Data-based Multi-observational Policy Extraction30 Jun 2026
- 2026When Life Gives You BC, Make Q-functions: Extracting Q-values from Behavior Cloning for On-Robot Reinforcement Learning6 May 2026
- 2026Optimizing Neurorobot Policy under Limited Demonstration Data through Preference Regret4 Apr 2026
- 2025Imitation Learning in the Deep Learning Era: A Novel Taxonomy and Recent Advances5 Nov 2025
- 2025GLUE: Global-Local Unified Encoding for Imitation Learning via Key-Patch Tracking27 Sep 2025
- 2025Learning to Evaluate Autonomous Behaviour in Human-Robot Interaction8 Jul 2025
- 2025Dexterous Manipulation through Imitation Learning: A Survey4 Apr 2025
- 2025Adapt3R: Adaptive 3D Scene Representation for Domain Transfer in Imitation Learning6 Mar 2025
- 2025A2Perf: Real-World Autonomous Agents Benchmark4 Mar 2025
12 newest of 56 All 56 (JSON)
- 2023Goal-Conditioned Imitation Learning using Score-based Diffusion Policiescited by 229
- 2023Policy Representation via Diffusion Probability Model for Reinforcement Learningcited by 105
- 2024CleanDiffuser: An Easy-to-use Modularized Library for Diffusion Models in Decision Makingcited by 38
- 2025IMLE Policy: Fast and Sample Efficient Visuomotor Policy Learning via Implicit Maximum Likelihood Estimationcited by 13
- 2025Robotic Compliant Object Prying Using Diffusion Policy Guided by Vision and Force Observationscited by 9
- 2023C3DM: Constrained-Context Conditional Diffusion Models for Imitation Learningcited by 7
- 2025GauDP: Reinventing Multi-Agent Collaboration through Gaussian-Image Synergy in Diffusion Policiescited by 2
- 2025PADiff: Predictive and Adaptive Diffusion Policies for Ad Hoc Teamworkcited by 0
- 2025PADiff: Predictive and Adaptive Diffusion Policies for Ad Hoc Teamwork10 Nov 2025
- 2025GauDP: Reinventing Multi-Agent Collaboration through Gaussian-Image Synergy in Diffusion Policies2 Nov 2025
- 2025Robotic Compliant Object Prying Using Diffusion Policy Guided by Vision and Force Observations6 Mar 2025
- 2025IMLE Policy: Fast and Sample Efficient Visuomotor Policy Learning via Implicit Maximum Likelihood Estimation17 Feb 2025
- 2024CleanDiffuser: An Easy-to-use Modularized Library for Diffusion Models in Decision Making13 Jun 2024
- 2023C3DM: Constrained-Context Conditional Diffusion Models for Imitation Learning2 Nov 2023
- 2023Policy Representation via Diffusion Probability Model for Reinforcement Learning22 May 2023
- 2023Goal-Conditioned Imitation Learning using Score-based Diffusion Policies5 Apr 2023
- 2024RoboCasa: Large-Scale Simulation of Everyday Tasks for Generalist Robotscited by 391
- 2025RoboTwin: Dual-Arm Robot Benchmark with Generative Digital Twinscited by 117
- 2022From Play to Policy: Conditional Behavior Generation from Uncurated Robot Datacited by 103
- 2022Robot Learning on the Job: Human-in-the-Loop Autonomy and Learning During Deploymentcited by 101
- 2021Learning by Watching: Physical Imitation of Manipulation Skills from Human Videoscited by 87
- 2022BITS: Bi-level Imitation for Traffic Simulationcited by 80
- 2022You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstrationcited by 79
- 2022Can Foundation Models Perform Zero-Shot Task Specification For Robot Manipulation?cited by 70
- 2021Understanding Domain Randomization for Sim-to-real Transfercited by 69
- 2024RoboTwin: Dual-Arm Robot Benchmark with Generative Digital Twins (early version)cited by 68
- 2024OKAMI: Teaching Humanoid Robots Manipulation Skills through Single Video Imitationcited by 62
- 2022Learning Neuro-Symbolic Skills for Bilevel Planningcited by 55
12 most cited of 41 All 41 (JSON)
- 2026Iterative Grasp Pose Refinement: A Deep Reinforcement Learning Approach for 2D Vision18 Aug 2026
- 2026GHOST: Hierarchical Sub-Goal Policies for Generalizing Robot Manipulation8 Jun 2026
- 2025From Power to Precision: Learning Fine-grained Dexterity for Multi-fingered Robotic Hands17 Nov 2025
- 2025Data-Efficient Multitask DAgger29 Sep 2025
- 2025PartInstruct: Part-level Instruction Following for Fine-grained Robot Manipulation27 May 2025
- 2025RoboTwin: Dual-Arm Robot Benchmark with Generative Digital Twins17 Apr 2025
- 2025Sim-and-Real Co-Training: A Simple Recipe for Vision-Based Robotic Manipulation31 Mar 2025
- 2025CoinRobot: Generalized End-to-end Robotic Learning for Physical Intelligence7 Mar 2025
- 2025Generative Models in Decision Making: A Survey24 Feb 2025
- 2025Temporal Representation Alignment: Successor Features Enable Emergent Compositionality in Robot Instruction Following8 Feb 2025
- 2025Horizon Generalization in Reinforcement Learning6 Jan 2025
- 2024P3-PO: Prescriptive Point Priors for Visuo-Spatial Generalization of Robot Policies9 Dec 2024
12 newest of 41 All 41 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2025Learning a Thousand Tasks in a Day
- 2021Towards More Generalizable One-shot Visual Imitation Learning
- 2019Relay Policy Learning: Solving Long-Horizon Tasks via Imitation and Reinforcement Learning
- 2022BC-Z: Zero-Shot Task Generalization with Robotic Imitation Learning
- 2024Offline Imitation Learning Through Graph Search and Retrieval
- 2025Generalization Capability for Imitation Learning
- 2026Beyond the Majority: Long-tail Imitation Learning for Robotic Manipulation
- 2020Scalable Multi-Task Imitation Learning with Autonomous Improvement