Recovering Low-Bit Failures in Vision-Language-Action Policies via Quantization-Conditioned Reinforcement Learning
cs.RO, cs.CV
Submitted: 2026-10-11
Updated: 2026-10-11
Code: https://github.com/apple-aiml-research/ml-stable-diffusion
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Quantization lowers deployment costs of vision-language-action (VLA) policies but can compromise task completion.
Terminology
Abstract
Quantization lowers deployment costs of vision-language-action (VLA) policies but can compromise task completion. Across five VLA models and 4,050 real-world trials, we systematically reveal that quantization-aware training (QAT) generally improves on post-training quantization (PTQ) but leaves substantial gaps to full precision (FP) on long-horizon tasks and under visual and semantic shifts, while its imitation-based objective does not explicitly use rollout success or partial completion. To exploit these outcome signals, we introduce Quantization-Conditioned Reinforcement Learning (QCRL), an offline RL method coupling progress-based advantage estimation with quantization-aware source conditioning. FP rollouts supply reference experience beyond action-level distillation, progress-scaled terminal penalty distinguishes partial completion, and quantization-aware conditioning selects advantageous reference behavior within the quantized policy. We evaluate weight-only INT4 W4A16 on π 0.5 and GR00T N1.5 and joint weight-activation NVFP4 W4A4 on π 0.5. Using only in-distribution data, QCRL reaches 99.3% of FP's mean score across three visual-shift conditions for π 0.5 with INT4 W4A16, alongside 42.9% model-memory savings. With NVFP4 W4A4, it reaches 100.7% of the same FP baseline, alongside a 1.73 times speedup using NVFP4/FP8 deployment. Hardware gains are relative to FP16 on Jetson Thor.
Sources
- A White Paper on Neural Network Quantization
- SmolVLA: A Vision-Language-Action Model for Affordable and Efficient Robotics
- GR00T N1: An Open Foundation Model for Generalist Humanoid Robots
- QuantVLA: Scale-Calibrated Post-Training Quantization for Vision-Language-Action Models
- QVLA: Not All Channels Are Equal in Vision-Language-Action Model's Quantization
- Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problems
- Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learning
- $\pi^{*}_{0.6}$: a VLA That Learns From Experience
Related papers
- FMT x: An Efficient and Asymptotically Optimal Extension of the Fast Marching Tree for Dynamic Replanning
- MPCFormer: A physics-informed data-driven approach for explainable socially-aware autonomous driving
- RoboLab: A High-Fidelity Simulation Benchmark for Analysis of Task Generalist Policies
- ARC: A Reasoning Recipe for Robot Foundation Models
- HRDexDB: A 4D Dexterous Grasping Dataset Across Human and Multiple Robot Embodiments
- APT: Action Expert Pretraining Improves Instruction Generalization of Vision-Language-Action Policies