Octo: An Open-Source Generalist Robot Policy
FeaturedDibya Ghosh, Homer Walke, Karl Pertsch, Kevin Black, Oier Mees, Sudeep Dasari, Joey Hejna, Tobias Kreiman, Charles Xu, Jianlan Luo, You Liang Tan, Lawrence Yunliang Chen, Pannag Sanketi, Quan Vuong, Ted Xiao, Dorsa Sadigh, Chelsea Finn, Sergey Levine · UC Berkeley, Stanford, TRI · 2024
Framework
JAX / Flax
License
MIT
Stars
1,652
Summary
Octo is a large open-source transformer-based generalist robot policy trained on 800k trajectories, supporting language-conditioned and goal-image-conditioned control across 9 robotic platforms with efficient fine-tuning on consumer GPUs.
Abstract Summary
Key Points
- Large open-source generalist policy trained on 800k trajectories from Open X-Embodiment.
- Supports language and goal-image conditioning with a flexible transformer architecture.
- Observation-action tokenizer enables cross-robot fine-tuning on consumer GPUs within hours.
- Evaluated on 9 distinct robot platforms with strong few-shot transfer to new tasks.
- Full training and fine-tuning stack released in JAX/Flax with Hugging Face integration.
The Octo Model
In this section, we describe the Octo model, our open-source generalist robot policy that can be adapted to new robots and tasks — including new sensory inputs and action spaces — via finetuning. We discuss the key design decisions, training objectives, training dataset, and infrastructure.
The design of the Octo model emphasizes flexibility and scale: it supports a variety of commonly used robots, sensor configurations, and actions while providing a generic and scalable recipe that can be trained on large amounts of data.
Experiments
Our experiments provide an empirical analysis of Octo, evaluating its ability to serve as a general robotic foundation model across several axes: Can Octo control multiple robot embodiments and solve language and goal tasks out of the box?
Do Octo weights serve as a good initialization for data-efficient finetuning to new tasks and robots, and does it improve over training from scratch and commonly used pretrained representations? Which design decisions in Octo matter most for building generalist robot policies? We evaluate Octo’s capabilities...
Sources and Demonstrations
Figures are reproduced from Ghosh et al. See the full paper for experimental details and the project page, when the paper links one, for demonstrations and videos.
Related Papers
X-Imitator: Spatial-Aware Imitation Learning via Bidirectional Action-Pose Interaction
Kai Xiong, Hongjie Fang, Lixin Yang et al.
Effectively handling the interplay between spatial perception and action generation remains a critical bottleneck in robotic manipulation.
DROID: A Large-Scale In-the-Wild Robot Manipulation Dataset
Alexander Khazatsky, Karl Pertsch, Suraj Nair et al.
A 350-hour dataset of diverse real-world robot manipulation across 22 robots in 71 scenes, designed to train scalable and generalist imitation learning policies.
LeRobot: A Library for Real-World Robot Learning
Remi Cadene, Simon Alibert, Alexander Soare et al.
Hugging Face's LeRobot is an open-source PyTorch framework providing pretrained models, datasets, and training scripts for imitation and reinforcement learning on real robots, lowering the entry barrier to robot learning.
Open X-Embodiment: Robotic Learning Datasets and RT-X Models
Open X-Embodiment Collaboration
The largest collaborative robot learning dataset ever assembled, spanning 22 robots from 21 institutions, enabling training of generalist RT-X policies that exhibit positive cross-embodiment transfer.