nielsr HF Staff commited on
Commit
a7e8f7d
·
verified ·
1 Parent(s): 9ca81f3

Add model card for DeepThinkVLA

Browse files

This 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.

Files changed (1) hide show
  1. README.md +43 -0
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
+ ```