FastVLA: Task-Conditioned and Temporally Adaptive Visual Token Compression for Efficient Vision-Language-Action Models
Organizations: Colorado School of Mines · Texas A&M University · Microsoft AI
Abstract
Vision-Language-Action (VLA) models present a promising paradigm for general-purpose robot manipulation, but closed-loop deployment is often limited by inference efficiency: at each timestep, high-resolution, often multi-view images are encoded into hundreds of visual tokens that dominate transformer computation and latency. Token compression via attention is a natural solution but is challenging for VLA control because (1) decision-critical tokens are determined by the task, not generic visual saliency, and (2) the token budget varies substantially across steps, so static metrics and budgets waste computation or discard control-critical tokens. We introduce FastVLA, a training-free, task-aware, and temporally adaptive visual token compression framework. At each step, FastVLA scores token importance via task-to-vision attention and adaptively sets the compression rate from the importance-distribution uniformity, measured by an inverse-Simpson metric with online sliding-window calibration. We evaluate FastVLA on OpenVLA, OpenVLA-OFT, and CogACT across LIBERO and SIMPLER, and on a physical Franka Research 3 robot. In simulation, it reduces LLM-backbone FLOPs by 35-59% and action latency by 9-32% (a 1.09-1.46 speedup) while maintaining task success, achieving a comparable or better efficiency-performance trade-off than existing plug-and-play token compression/pruning/reuse baselines. On the real robot, it cuts end-to-end latency by 20% (1.25 speedup) at comparable success. Code is available at https://github.com/jiuyixu25/FastVLA.
Figures & tables
| Method | Success Rate (%) | FLOPs (T) | Latency (ms) | Control Freq. (Hz) | CUDA Late. (ms) | Speedup | ||||
| Spatial | Object | Goal | 10 | Average | ||||||
| OpenVLA | 85.8 | 88.8 | 77.4 | 53.4 | 76.4 | 1.86 | 186.64 | 5.36 | 42.28 | 1.00 |
| + VLA-Cache | 86.0 | 88.0 | 75.4 | 54.4 | 76.0 | 1.36 | 203.37 | 4.92 | 31.83 | 0.92 |
| + FlashVLA ∗ | 82.6 | 86.6 | 78.8 | 46.8 | 73.7 | 1.08 | - | - | - | 1.56 |
| + FastVLA | 85.3 0.1 | 87.2 0.2 | 77.6 0.2 | 53.2 0.4 | 75.8 0.2 | 1.20 | 161.53 | 6.19 | 26.91 | 1.16 |
| OpenVLA-OFT | 98.2 | 98.2 | 97.6 | 94.0 | 97.0 | 3.98 | 118.52 | 67.50 | 65.99 | 1.00 |
| SIMPLER | Method | Success Rate (%) | FLOPs (T) | Latency (ms) | Control Freq. (Hz) | CUDA Late. (ms) | Speedup | ||||
| PickCan | MoveNear | Drawer | DrawerApple | Average | |||||||
| Visual Matching | CogACT-base | 93.7 | 86.7 | 75.9 | 53.7 | 77.5 | 1.85 | 174.48 | 5.73 | 82.53 | 1.00 |
| + FastVLA | 93.7 2.0 | 85.4 0.7 | 75.5 0.9 | 53.6 0.2 | 77.1 0.8 | 1.16 | 159.59 | 6.27 | 66.10 | 1.09 | |
| Variant Aggregation | CogACT-base | 89.1 | 76.7 | 50.8 | 27.0 | 60.9 | 1.81 | 174.94 | 5.72 | 82.74 | 1.00 |
| + FastVLA | 89.5 1.3 | 77.2 1.0 | 52.4 0.3 | 28.3 0.4 | 61.9 0.6 | 1.13 | 157.06 | 6.37 | 65.34 | 1.11 | |
| Task | Method | Success Rate | Latency (ms) | Control Freq. (Hz) | Speedup |
| Pick | OpenVLA-OFT | 84.0% | 121.13 | 66.04 | 1.0 |
| + FastVLA | 80.0% | 96.90 | 82.56 | 1.25 | |
| Pick and place | OpenVLA-OFT | 68.0% | 120.01 | 66.67 | 1.0 |
| + FastVLA | 66.0% | 97.77 | 81.82 | 1.23 |
| Method | Spatial | Object | Goal | 10 | ||||
| Tokens | SR (%) | Tokens | SR (%) | Tokens | SR (%) | Tokens | SR (%) | |
| Fixed @ nominal | 102 (80%) | 96.7 0.1 | 154 (70%) | 97.0 0.2 | 128 (75%) | 95.1 0.5 | 205 (60%) | 91.9 0.4 |
| Fixed, budget-matched | 105 | 96.5 0.1 | 157 | 97.1 0.1 | 132 | 95.7 0.3 | 207 | 92.0 0.3 |
| Adaptive | 104.3 | 97.8 0.1 | 156.3 | 98.0 0.1 | 131.9 | 97.0 0.5 | 206.8 | 93.1 0.5 |
| Calibration | Spatial | Object | ||
| SR (%) | Tokens | SR (%) | Tokens | |
| Min–Max | 97.8 | 104.3 | 98.0 | 156.3 |
| Rank | 97.6 | 103.0 | 98.0 | 159.5 |
| Calibration | Goal | 10 | ||
| SR (%) | Tokens | SR (%) | Tokens | |
| Min–Max | 97.0 | 131.9 | 93.1 | 206.8 |
| Diversity Metric | Goal | 10 | ||
| SR (%) | Tokens | SR (%) | Tokens | |
| Shannon | 94.0 | 129.6 | 90.8 | 205.5 |
| Gini | 95.3 | 130.3 | 91.1 | 205.2 |
| Inverse-Simpson | 97.0 | 131.9 | 93.1 | 206.8 |
Efficient VLA Model Inference
Efficient vision-language-action (VLA) model inference reduces robot-control latency, memory use, and energy while preserving task performance.
See all 174 papers