KALIE: Fine-Tuning Vision-Language Models for Open-World Manipulation without Robot Data

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Tang, Grace, Rajkumar, Swetha, Zhou, Yifei, Walke, Homer Rich, Levine, Sergey, Fang, Kuan
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913512322760704
author Tang, Grace
Rajkumar, Swetha
Zhou, Yifei
Walke, Homer Rich
Levine, Sergey
Fang, Kuan
author_facet Tang, Grace
Rajkumar, Swetha
Zhou, Yifei
Walke, Homer Rich
Levine, Sergey
Fang, Kuan
contents Building generalist robotic systems involves effectively endowing robots with the capabilities to handle novel objects in an open-world setting. Inspired by the advances of large pre-trained models, we propose Keypoint Affordance Learning from Imagined Environments (KALIE), which adapts pre-trained Vision Language Models (VLMs) for robotic control in a scalable manner. Instead of directly producing motor commands, KALIE controls the robot by predicting point-based affordance representations based on natural language instructions and visual observations of the scene. The VLM is trained on 2D images with affordances labeled by humans, bypassing the need for training data collected on robotic systems. Through an affordance-aware data synthesis pipeline, KALIE automatically creates massive high-quality training data based on limited example data manually collected by humans. We demonstrate that KALIE can learn to robustly solve new manipulation tasks with unseen objects given only 50 example data points. Compared to baselines using pre-trained VLMs, our approach consistently achieves superior performance.
format Preprint
id arxiv_https___arxiv_org_abs_2409_14066
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle KALIE: Fine-Tuning Vision-Language Models for Open-World Manipulation without Robot Data
Tang, Grace
Rajkumar, Swetha
Zhou, Yifei
Walke, Homer Rich
Levine, Sergey
Fang, Kuan
Robotics
Artificial Intelligence
Machine Learning
Building generalist robotic systems involves effectively endowing robots with the capabilities to handle novel objects in an open-world setting. Inspired by the advances of large pre-trained models, we propose Keypoint Affordance Learning from Imagined Environments (KALIE), which adapts pre-trained Vision Language Models (VLMs) for robotic control in a scalable manner. Instead of directly producing motor commands, KALIE controls the robot by predicting point-based affordance representations based on natural language instructions and visual observations of the scene. The VLM is trained on 2D images with affordances labeled by humans, bypassing the need for training data collected on robotic systems. Through an affordance-aware data synthesis pipeline, KALIE automatically creates massive high-quality training data based on limited example data manually collected by humans. We demonstrate that KALIE can learn to robustly solve new manipulation tasks with unseen objects given only 50 example data points. Compared to baselines using pre-trained VLMs, our approach consistently achieves superior performance.
title KALIE: Fine-Tuning Vision-Language Models for Open-World Manipulation without Robot Data
topic Robotics
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2409.14066