s1ghhh commited on
Commit
816a9db
·
verified ·
1 Parent(s): 40f89cd

Add VLADrop-pi05-LIBERO-metric-drop9-taylor-igia (VLADrop / DTR, arXiv:2606.27755)

Browse files
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