TinyVLA: Towards Fast, Data-Efficient Vision-Language-Action Models for Robotic Manipulation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wen, Junjie, Zhu, Yichen, Li, Jinming, Zhu, Minjie, Wu, Kun, Xu, Zhiyuan, Liu, Ning, Cheng, Ran, Shen, Chaomin, Peng, Yaxin, Feng, Feifei, Tang, Jian
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910939695022080
author Wen, Junjie
Zhu, Yichen
Li, Jinming
Zhu, Minjie
Wu, Kun
Xu, Zhiyuan
Liu, Ning
Cheng, Ran
Shen, Chaomin
Peng, Yaxin
Feng, Feifei
Tang, Jian
author_facet Wen, Junjie
Zhu, Yichen
Li, Jinming
Zhu, Minjie
Wu, Kun
Xu, Zhiyuan
Liu, Ning
Cheng, Ran
Shen, Chaomin
Peng, Yaxin
Feng, Feifei
Tang, Jian
contents Vision-Language-Action (VLA) models have shown remarkable potential in visuomotor control and instruction comprehension through end-to-end learning processes. However, current VLA models face significant challenges: they are slow during inference and require extensive pre-training on large amounts of robotic data, making real-world deployment difficult. In this paper, we introduce a new family of compact vision-language-action models, called TinyVLA, which offers two key advantages over existing VLA models: (1) faster inference speeds, and (2) improved data efficiency, eliminating the need for pre-training stage. Our framework incorporates two essential components to build TinyVLA: (1) initializing the policy backbone with robust, high-speed multimodal models, and (2) integrating a diffusion policy decoder during fine-tuning to enable precise robot actions. We conducted extensive evaluations of TinyVLA in both simulation and on real robots, demonstrating that our approach significantly outperforms the state-of-the-art VLA model, OpenVLA, in terms of speed and data efficiency, while delivering comparable or superior performance. Additionally, TinyVLA exhibits strong generalization capabilities across various dimensions, including language instructions, novel objects, unseen positions, changes in object appearance, background variations, and environmental shifts, often matching or exceeding the performance of OpenVLA. We believe that \methodname offers an interesting perspective on utilizing pre-trained multimodal models for policy learning. Our project is at https://tiny-vla.github.io.
format Preprint
id arxiv_https___arxiv_org_abs_2409_12514
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TinyVLA: Towards Fast, Data-Efficient Vision-Language-Action Models for Robotic Manipulation
Wen, Junjie
Zhu, Yichen
Li, Jinming
Zhu, Minjie
Wu, Kun
Xu, Zhiyuan
Liu, Ning
Cheng, Ran
Shen, Chaomin
Peng, Yaxin
Feng, Feifei
Tang, Jian
Robotics
Computer Vision and Pattern Recognition
Vision-Language-Action (VLA) models have shown remarkable potential in visuomotor control and instruction comprehension through end-to-end learning processes. However, current VLA models face significant challenges: they are slow during inference and require extensive pre-training on large amounts of robotic data, making real-world deployment difficult. In this paper, we introduce a new family of compact vision-language-action models, called TinyVLA, which offers two key advantages over existing VLA models: (1) faster inference speeds, and (2) improved data efficiency, eliminating the need for pre-training stage. Our framework incorporates two essential components to build TinyVLA: (1) initializing the policy backbone with robust, high-speed multimodal models, and (2) integrating a diffusion policy decoder during fine-tuning to enable precise robot actions. We conducted extensive evaluations of TinyVLA in both simulation and on real robots, demonstrating that our approach significantly outperforms the state-of-the-art VLA model, OpenVLA, in terms of speed and data efficiency, while delivering comparable or superior performance. Additionally, TinyVLA exhibits strong generalization capabilities across various dimensions, including language instructions, novel objects, unseen positions, changes in object appearance, background variations, and environmental shifts, often matching or exceeding the performance of OpenVLA. We believe that \methodname offers an interesting perspective on utilizing pre-trained multimodal models for policy learning. Our project is at https://tiny-vla.github.io.
title TinyVLA: Towards Fast, Data-Efficient Vision-Language-Action Models for Robotic Manipulation
topic Robotics
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.12514