|
1 | 1 | <div align="center"> |
2 | 2 |
|
3 | | -<img src="assets/icon.png" alt="QuantVLA Logo" width="160"> |
| 3 | +<img src="assets/icon.png" alt="QuantVLA Logo" width="180"> |
| 4 | + |
| 5 | +<br> |
4 | 6 |
|
5 | 7 | # QuantVLA |
6 | 8 |
|
7 | | -### Scale-Calibrated Post-Training Quantization for Vision-Language-Action Models |
| 9 | +**Scale-Calibrated Post-Training Quantization for Vision-Language-Action Models** |
8 | 10 |
|
9 | | -[](https://cvpr.thecvf.com/Conferences/2026) |
| 11 | +<a href="https://cvpr.thecvf.com/Conferences/2026"><img src="https://img.shields.io/badge/CVPR-2026-6B46C1?style=for-the-badge&logo=ieee&logoColor=white" alt="CVPR 2026"></a> |
10 | 12 |
|
11 | | -[](https://arxiv.org/pdf/2602.20309) |
12 | | -[](https://arxiv.org/abs/2602.20309) |
13 | | -[](https://quantvla.github.io/) |
14 | | -[](https://github.com/AIoT-MLSys-Lab/QuantVLA) |
| 13 | +<br> |
15 | 14 |
|
16 | | ---- |
| 15 | +<a href="https://arxiv.org/pdf/2602.20309"><img src="https://img.shields.io/badge/📄_Paper-PDF-d32f2f?style=for-the-badge" alt="Paper"></a> |
| 16 | +<a href="https://arxiv.org/abs/2602.20309"><img src="https://img.shields.io/badge/📝_arXiv-2602.20309-b31b1b?style=for-the-badge" alt="arXiv"></a> |
| 17 | +<a href="https://quantvla.github.io/"><img src="https://img.shields.io/badge/🌐_Project-Page-7c4dff?style=for-the-badge" alt="Project Page"></a> |
| 18 | +<a href="https://github.com/AIoT-MLSys-Lab/QuantVLA"><img src="https://img.shields.io/badge/💻_GitHub-Code-181717?style=for-the-badge" alt="Code"></a> |
| 19 | + |
| 20 | +<br> |
| 21 | + |
| 22 | +[Jingxuan Zhang](https://github.com)<sup>2†</sup> |
| 23 | +[Yunta Hsieh](https://github.com)<sup>3†</sup> |
| 24 | +[Zhongwei Wan](https://github.com)<sup>1</sup> |
| 25 | +[Haokun Lin](https://github.com)<sup>4</sup> |
| 26 | +[Xin Wang](https://github.com)<sup>1</sup> |
| 27 | +[Ziqi Wang](https://github.com)<sup>1</sup> |
| 28 | +[Yingtie Lei](https://github.com)<sup>1</sup> |
| 29 | +[Mi Zhang](https://github.com)<sup>1*</sup> |
| 30 | + |
| 31 | +<sup>1</sup>The Ohio State University <sup>2</sup>Indiana University <sup>3</sup>University of Michigan <sup>4</sup>City University of Hong Kong |
| 32 | + |
| 33 | +<sup>†</sup>Equal Contribution <sup>*</sup>Corresponding Author |
| 34 | + |
| 35 | +</div> |
17 | 36 |
|
18 | | -🏆 **First Post-Training Quantization for VLA Systems** | 💾 **~70% Memory Savings** | ⚡ **Training-Free** |
| 37 | +<br> |
19 | 38 |
|
| 39 | +<div align="center"> |
| 40 | + |
| 41 | +| 🏆 First PTQ for VLA | 💾 ~70% Memory Savings | ⚡ Training-Free | 🚀 1.22× Speedup | |
| 42 | +|:---:|:---:|:---:|:---:| |
| 43 | +| First post-training quantization framework for Vision-Language-Action systems | Significant memory reduction on quantized components | Uses only a small unlabeled calibration buffer — no retraining needed | End-to-end inference latency improvement | |
| 44 | + |
| 45 | +</div> |
| 46 | + |
| 47 | +<br> |
| 48 | + |
| 49 | +<div align="center"> |
| 50 | +<img src="assets/pipeline.svg" alt="QuantVLA Pipeline" width="100%"> |
| 51 | +<br> |
| 52 | +<em>Overview of the QuantVLA framework: selective quantization layout + attention temperature matching + output head balancing.</em> |
20 | 53 | </div> |
21 | 54 |
|
| 55 | +<br> |
| 56 | + |
22 | 57 | ## Abstract |
23 | 58 |
|
24 | | -Vision-language-action (VLA) models unify perception, language, and control for embodied agents but face significant challenges in practical deployment due to rapidly increasing compute and memory demands, especially as models scale to longer horizons and larger backbones. To address these bottlenecks, we introduce **QuantVLA**, a training-free post-training quantization (PTQ) framework that, to our knowledge, is the first PTQ approach for VLA systems and the first to successfully quantize a diffusion transformer (DiT) action head. QuantVLA incorporates three scale-calibrated components: (1) a selective quantization layout that integerizes all linear layers in both the language backbone and the DiT while keeping attention projections in floating point to preserve the original operator schedule; (2) attention temperature matching, a lightweight per-head scaling mechanism that stabilizes attention logits and is folded into the dequantization scales at inference; and (3) output head balancing, a per-layer residual interface calibration that mitigates post-projection energy drift. The framework requires no additional training, uses only a small unlabeled calibration buffer, and supports integer kernels for low-bit weights and activations while leaving the architecture unchanged. Across representative VLA models on LIBERO, QuantVLA exceeds the task success rates of full-precision baselines, achieves about **70% relative memory savings** on the quantized components, providing a practical pathway toward scalable low-bit embodied intelligence under strict compute, memory, and power constraints. |
| 59 | +Vision-language-action (VLA) models unify perception, language, and control for embodied agents but face significant challenges in practical deployment due to rapidly increasing compute and memory demands, especially as models scale to longer horizons and larger backbones. To address these bottlenecks, we introduce **QuantVLA**, a training-free post-training quantization (PTQ) framework that, to our knowledge, is the first PTQ approach for VLA systems and the first to successfully quantize a diffusion transformer (DiT) action head. QuantVLA incorporates three scale-calibrated components: (1) a selective quantization layout that integerizes all linear layers in both the language backbone and the DiT while keeping attention projections in floating point to preserve the original operator schedule; (2) attention temperature matching, a lightweight per-head scaling mechanism that stabilizes attention logits and is folded into the dequantization scales at inference; and (3) output head balancing, a per-layer residual interface calibration that mitigates post-projection energy drift. The framework requires no additional training, uses only a small unlabeled calibration buffer, and supports integer kernels for low-bit weights and activations while leaving the architecture unchanged. Across representative VLA models on LIBERO, QuantVLA exceeds the task success rates of full-precision baselines, achieves about **70% relative memory savings** on the quantized components, and delivers a **1.22× speedup** in end-to-end inference latency, providing a practical pathway toward scalable low-bit embodied intelligence under strict compute, memory, and power constraints. |
25 | 60 |
|
26 | | -- 📄 Paper: https://arxiv.org/abs/2602.20309 |
27 | | -- 🌐 Project page: https://quantvla.github.io/ |
| 61 | +<p align="center"> |
| 62 | + 📄 <a href="https://arxiv.org/abs/2602.20309">Paper</a> | |
| 63 | + 🌐 <a href="https://quantvla.github.io/">Project Page</a> | |
| 64 | + 💻 <a href="https://github.com/AIoT-MLSys-Lab/QuantVLA">Code</a> |
| 65 | +</p> |
28 | 66 |
|
29 | 67 |
|
30 | 68 | # QuantVLA GR00T Environment Setup Guide |
|
0 commit comments