Add VLADrop-pi05-LIBERO-metric-drop9-taylor-igia (VLADrop / DTR, arXiv:2606.27755)
Browse files- README.md +54 -0
- assets/physical-intelligence/libero/norm_stats.json +84 -0
- metadata.pt +3 -0
- model.safetensors +3 -0
README.md
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- robotics
|
| 5 |
+
- vla
|
| 6 |
+
- vision-language-action
|
| 7 |
+
- libero
|
| 8 |
+
- model-compression
|
| 9 |
+
pipeline_tag: robotics
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# VLADrop-pi05-LIBERO-metric-drop9-taylor-igia
|
| 13 |
+
|
| 14 |
+
Checkpoint for [Drop-Then-Recovery: How Redundant Are Vision-Language-Action Models?](https://arxiv.org/abs/2606.27755).
|
| 15 |
+
|
| 16 |
+
DTR (Drop-Then-Recovery) removes transformer blocks from a pretrained VLA model and
|
| 17 |
+
recovery-fine-tunes the smaller dense model. Code: https://github.com/s1ghhh/VLADrop
|
| 18 |
+
|
| 19 |
+
## This checkpoint
|
| 20 |
+
|
| 21 |
+
| | |
|
| 22 |
+
|---|---|
|
| 23 |
+
| Paper row | Table 3: Taylor (= IGIA, identical block selection at this level) (Drop-9 importance-metric comparison) |
|
| 24 |
+
| Dropped blocks | Language backbone (PaliGemma, 18 layers): drop 9 whole blocks selected by Taylor (= IGIA, identical block selection at this level); keep blocks [0,1,3,4,5,6,7,8,9]. Vision and action untouched. |
|
| 25 |
+
| Recovery training | batch size 32, 30K steps, lr 5e-5 |
|
| 26 |
+
| LIBERO success rate | Spatial / Object / Goal / Long / Avg = 97.0 / 97.0 / 94.0 / 88.6 / 94.2 |
|
| 27 |
+
|
| 28 |
+
## Usage
|
| 29 |
+
|
| 30 |
+
This is an [openpi](https://github.com/Physical-Intelligence/openpi)-format pi0.5 checkpoint
|
| 31 |
+
(PyTorch). Use with the VLADrop fork: https://github.com/s1ghhh/VLADrop
|
| 32 |
+
|
| 33 |
+
```bash
|
| 34 |
+
python scripts/serve_policy_batch_drop.py \
|
| 35 |
+
--config pi05_libero_dropped \
|
| 36 |
+
--dir <this_repo_local_path> \
|
| 37 |
+
--port 8000
|
| 38 |
+
```
|
| 39 |
+
|
| 40 |
+
**Important:** the drop lists are NOT stored inside the checkpoint. Pass the exact
|
| 41 |
+
`llm_drop_attn_list` / `llm_drop_mlp_list` shown above (via config or CLI) when serving,
|
| 42 |
+
otherwise layers will be mismatched. `assets/` contains the LIBERO norm stats.
|
| 43 |
+
The optimizer state (`train_state/`) is not included.
|
| 44 |
+
|
| 45 |
+
## Citation
|
| 46 |
+
|
| 47 |
+
```bibtex
|
| 48 |
+
@article{sun2026vladrop,
|
| 49 |
+
title={Drop-Then-Recovery: How Redundant Are Vision-Language-Action Models?},
|
| 50 |
+
author={Sun, Guoheng and Feng, Kaixi and He, Shwai and Gong, Xiaochuan and He, Yexiao and Wang, Ziyao and Shen, Zheyu and Ye, Wanghao and Kompella, Ramana Rao and Liu, Gaowen and Li, Ang},
|
| 51 |
+
journal={arXiv preprint arXiv:2606.27755},
|
| 52 |
+
year={2026}
|
| 53 |
+
}
|
| 54 |
+
```
|
assets/physical-intelligence/libero/norm_stats.json
ADDED
|
@@ -0,0 +1,84 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"norm_stats": {
|
| 3 |
+
"state": {
|
| 4 |
+
"mean": [
|
| 5 |
+
-0.04652611166238785,
|
| 6 |
+
0.034395575523376465,
|
| 7 |
+
0.7645326256752014,
|
| 8 |
+
2.972200632095337,
|
| 9 |
+
-0.22040514647960663,
|
| 10 |
+
-0.12562009692192078,
|
| 11 |
+
0.026915626600384712,
|
| 12 |
+
-0.02719222754240036
|
| 13 |
+
],
|
| 14 |
+
"std": [
|
| 15 |
+
0.10494592040777206,
|
| 16 |
+
0.15176665782928467,
|
| 17 |
+
0.37852945923805237,
|
| 18 |
+
0.3442988693714142,
|
| 19 |
+
0.9069637656211853,
|
| 20 |
+
0.3253786265850067,
|
| 21 |
+
0.014175431802868843,
|
| 22 |
+
0.014058168977499008
|
| 23 |
+
],
|
| 24 |
+
"q01": [
|
| 25 |
+
-0.3992278680294752,
|
| 26 |
+
-0.2688781352341175,
|
| 27 |
+
0.0380015922319144,
|
| 28 |
+
1.5083277968406676,
|
| 29 |
+
-2.721004430627823,
|
| 30 |
+
-1.0813218128442763,
|
| 31 |
+
0.00173520012809895,
|
| 32 |
+
-0.04004424215038307
|
| 33 |
+
],
|
| 34 |
+
"q99": [
|
| 35 |
+
0.13532010269463068,
|
| 36 |
+
0.33566314613819126,
|
| 37 |
+
1.2698724751349537,
|
| 38 |
+
3.2765059180259706,
|
| 39 |
+
2.405436800289154,
|
| 40 |
+
0.5971528978824616,
|
| 41 |
+
0.04030334069116507,
|
| 42 |
+
-0.0017878379381960213
|
| 43 |
+
]
|
| 44 |
+
},
|
| 45 |
+
"actions": {
|
| 46 |
+
"mean": [
|
| 47 |
+
0.05515800043940544,
|
| 48 |
+
0.08770468831062317,
|
| 49 |
+
-0.08735653758049011,
|
| 50 |
+
0.0007100921357050538,
|
| 51 |
+
0.005471237003803253,
|
| 52 |
+
-0.005537765566259623,
|
| 53 |
+
-0.04559307545423508
|
| 54 |
+
],
|
| 55 |
+
"std": [
|
| 56 |
+
0.331879198551178,
|
| 57 |
+
0.3758169412612915,
|
| 58 |
+
0.448102205991745,
|
| 59 |
+
0.039334457367658615,
|
| 60 |
+
0.06358090043067932,
|
| 61 |
+
0.07857435941696167,
|
| 62 |
+
0.9989600777626038
|
| 63 |
+
],
|
| 64 |
+
"q01": [
|
| 65 |
+
-0.7072499999999999,
|
| 66 |
+
-0.7983750000000001,
|
| 67 |
+
-0.9375,
|
| 68 |
+
-0.11467779144644738,
|
| 69 |
+
-0.16425,
|
| 70 |
+
-0.2240490057706833,
|
| 71 |
+
-1.0
|
| 72 |
+
],
|
| 73 |
+
"q99": [
|
| 74 |
+
0.937125,
|
| 75 |
+
0.86775,
|
| 76 |
+
0.937125,
|
| 77 |
+
0.1315075716376305,
|
| 78 |
+
0.19169999999999998,
|
| 79 |
+
0.33742949963808055,
|
| 80 |
+
0.9996
|
| 81 |
+
]
|
| 82 |
+
}
|
| 83 |
+
}
|
| 84 |
+
}
|
metadata.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:23b6c26a6050dacaeb36049b634f8c9454cbac1f98836a652ca3a8b1ac737a31
|
| 3 |
+
size 3379
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5e1122ae52812394df9b573ee05017864a8d91082e5b9a83ad676f897303e3bc
|
| 3 |
+
size 7473091464
|