File size: 5,958 Bytes
7c02222 ca48274 7c02222 ca48274 7c02222 70562e0 7c02222 70562e0 7c02222 ca48274 7c02222 ca48274 7c02222 ca48274 7c02222 70562e0 7c02222 70562e0 7c02222 ca48274 7c02222 ca48274 7c02222 415b867 7c02222 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 | ---
library_name: lerobot
license: apache-2.0
language:
- en
base_model:
- SberRoboticsCenter/GreenVLA-5b-base-stride-1
pipeline_tag: robotics
tags:
- robotics
- vla
- vision-language-action
- manipulation
- flow-matching
- action-prediction
- green-vla
- bridge
- widowx
- reinforcement-learning
datasets:
- IPEC-COMMUNITY/bridge_orig_lerobot
model-index:
- name: GreenVLA-5b-stride-1-R2-bridge
results:
- task:
type: robotics
name: SimplerEnv WidowX (Bridge)
dataset:
type: IPEC-COMMUNITY/bridge_orig_lerobot
name: Bridge
metrics:
- type: success_rate
name: Partial Average
value: 94.5
- type: success_rate
name: Entire Average
value: 80.5
---
<div align="center">
# GreenVLA-5b-stride-1-R2-bridge
### RL-Aligned VLA for Bridge (WidowX)
**Sber Robotics Center · Manipulation Team**
[](https://arxiv.org/abs/2602.00919)
[](https://greenvla.github.io/)
[](https://github.com/greenvla/GreenVLA)
</div>
---
## Overview
**GreenVLA-5b-stride-1-R2-bridge** is the R2 (RL-aligned) checkpoint of the [Green-VLA](https://arxiv.org/abs/2602.00919) family, fine-tuned on the [Bridge](https://huggingface.co/datasets/IPEC-COMMUNITY/bridge_orig_lerobot) dataset for the WidowX robot arm joined with additional trajectories collected in SimplerEnv Bridge environments. Trajectory collection and RL fine-tuning were conducted according to the Trajectory Optimization approach described in the [technical report](https://arxiv.org/abs/2602.00919).
Starting from [GreenVLA-5b-base-stride-1](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-base-stride-1), this model went through both R1 (supervised fine-tuning) and R2 (RL policy alignment) stages, resulting in significant performance gains over behavior cloning alone.
## Evaluation
Evaluated on **SimplerEnv WidowX (Bridge)** benchmark.
> **Note:** Bridge benchmark results can vary up to ±6% between runs. We recommend averaging over multiple evaluation runs for reliable comparisons.
### Partial Success Rate
| Task | Success Rate |
|------|:---:|
| Put Spoon on Towel | 90.6% |
| Put Carrot on Plate | 89.6% |
| Stack Blocks | 99.0% |
| Put Eggplant in Basket | 99.0% |
| **Average** | **94.5%** |
### Entire Success Rate
| Task | Success Rate |
|------|:---:|
| Put Spoon on Towel | 80.2% |
| Put Carrot on Plate | 76.1% |
| Stack Blocks | 70.8% |
| Put Eggplant in Basket | 94.8% |
| **Average** | **80.5%** |
## Training
| | Details |
|---|---|
| **Base checkpoint** | [GreenVLA-5b-base-stride-1](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-base-stride-1) |
| **Stage** | R2 — RL policy alignment |
| **Method** | Trajectory optimization (SFT + RL on collected trajectories) |
| **Dataset** | [IPEC-COMMUNITY/bridge_orig_lerobot](https://huggingface.co/datasets/IPEC-COMMUNITY/bridge_orig_lerobot) + SimplerEnv rollouts |
| **Robot** | WidowX (Bridge) |
| **Parameters** | ~5B |
## Quick Start
### Installation
```bash
git clone https://github.com/greenvla/GreenVLA.git
cd GreenVLA
uv sync # or: pip install -e .
```
### Inference
```python
import numpy as np
import torch
from lerobot.common.policies.factory import load_pretrained_policy
from lerobot.common.utils.torch_observation import (
move_dict_to_batch_for_inference,
torch_preprocess_dict_inference,
)
# 1. Load policy and transforms.
policy, input_transforms, output_transforms = load_pretrained_policy(
"SberRoboticsCenter/GreenVLA-5b-stride-1-R2-bridge",
data_config_name="bridge",
)
policy.to("cuda").eval()
# 2. Build an observation (replace with real sensor data).
raw_obs = {
"observation/state": np.random.rand(8).astype(np.float32), # x y z roll pitch yaw _pad_ gripper
"observation/image": np.random.randint(0, 256, size=(224, 224, 3), dtype=np.uint8),
"prompt": "pick up the green block and place it on the plate",
}
# 3. Transform, preprocess, and batch.
obs = input_transforms(raw_obs)
obs = torch_preprocess_dict_inference(obs)
batch = move_dict_to_batch_for_inference(obs, device="cuda")
# 4. Predict actions and post-process.
with torch.inference_mode():
raw_actions = policy.select_action(batch).cpu().numpy()
actions = output_transforms(
{"actions": raw_actions, "state": batch["state"].cpu().numpy()}
)["actions"]
# actions shape: (action_horizon, 7) — [x, y, z, roll, pitch, yaw, gripper]
```
See [`examples/example_inference_bridge.py`](https://github.com/greenvla/GreenVLA/blob/main/examples/example_inference_bridge.py) for the full runnable script with argument parsing.
## Citation
```bibtex
@misc{apanasevich2026greenvlastagedvisionlanguageactionmodel,
title = {Green-VLA: Staged Vision-Language-Action Model for Generalist Robots},
author = {I. Apanasevich and M. Artemyev and R. Babakyan and P. Fedotova and
D. Grankin and E. Kupryashin and A. Misailidi and D. Nerus and
A. Nutalapati and G. Sidorov and I. Efremov and M. Gerasyov and
D. Pikurov and Y. Senchenko and S. Davidenko and D. Kulikov and
M. Sultankin and K. Askarbek and O. Shamanin and D. Statovoy and
E. Zalyaev and I. Zorin and A. Letkin and E. Rusakov and
A. Silchenko and V. Vorobyov and S. Sobolnikov and A. Postnikov},
year = {2026},
eprint = {2602.00919},
archivePrefix = {arXiv},
primaryClass = {cs.RO},
url = {https://arxiv.org/abs/2602.00919},
}
```
<div align="center">
© 2026 Sber Robotics Center · Manipulation Team
</div>
|