Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ST-Prune: Training-Free Spatio-Temporal Token Pruning for Vision-Language Models in Autonomous Driving

About

Vision-Language Models (VLMs) have become central to autonomous driving systems, yet their deployment is severely bottlenecked by the massive computational overhead of multi-view camera and multi-frame video input. Existing token pruning methods, primarily designed for single-image inputs, treat each frame or view in isolation and thus fail to exploit the inherent spatio-temporal redundancies in driving scenarios. To bridge this gap, we propose ST-Prune, a training-free, plug-and-play framework comprising two complementary modules: Motion-aware Temporal Pruning (MTP) and Ring-view Spatial Pruning (RSP). MTP addresses temporal redundancy by encoding motion volatility and temporal recency as soft constraints within the diversity selection objective, prioritizing dynamic trajectories and current-frame content over static historical background. RSP further resolves spatial redundancy by exploiting the ring-view camera geometry to penalize bilateral cross-view similarity, eliminating duplicate projections and residual background that temporal pruning alone cannot suppress. These two modules together constitute a complete spatio-temporal pruning process, preserving key scene information under strict compression. Validated across four benchmarks spanning perception, prediction, and planning, ST-Prune establishes new state-of-the-art for training-free token pruning. Notably, even at 90\% token reduction, ST-Prune achieves near-lossless performance with certain metrics surpassing the full-model baseline, while maintaining inference speeds comparable to existing pruning approaches.

Lin Sha, Haiyun Guo, Tao Wang, Cong Zhang, Min Huang, Jinqiao Wang, Qinghai Miao• 2026

Related benchmarks

TaskDatasetResultRank
Autonomous Driving Instruction FollowingNuInstruct
MAE3.52
14
Multi-view Driving Question AnsweringOmniDrive 25% Token Retention
BLEU40.82
5
Vision-Language Instruction FollowingNuInstruct 25% Token Retention
MAE3.49
5
Autonomous Driving ReasoningDriveLM 25% Token Retention
Accuracy81.1
5
Driving Question AnsweringDriveLM
Accuracy77.6
5
Driving Video Question AnsweringLingoQA 25% Token Retention
LingoJudge Score68.2
5
Driving Video Question AnsweringLingoQA
LingoJudge65.2
5
Multimodal Driving Scene ReasoningOmniDrive
BLEU40.78
5
Reasoning and GenerationDriveLM (test)
Accuracy81
5
Showing 9 of 9 rows

Other info

Follow for update