English

SpecPrune-VLA: Accelerating Vision-Language-Action Models via Action-Aware Self-Speculative Pruning

Computer Vision and Pattern Recognition 2026-05-26 v3 Artificial Intelligence Robotics

Abstract

Pruning is a typical acceleration technique for compute-bound models by removing computation on unimportant values. Recently, it has been applied to accelerate Vision-Language-Action (VLA) model inference. However, existing acceleration methods focus on local information from the current action step and ignore the global context, leading to >20% success rate drop and limited speedup in some scenarios. In this paper, we point out spatial-temporal consistency in VLA tasks: input images in consecutive steps exhibit high similarity, and propose the key insight that token selection should combine local information with global context of the model. Based on this, we propose SpecPrune-VLA, a training-free, two-level pruning method with heuristic control. (1) Action-level static pruning. We leverage global history and local attention to statically reduce visual tokens per action. (2) Layer-level dynamic pruning. We prune tokens adaptively per layer based on layer-wise importance. (3) Lightweight action-aware controller: We classify actions as coarse- or fine-grained by the speed of the end effector and adjust pruning aggressiveness accordingly. Extensive experiments show that SpecPrune-VLA achieves up to 1.57×\times speedup in LIBERO simulation and 1.70×\times on real-world tasks, with negligible success rate degradation.

Keywords

Cite

@article{arxiv.2509.05614,
  title  = {SpecPrune-VLA: Accelerating Vision-Language-Action Models via Action-Aware Self-Speculative Pruning},
  author = {Hanzhen Wang and Jiaming Xu and Yushun Xiang and Jiayi Pan and Yongkang Zhou and Yong-Lu Li and Guohao Dai},
  journal= {arXiv preprint arXiv:2509.05614},
  year   = {2026}
}

Comments

Accepted to ICML 2026

R2 v1 2026-07-01T05:24:10.283Z