VLADrop-pi05-RoboTwin-turn-switch-drop12-gate
Checkpoint for Drop-Then-Recovery: How Redundant Are Vision-Language-Action Models?.
DTR (Drop-Then-Recovery) removes transformer blocks from a pretrained VLA model and recovery-fine-tunes the smaller dense model. Code: https://github.com/s1ghhh/VLADrop
This checkpoint
| Paper row | Table 17 / Figure 4: RoboTwin 2.0 'turn switch' Drop-12 |
| Dropped blocks | Language backbone (PaliGemma, 18 layers): drop 12 whole blocks selected by per-task GateProbe; keep blocks [0,1,2,3,5,6]. Vision and action untouched. |
| Recovery training | single-task RoboTwin 2.0, FLOPs-matched to baseline (step 34234), lr 5e-5 |
| RoboTwin 2.0 success rate | Easy 12 / Hard 17 |
Usage
This is an openpi-format pi0.5 checkpoint (PyTorch). Use with the VLADrop fork: https://github.com/s1ghhh/VLADrop
python scripts/serve_policy_batch_drop.py \
--config pi05_libero_dropped \
--dir <this_repo_local_path> \
--port 8000
Important: the drop lists are NOT stored inside the checkpoint. Pass the exact
llm_drop_attn_list / llm_drop_mlp_list shown above (via config or CLI) when serving,
otherwise layers will be mismatched. assets/ contains the LIBERO norm stats.
The optimizer state (train_state/) is not included.
Citation
@article{sun2026vladrop,
title={Drop-Then-Recovery: How Redundant Are Vision-Language-Action Models?},
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},
journal={arXiv preprint arXiv:2606.27755},
year={2026}
}