MILES: Making Imitation Learning Easy with Self-Supervision

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Papagiannis, Georgios, Johns, Edward
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929558611034112
author Papagiannis, Georgios
Johns, Edward
author_facet Papagiannis, Georgios
Johns, Edward
contents Data collection in imitation learning often requires significant, laborious human supervision, such as numerous demonstrations, and/or frequent environment resets for methods that incorporate reinforcement learning. In this work, we propose an alternative approach, MILES: a fully autonomous, self-supervised data collection paradigm, and we show that this enables efficient policy learning from just a single demonstration and a single environment reset. MILES autonomously learns a policy for returning to and then following the single demonstration, whilst being self-guided during data collection, eliminating the need for additional human interventions. We evaluated MILES across several real-world tasks, including tasks that require precise contact-rich manipulation such as locking a lock with a key. We found that, under the constraints of a single demonstration and no repeated environment resetting, MILES significantly outperforms state-of-the-art alternatives like imitation learning methods that leverage reinforcement learning. Videos of our experiments and code can be found on our webpage: www.robot-learning.uk/miles.
format Preprint
id arxiv_https___arxiv_org_abs_2410_19693
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MILES: Making Imitation Learning Easy with Self-Supervision
Papagiannis, Georgios
Johns, Edward
Robotics
Artificial Intelligence
Machine Learning
Data collection in imitation learning often requires significant, laborious human supervision, such as numerous demonstrations, and/or frequent environment resets for methods that incorporate reinforcement learning. In this work, we propose an alternative approach, MILES: a fully autonomous, self-supervised data collection paradigm, and we show that this enables efficient policy learning from just a single demonstration and a single environment reset. MILES autonomously learns a policy for returning to and then following the single demonstration, whilst being self-guided during data collection, eliminating the need for additional human interventions. We evaluated MILES across several real-world tasks, including tasks that require precise contact-rich manipulation such as locking a lock with a key. We found that, under the constraints of a single demonstration and no repeated environment resetting, MILES significantly outperforms state-of-the-art alternatives like imitation learning methods that leverage reinforcement learning. Videos of our experiments and code can be found on our webpage: www.robot-learning.uk/miles.
title MILES: Making Imitation Learning Easy with Self-Supervision
topic Robotics
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2410.19693