Add model card for DeepThinkVLA
Browse filesThis PR adds a comprehensive model card for DeepThinkVLA. It includes:
- A link to the paper: [DeepThinkVLA: Enhancing Reasoning Capability of Vision-Language-Action Models](https://huggingface.co/papers/2511.15669).
- A link to the official GitHub repository.
- The `robotics` pipeline tag to improve discoverability on the Hub.
- Information regarding the model's hybrid-attention architecture and performance results on the LIBERO benchmark.
README.md
ADDED
|
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
pipeline_tag: robotics
|
| 3 |
+
---
|
| 4 |
+
|
| 5 |
+
# DeepThinkVLA: Enhancing Reasoning Capability of Vision-Language-Action Models
|
| 6 |
+
|
| 7 |
+
DeepThinkVLA is a Vision-Language-Action (VLA) model designed to enhance the reasoning capabilities of robotic agents through explicit deliberation (Chain-of-Thought). It refactors the policy into a hybrid-attention decoder that separates autoregressive reasoning from parallel action generation.
|
| 8 |
+
|
| 9 |
+
- **Paper:** [DeepThinkVLA: Enhancing Reasoning Capability of Vision-Language-Action Models](https://huggingface.co/papers/2511.15669)
|
| 10 |
+
- **Repository:** [https://github.com/OpenBMB/DeepThinkVLA](https://github.com/OpenBMB/DeepThinkVLA)
|
| 11 |
+
|
| 12 |
+
## Overview
|
| 13 |
+
DeepThinkVLA introduces two key innovations for effective Chain-of-Thought (CoT) in robotics:
|
| 14 |
+
1. **Decoding Alignment:** A hybrid-attention decoder that pairs causal attention for language (reasoning tokens) with bidirectional attention for parallel action decoding.
|
| 15 |
+
2. **Causal Alignment:** A two-stage SFT-then-RL pipeline that aligns the reasoning-action chain with sparse task-success rewards, ensuring that the generated "thoughts" are causally linked to task success.
|
| 16 |
+
|
| 17 |
+
The model achieves a 97.0% success rate on the LIBERO benchmark and demonstrates strong robustness under distribution shifts in LIBERO-Plus (79.0% zero-shot success).
|
| 18 |
+
|
| 19 |
+
## Architecture
|
| 20 |
+
DeepThinkVLA inserts a `<think>` segment between visual observations and robot actions. Starting from the `pi0-FAST` checkpoint, it uses a 2.9B parameter hybrid decoder. Reasoning tokens are generated autoregressively, after which the model switches to parallel decoding for action chunks to maintain low latency.
|
| 21 |
+
|
| 22 |
+
## Performance
|
| 23 |
+
- **LIBERO:** 97.0% average success rate (Object 99.0, Spatial 96.6, Goal 96.4, Long 96.2).
|
| 24 |
+
- **LIBERO-Plus:** 79.0% zero-shot robustness.
|
| 25 |
+
- **RoboTwin 2.0:** 59.3% success rate.
|
| 26 |
+
|
| 27 |
+
## Setup and Evaluation
|
| 28 |
+
Please refer to the [official GitHub repository](https://github.com/OpenBMB/DeepThinkVLA) for detailed instructions on environment setup, dataset acquisition, and running evaluation scripts.
|
| 29 |
+
|
| 30 |
+
Example evaluation command:
|
| 31 |
+
```bash
|
| 32 |
+
bash scripts/eval.sh --pretrained_checkpoint yinchenghust/deepthinkvla_libero_cot_sft
|
| 33 |
+
```
|
| 34 |
+
|
| 35 |
+
## Citation
|
| 36 |
+
```bibtex
|
| 37 |
+
@article{yin2025deepthinkvla,
|
| 38 |
+
title={DeepThinkVLA: Enhancing Reasoning Capability of Vision-Language-Action Models},
|
| 39 |
+
author={Yin, Cheng and Lin, Yankai and Xu, Wang and Tam, Sikyuen and Zeng, Xiangrui and Liu, Zhiyuan and Yin, Zhouping},
|
| 40 |
+
journal={arXiv preprint arXiv:2511.15669},
|
| 41 |
+
year={2025}
|
| 42 |
+
}
|
| 43 |
+
```
|