FoldQuantVLA: Native Low-Bit Quantization of Vision-Language-Action Models via Consistent Folding
En palabras de los autores
Low-bit vision-language-action inference must reduce observation-to-action latency while preserving robot behavior. We present FoldQuantVLA, a post-training quantization framework that carries a consistent activation representation through calibration, weight rounding, and native integer execution. It combines channel scaling and block Hadamard transforms with dynamic per-token quantization, without policy retraining. Custom TensorRT plugins execute projections in both the language backbone and iterative action expert with four-bit weights and activations (W4A4) on Ada GPUs and Jetson AGX Orin. Evaluation spans LIBERO, SimplerEnv, and two robot platforms. Across three GR00T checkpoints and , W4A4 achieves to speedups over floating-point TensorRT on Orin and to on desktop. Retaining language attention-output and feed-forward down projections at eight bits (W8A8) improves held-out action fidelity on all four checkpoints. Across four real-robot tasks, this configuration raises observed GR00T N1.7 success from with uniform W4A4 to over 80 trials per configuration, with a measured additional Orin latency of 1 ms.
Apareció: martes, 22 de septiembre. arXiv. Preprint, todavía sin revisión por pares.
Comentario de los autores: 8 pages, 5 figures, 7 tables. Code: https://github.com/cair-vinuni/FoldQuantVLA