AI Models Just Got 73% Lighter and 1.5x Faster — Thanks to a Clever New Quantization Trick
New quantization method cuts OpenVLA memory from 15.4GB to 4.1GB while keeping accuracy high.
Vision-Language-Action (VLA) models power next-gen robotics, but their size (e.g., OpenVLA-OFT at 15.4GB) makes real-time edge deployment impractical. Standard quantization often destroys task-critical evidence hidden in cross-modal attention layers. To solve this, Navin Ranjan and Andreas Savakis introduce Mix-QVLA, a post-training mixed-precision quantization framework that preserves decision-relevant information. Instead of applying uniform bit-widths, Mix-QVLA computes normalized gradient-weighted task-evidence maps from activations at key functional boundaries (vision encoder, language model, action head). It compares full-precision and quantized maps using evidence-mass and attribution-distortion metrics, then aggregates degradation into layer-wise sensitivity scores via a soft-bottleneck objective. Crucially, Mix-QVLA models sensitivity throughout task execution rather than assuming a static profile, capturing phase-dependent shifts in which layers matter most. The resulting time- and evidence-aware sensitivity scores guide bit allocation under model-size and BitOps budgets.
Evaluated on OpenVLA-style policies running LIBERO benchmark tasks, Mix-QVLA delivers dramatic efficiency gains: memory drops from 15.4GB to 4.1GB (a 73% reduction) while the average success rate only dips from 97.1% (BF16 baseline) to 96.3%. Inference speed increases by 1.52x, and the mixed-precision model also reduces BitOps by over 60%. The method consistently outperforms uniform quantization and prior mixed-precision approaches across different VLA architectures and task scenarios. By intelligently allocating precision where it matters most — and adapting to shifting importance during a robot’s decision sequence — Mix-QVLA makes it feasible to deploy large VLA models on resource-constrained hardware like drones, mobile manipulators, or edge servers. This work, published on arXiv, represents a practical step toward scalable, real-world robotic policy deployment without sacrificing task success.
- Reduces OpenVLA-OFT memory from 15.4GB to 4.1GB (73% reduction) on LIBERO benchmarks
- Retains 96.3% average success rate vs. 97.1% for BF16 baseline — only 0.8% drop
- Achieves 1.52x inference speedup while cutting BitOps by over 60%
Why It Matters
Enables real-time, on-device deployment of large VLA models for robotics without sacrificing accuracy.