Language-Conditioned Imitation Learning with Base Skill Priors under Unstructured Data

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhou, Hongkuan, Bing, Zhenshan, Yao, Xiangtong, Su, Xiaojie, Yang, Chenguang, Huang, Kai, Knoll, Alois
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912023265148928
author Zhou, Hongkuan
Bing, Zhenshan
Yao, Xiangtong
Su, Xiaojie
Yang, Chenguang
Huang, Kai
Knoll, Alois
author_facet Zhou, Hongkuan
Bing, Zhenshan
Yao, Xiangtong
Su, Xiaojie
Yang, Chenguang
Huang, Kai
Knoll, Alois
contents The growing interest in language-conditioned robot manipulation aims to develop robots capable of understanding and executing complex tasks, with the objective of enabling robots to interpret language commands and manipulate objects accordingly. While language-conditioned approaches demonstrate impressive capabilities for addressing tasks in familiar environments, they encounter limitations in adapting to unfamiliar environment settings. In this study, we propose a general-purpose, language-conditioned approach that combines base skill priors and imitation learning under unstructured data to enhance the algorithm's generalization in adapting to unfamiliar environments. We assess our model's performance in both simulated and real-world environments using a zero-shot setting. In the simulated environment, the proposed approach surpasses previously reported scores for CALVIN benchmark, especially in the challenging Zero-Shot Multi-Environment setting. The average completed task length, indicating the average number of tasks the agent can continuously complete, improves more than 2.5 times compared to the state-of-the-art method HULC. In addition, we conduct a zero-shot evaluation of our policy in a real-world setting, following training exclusively in simulated environments without additional specific adaptations. In this evaluation, we set up ten tasks and achieved an average 30% improvement in our approach compared to the current state-of-the-art approach, demonstrating a high generalization capability in both simulated environments and the real world. For further details, including access to our code and videos, please refer to https://hk-zh.github.io/spil/
format Preprint
id arxiv_https___arxiv_org_abs_2305_19075
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Language-Conditioned Imitation Learning with Base Skill Priors under Unstructured Data
Zhou, Hongkuan
Bing, Zhenshan
Yao, Xiangtong
Su, Xiaojie
Yang, Chenguang
Huang, Kai
Knoll, Alois
Robotics
Artificial Intelligence
The growing interest in language-conditioned robot manipulation aims to develop robots capable of understanding and executing complex tasks, with the objective of enabling robots to interpret language commands and manipulate objects accordingly. While language-conditioned approaches demonstrate impressive capabilities for addressing tasks in familiar environments, they encounter limitations in adapting to unfamiliar environment settings. In this study, we propose a general-purpose, language-conditioned approach that combines base skill priors and imitation learning under unstructured data to enhance the algorithm's generalization in adapting to unfamiliar environments. We assess our model's performance in both simulated and real-world environments using a zero-shot setting. In the simulated environment, the proposed approach surpasses previously reported scores for CALVIN benchmark, especially in the challenging Zero-Shot Multi-Environment setting. The average completed task length, indicating the average number of tasks the agent can continuously complete, improves more than 2.5 times compared to the state-of-the-art method HULC. In addition, we conduct a zero-shot evaluation of our policy in a real-world setting, following training exclusively in simulated environments without additional specific adaptations. In this evaluation, we set up ten tasks and achieved an average 30% improvement in our approach compared to the current state-of-the-art approach, demonstrating a high generalization capability in both simulated environments and the real world. For further details, including access to our code and videos, please refer to https://hk-zh.github.io/spil/
title Language-Conditioned Imitation Learning with Base Skill Priors under Unstructured Data
topic Robotics
Artificial Intelligence
url https://arxiv.org/abs/2305.19075