Learning to Generalize Across Long-Horizon Tasks from Human Demonstrations
Open accessAbstract
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/ .
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
The tasks MimicGen was demonstrated on have pose variation well beyond typical prior works using BC from human demos.
“The tasks also have pose variation well beyond typical prior works using BC from human demos [1,3-7,30,33,55,56].”
From MimicGen: A Data Generation System for Scalable Robot Learning using Human Demonstrations (2023) · cites this paper as [33]
The Real World Robot Manipulation Experiments section (p. 7) describes two starting configurations for the PandaKitchen task, but does not quantify pose variation or compare it with prior behavioral-cloning work. Searches for “pose variation” and varying object positions found no passage establishing that comparison. (LLM, conf=0.73)
Experimental — accuracy not yet measured
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
Auto-Encoding Variational Bayes
Kingma, D. P., & Welling, M. (2013) · arXiv · doi:10.48550/arxiv.1312.6114 · Cited by 20,067 · see citing works
Mastering the game of Go with deep neural networks and tree search
Silver, D., Huang, A., Maddison, C. J., Guez, A., Sifre, L., van den Driessche, G., Schrittwieser, J., Antonoglou, I., Panneershelvam, V., Lanctot, M., et al. (2016) · Nature · doi:10.1038/nature16961 · Cited by 15,612 · see citing works
Khatib, O. (1987) · IEEE J. Robotics Autom. · doi:10.1109/jra.1987.1087068 · Cited by 3,233 · see citing works
End to End Learning for Self-Driving Cars
Bojarski, M., Del Testa, D., Dworakowski, D., Firner, B., Flepp, B., Goyal, P., Jackel, L. D., Monfort, M., Muller, U., Zhang, J., et al. (2016) · arXiv · doi:10.48550/arxiv.1604.07316 · Cited by 3,198 · see citing works
Between MDPs and semi-MDPs: A framework for temporal abstraction in reinforcement learning
Sutton, R. S., Precup, D., & Singh, S. (1999) · Artificial Intelligence · doi:10.1016/s0004-3702(99)00052-1 · Cited by 3,119 · see citing works
A Reduction of Imitation Learning and Structured Prediction to No-Regret Online Learning
Ross, S., Gordon, G. J., & Bagnell, J. A. (2018) · Carnegie Mellon University · doi:10.1184/r1/6550949 · Cited by 1,941 · see citing works
ALVINN, an autonomous land vehicle in a neural network
Pomerleau, D. A. (2018) · doi:10.1184/r1/6603146 · Cited by 1,403 · see citing works
Is imitation learning the route to humanoid robots?
Schaal, S. (1999) · Trends in Cognitive Sciences · doi:10.1016/s1364-6613(99)01327-3 · Cited by 1,325 · see citing works
Movement imitation with nonlinear dynamical systems in humanoid robots
Ijspeert, A. J., Nakanishi, J., & Stefan Schaal, A. (2003) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/robot.2002.1014739 · Cited by 862 · see citing works
Learning and Reproduction of Gestures by Imitation
Calinon, S., D'halluin, F., Sauser, E. L., Caldwell, D. G., & Billard, A. (2010) · IEEE Robotics & Automation Magazine · doi:10.1109/mra.2010.936947 · Cited by 461 · see citing works
One-Shot Imitation LearningPreprint
Duan, Y., Andrychowicz, M., Stadie, B. C., Ho, O. J., Schneider, J., Sutskever, I., Abbeel, P., & Zaremba, W. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1703.07326 · Cited by 449 · see citing works
Deep spatial autoencoders for visuomotor learning
Finn, C., Tan, X., Duan, Y., Darrell, T., Levine, S., & Abbeel, P. (2016) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/icra.2016.7487173 · Cited by 446 · see citing works
A framework for behavioural cloning
Bain, M., & Sammut, C. (2000) · doi:10.1093/oso/9780198538677.003.0006 · Cited by 370 · see citing works
One-Shot Visual Imitation Learning via Meta-Learning
Finn, C., Yu, T., Zhang, T., Abbeel, P., & Levine, S. (2017) · arXiv · doi:10.48550/arxiv.1709.04905 · Cited by 341 · see citing works
Learning motor primitives for robotics
Kober, J., & Peters, J. (2009) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/robot.2009.5152577 · Cited by 226 · see citing works
The Option-Critic Architecture
Bacon, P., Harb, J., & Precup, D. (2017) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v31i1.10916 · Cited by 219 · see citing works
Learning to Optimize Join Queries With Deep Reinforcement LearningPreprint
Krishnan, S., Yang, Z., Goldberg, K., Hellerstein, J. M., & Stoica, I. G. (2018) · arXiv (Cornell University) · doi:10.48550/arxiv.1808.03196 · Cited by 201 · see citing works
Learning Latent Plans from PlayPreprint
Lynch, C., Khansari, M., Xiao, T. G., Kumar, V., Tompson, J., Levine, S., & Sermanet, P. (2019) · arXiv (Cornell University) · doi:10.48550/arxiv.1903.01973 · Cited by 195 · see citing works
Pathak, D., Mahmoudieh, P., Luo, G., Agrawal, P., Chen, D., Shentu, F., Shelhamer, E., Malik, J., Efros, A. A., & Darrell, T. (2018) · IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW) · doi:10.1109/cvprw.2018.00278 · Cited by 188 · see citing works
Neural Task Programming: Learning to Generalize Across Hierarchical Tasks
Xu, D., Nair, S., Zhu, Y., Gao, J., Garg, A., Fei-Fei, L., & Savarese, S. (2018) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/icra.2018.8460689 · Cited by 164 · see citing works
Self-citationHierarchical Relative Entropy Policy Search
Daniel, C., Neumann, G. M., Kroemer, O., & Peters, J. (2014) · Lincoln Repository (University of Lincoln) · Cited by 160 · see citing works
ROBOTURK: A Crowdsourcing Platform for Robotic Skill Learning through Imitation
Mandlekar, A., Zhu, Y., Garg, A., Booher, J., Spero, M., Tung, A., Gao, J., Emmons, J., Gupta, A., Orbay, E., et al. (2018) · arXiv · doi:10.48550/arxiv.1811.02790 · Cited by 134 · see citing works
Self-citationImitation and Reinforcement Learning
Kober, J., & Peters, J. (2010) · IEEE Robotics & Automation Magazine · doi:10.1109/mra.2010.936952 · Cited by 132 · see citing works
End-to-End Driving Via Conditional Imitation Learning
Codevilla, F., Muller, M., Lopez, A., Koltun, V., & Dosovitskiy, A. (2018) · 2018 IEEE International Conference on Robotics and Automation (ICRA) · doi:10.1109/icra.2018.8460487 · Cited by 129 · see citing works
Robust Imitation of Diverse BehaviorsPreprint
Wang, Z., Merel, J., Reed, S., Wayne, G., de Freitas, N., & Heess, N. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1707.02747 · Cited by 125 · see citing works
Neural Task Graphs: Generalizing to Unseen Tasks From a Single Video Demonstration
Huang, D.-A., Nair, S., Xu, D., Zhu, Y., Garg, A., Fei-Fei, L., Savarese, S., & Niebles, J. C. (2019) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr.2019.00876 · Cited by 122 · see citing works
Self-citationAVID: Learning Multi-Stage Tasks via Pixel-Level Translation of Human Videos
Smith, L., Dhawan, N., Zhang, M., Abbeel, P., & Levine, S. (2020) · Robotics: Science and Systems XVI · doi:10.15607/rss.2020.xvi.024 · Cited by 102 · see citing works
Mandlekar, A., Ramos, F. T., Boots, B., Savarese, S., Fei-Fei, L., Garg, A., & Fox, D. (2020) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/icra40945.2020.9196935 · Cited by 79 · see citing works
Self-citationDeep Local Trajectory Replanning and Control for Robot Navigation
Pokle, A., Martín-Martín, R., Goebel, P., Chow, V., Ewald, H. M., Yang, J., Wang, Z., Sadeghian, A., Sadigh, D., Savarese, S., et al. (2019) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/icra.2019.8794062 · Cited by 72 · see citing works
Self-citationStochastic Neural Networks for Hierarchical Reinforcement Learning
Florensa, C., Duan, Y., & Abbeel, P. (2016) · International Conference on Learning Representations · Cited by 57 · see citing works
Learning manipulation skills from a single demonstration
Englert, P. A. J., & Toussaint, M. (2017) · The International Journal of Robotics Research · doi:10.1177/0278364917743795 · Cited by 49 · see citing works
Mandlekar, A., Booher, J., Spero, M., Tung, A., Gupta, A., Zhu, Y., Garg, A., Savarese, S., & Fei-Fei, L. (2019) · arXiv · doi:10.48550/arxiv.1911.04052 · Cited by 48 · see citing works
Self-citationMandlekar, A., Booher, J., Spero, M., Tung, A. T. Y., Gupta, A., Zhu, Y., Garg, A., Savarese, S., & Fei-Fei, L. (2019) · IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS) · doi:10.1109/iros40897.2019.8968114 · Cited by 47 · see citing works
Self-citationLearning Deep Parameterized Skills from Demonstration for Re-targetable Visuomotor ControlPreprint
Chang, J., Kumar, N., Hastings, S., Gokaslan, A., Romeres, D., Jha, D. K., Nikovski, D. N., Konidaris, G., & Tellex, S. (2019) · arXiv (Cornell University) · doi:10.48550/arxiv.1910.10628 · Cited by 4 · see citing works
Flagged first, then by citations
2 of the 36 references couldn’t be matched to a known paper.
Research works printed but not linked: 8
- [3][3] Aude Billard, Sylvain Calinon, R¨ udiger Dillmann, and Stefan Schaal. Robot programming by demonstration. In Springer Handbook of Robotics , 2008. -
- [7][7] John Co-Reyes, YuXuan Liu, Abhishek Gupta, Benjamin Eysenbach, Pieter Abbeel, and Sergey Levine. Selfconsistent trajectory autoencoder: Hierarchical reinforcement learning with trajectory embeddings. In Proceedings of the 35th International Conference on Machine Learning , volume 80 of Proceedings of Machine Learning Research , pages 1009-1018, Stockholmsm¨ assan, Stockholm Sweden, 10-15 Jul 2018. PMLR. -
- [12][12] Benjamin Eysenbach, Abhishek Gupta, Julian Ibarz, and Sergey Levine. Diversity is all you need: Learning diverse skills without a reward function. In International Conference of Learning Representations , 2019. -
- [18][18] Dinesh Jayaraman, Frederik Ebert, Alexei A Efros, and Sergey Levine. Time-agnostic prediction: Predicting predictable video frames. In International Conference of Learning Representations , 2019. -
- [21][21] Thomas Kipf, Yujia Li, Hanjun Dai, Vinicius Zambaldi, Alvaro Sanchez-Gonzalez, Edward Grefenstette, Pushmeet Kohli, and Peter Battaglia. Compile: Compositional imitation learning and execution. In International Conference on Machine Learning (ICML) , 2019. -
- [26][26] Corey Lynch, Mohi Khansari, Ted Xiao, Vikash Kumar, Jonathan Tompson, Sergey Levine, and Pierre Sermanet. Learning latent plans from play. In Conference on Robot Learning , 2019. -
- [33][33] St´ ephane Ross, Geoffrey J. Gordon, and J. Andrew Bagnell. A reduction of imitation learning and structured prediction to no-regret online learning. In AISTATS , 2010. -
- [35][35] Stuart Russell. Learning agents for uncertain environments. In Proceedings of the eleventh annual conference on Computational learning theory , pages 101-103, 1998. -
Cited by
Sorted by citations
Diffusion policy: Visuomotor policy learning via action diffusion
Chi, C., Xu, Z., Feng, S., Cousineau, E., Du, Y., Burchfiel, B., Tedrake, R., & Song, S. (2025) · The International Journal of Robotics Research · doi:10.1177/02783649241273668 · Cited by 613 · see citing works
Diffusion Policy: Visuomotor Policy Learning via Action Diffusion
Chi, C., Feng, S., Du, Y., Xu, Z., Cousineau, E., Burchfiel, B., & Song, S. (2023) · Robotics: Science and Systems XIX · doi:10.15607/rss.2023.xix.026 · Cited by 530 · see citing works
Sim2real transfer for reinforcement learning without dynamics randomization
Kaspar, M., Munoz Osorio, J. D., & Bock, J. (2020) · 2020 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS) · doi:10.1109/iros45743.2020.9341260 · Cited by 87 · see citing works
You Only Demonstrate Once: Category-Level Manipulation from Single Visual Demonstration
Wen, B., Lian, W., Bekris, K., & Schaal, S. (2022) · Robotics: Science and Systems XVIII · doi:10.15607/rss.2022.xviii.044 · Cited by 70 · see citing works
Learning by Watching: Physical Imitation of Manipulation Skills from Human Videos
Xiong, H., Li, Q., Chen, Y.-C., Bharadhwaj, H., Sinha, S., & Garg, A. (2021) · IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS) · doi:10.1109/iros51168.2021.9636080 · Cited by 49 · see citing works
A Comprehensive Review of Humanoid Robots
Sheng, Q., Zhou, Z., Li, J., Mi, X., Xiang, P., Chen, Z., Xu, H., Jia, S., Wu, X., Cui, Y., et al. (2025) · SmartBot · doi:10.1002/smb2.12008 · Cited by 51 · see citing works
Autonomous reinforcement learning agent for chemical vapor deposition synthesis of quantum materials
Rajak, P., Krishnamoorthy, A., Mishra, A., Kalia, R. K., Nakano, A., & Vashishta, P. D. (2021) · npj Computational Materials · doi:10.1038/s41524-021-00535-3 · Cited by 46 · see citing works
COG: Connecting New Skills to Past Experience with Offline Reinforcement LearningPreprint
Singh, A., Yu, A., Yang, J., Zhang, J. M., Kumar, A., & Levine, S. (2020) · arXiv (Cornell University) · doi:10.48550/arxiv.2010.14500 · Cited by 48 · see citing works
Human-in-the-Loop Imitation Learning using Remote Teleoperation
Mandlekar, A., Xu, D., Martín-Martín, R., Zhu, Y., Fei-Fei, L., & Savarese, S. (2020) · arXiv · doi:10.48550/arxiv.2012.06733 · Cited by 35 · see citing works
Self-citationA survey of demonstration learning
Correia, A., & Alexandre, L. A. (2024) · Robotics and Autonomous Systems · doi:10.1016/j.robot.2024.104812 · Cited by 34 · see citing works
Robot Learning on the Job: Human-in-the-Loop Autonomy and Learning During Deployment
Liu, H., Nasiriany, S., Zhang, L., Bao, Z., & Zhu, Y. (2023) · Robotics: Science and Systems · doi:10.15607/rss.2023.xix.005 · Cited by 30 · see citing works
Wang, C., Wang, R., Mandlekar, A., Fei-Fei, L., Savarese, S., & Xu, D. (2021) · IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS) · doi:10.1109/iros51168.2021.9636023 · Cited by 23 · see citing works
Self-citationDiffusion Policy: Visuomotor Policy Learning via Action Diffusion
Chi, C., Xu, Z., Feng, S., Cousineau, E., Du, Y., Burchfiel, B., Tedrake, R., & Song, S. (2023) · arXiv · doi:10.48550/arxiv.2303.04137 · Cited by 12 · see citing works
PRIME: Scaffolding Manipulation Tasks With Behavior Primitives for Data-Efficient Imitation Learning
Gao, T. Z., Nasiriany, S., Liu, H., Yang, Q., & Zhu, Y. (2024) · IEEE Robotics and Automation Letters · doi:10.1109/lra.2024.3443610 · Cited by 13 · see citing works
Manipulator-Independent Representations for Visual Imitation
Zhou, Y., Aytar, Y., & Bousmalis, K. (2021) · Robotics: Science and Systems · doi:10.15607/rss.2021.xvii.002 · Cited by 11 · see citing works
A Survey of Demonstration LearningPreprint
Correia, A. R. D. S. P., & Alexandre, L. A. (2023) · Robotics Auton. Syst. · doi:10.2139/ssrn.4390650 · Cited by 56 · see citing works
A survey on learning an autonomous dynamic system for human–robot skills transfer from demonstration
Fu, J., Huang, H., Jin, Z., Liu, A., Zhang, W.-A., Yu, L. J., Si, W., & Yang, C. (2025) · Robotics and Computer-Integrated Manufacturing · doi:10.1016/j.rcim.2025.103092 · Cited by 6 · see citing works
MimicGen: A Data Generation System for Scalable Robot Learning using Human Demonstrations
Mandlekar, A., Nasiriany, S., Wen, B., Akinola, I., Narang, Y., Fan, L., Zhu, Y., & Fox, D. (2023) · arXiv · doi:10.48550/arxiv.2310.17596 · Cited by 5 · see citing works
Self-citationOne-Shot Transfer of Long-Horizon Extrinsic Manipulation Through Contact Retargeting
Wu, A., Wang, R., Chen, S., Eppner, C., & Liu, C. K. (2024) · IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS) · doi:10.1109/iros58592.2024.10801356 · Cited by 3 · see citing works
Tsinganos, K., Chatzilygeroudis, K. I., Hadjivelichkov, D., Komninos, T., Dermatas, E., & Kanoulas, D. (2022) · Frontiers in Robotics and AI · doi:10.3389/frobt.2022.974537 · Cited by 2 · see citing works
Compositional Servoing by Recombining Demonstrations
Argus, M. J., Nayak, A., Büchner, M., Galesso, S., Valada, A., & Brox, T. (2024) · Proceedings - IEEE International Conference on Robotics and Automation/Proceedings · doi:10.1109/icra57147.2024.10609978 · Cited by 1 · see citing works
- Authors
- Mandlekar, A., Xu, D., Martín-Martín, R., Savarese, S., & Fei-Fei, L.
- Affiliations
- Stanford University (US)
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2020
- ISSN
- 2331-8422 (ISSN-L)
- arXiv
- 2003.06085
- OpenAlex
- W3010709437
- Cited by
- 17
- References
- 36
- Citations by year
- 2025: 2 · 2024: 4 · 2023: 3 · 2022: 2 · 2021: 4 · 2020: 2
- Type
- posted content
- Language
- en
- Open access
- green
- arxiv.org
- Locations
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Topics
- Physical Sciences › Computer Science › Artificial Intelligence › Reinforcement Learning in Robotics
- Physical Sciences › Engineering › Control and Systems Engineering › Robot Manipulation and Learning
- Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
- Keywords
- imitation learning, long-horizon tasks, trajectory intersection, multi-task learning, stochastic policies, robotic manipulation, offline learning, behavior composition
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review