OpenVLA: An Open-Source Vision-Language-Action Model
FeaturedMoo Jin Kim, Karl Pertsch, Siddharth Karamcheti, Ted Xiao, Ashwin Balakrishna, Suraj Nair, Rafael Rafailov, Erin Foster, Grace Lam, Pannag Sanketi, Quan Vuong, Thomas Kollar, Benjamin Burchfiel, Russ Tedrake, Dorsa Sadigh, Sergey Levine, Peng Xu, Chelsea Finn · Stanford, UC Berkeley, Toyota Research Institute, Google DeepMind, MIT · 2024
Framework
PyTorch
License
Apache-2.0
Stars
6,260
Summary
A 7B-parameter open-source Vision-Language-Action model pre-trained on 970k real-world robot demonstrations, achieving strong generalization across robots and tasks.
Abstract Summary
Key Points
- First fully open-source 7B VLA model with publicly available weights and training code.
- Pre-trained on 970k real robot trajectories from the Open X-Embodiment dataset (22 robots).
- Uses Llama 2 + DINOv2/SigLIP vision encoder; actions quantized into discrete tokens.
- Strong few-shot transfer: fine-tune on ~10-100 demos with LoRA for a new robot.
- Supports natural language task specification and multi-step long-horizon tasks.
The OpenVLA Model
We introduce the OpenVLA model, a 7B-parameter vision-language-action model (VLA) trained on 970k robot demonstrations from the Open X-Embodiment dataset [ 1 ] . There are many, largely unexplored, questions around best practices for developing VLA models, e.g., what are the best model backbones, datasets, and hyperparameters to use for training.
Below, we detail our approach for developing OpenVLA and summarize our key learnings.
The OpenVLA Codebase
Along with our model, we release the OpenVLA codebase, a modular PyTorch codebase for training VLA models (see https://openvla.github.io ).
It scales from fine-tuning VLAs on individual GPUs to training billion-parameter VLAs on multi-node GPU clusters, and supports modern techniques for large transformer model training such as automatic mixed precision (AMP, PyTorch [75] ), FlashAttention [ 76 ] , and fully sharded data parallelism (FSDP, Zhao et al. [77] ).
Experiments
The goal of our experimental evaluations is to test OpenVLA’s ability to serve as a powerful multi-robot control policy out of the box, as well as be a good initialization for fine-tuning to new robot tasks.
Sources and Demonstrations
Figures are reproduced from Kim et al. See the full paper for experimental details and the project page, when the paper links one, for demonstrations and videos.
Related Papers
DynaFLIP: Rethinking Robotics Perception via Tri-Modal-Dynamics Guided Representation
Jusuk Lee, Seungjae Lee, Jonghun Shin et al.
Robot manipulation critically depends on perception that preserves the action-relevant aspects of a scene. Yet most robot learning pipelines are built upon visual encoders pre-trained for static recog
GuidedVLA: Specifying Task-Relevant Factors via Plug-and-Play Action Attention Specialization
Xiaosong Jia, Bowen Yang, Zuhao Ge et al.
Vision-Language-Action (VLA) models aim for general robot learning by aligning action as a modality within powerful Vision-Language Models (VLMs). Existing VLAs rely on end-to-end supervision to implicitly enable the action decoding process to learn task-relevant features.
From Imagined Futures to Executable Actions: Mixture of Latent Actions for Robot Manipulation
Yajie Li, Bozhou Zhang, Chun Gu et al.
Video foundation-models models offer a promising imagination mechanism for robot manipulation by predicting long-horizon future observations, but effectively exploiting these imagined futures for action execution remains challenging.
Qwen-VLA: Unifying Vision-Language-Action Modeling across Tasks, Environments, and Robot Embodiments
Qiuyue Wang, Mingsheng Li, Jian Guan et al.
Embodied intelligence is often studied through specialized models for individual tasks such as manipulation or navigation, resulting in fragmented capabilities and limited generalization across tasks,