arXiv:cs.LG· Shen Ruan, Wenchang Gao, Jin Wang, Siao Liu, Zhaxizhuoma, Dongchun Ren, Xin Zheng·· 13 小时前
量化 VLA 泛化性恢复:PIVOT-Q 用脆弱状态定向调优,仅需 7.4% 蒸馏预算
Beyond In-Distribution Preservation: Recovering Generalization in Quantized VLAs via Vulnerability-Oriented Tuning
AI 导读
针对量化 VLA 在环境扰动下易失效的问题,研究提出脆弱性导向的 On-Policy Distillation 框架 PIVOT-Q,以冻结的全精度策略为教师,选择性纠正量化学生 rollout 中的脆弱状态。在 LIBERO-Plus 的七种环境变化下,该方法在多种 VLA 骨干与量化方法上均稳定恢复性能,且全部设置下优于全状态蒸馏,状态级蒸馏预算仅为其 7.4%。
正文
Abstract:Post-training quantization has been shown to preserve VLA performance under standard evaluation conditions, but whether it preserves the full-precision model's robustness and generalization remains underexplored. In this study, we systematically study the robustness and generalization of post-quantized VLA policies under environmental disturbances. Empirical results show that quantized policies can become fragile to subtle environmental variations despite retaining comparable in-distribution performance. We further observe that action discrepancies are concentrated in a small subset of rollout states, while teacher guidance has opposite effects depending on discrepancy: it improves generalization at high-discrepancy states but can degrade it at low-discrepancy states. These findings reveal that effective post-quantization recovery requires selectively intervening on vulnerable states rather than globally distilling the student. We therefore propose Policy-Induced Vulnerability-Oriented Tuning (PIVOT-Q), a vulnerability-aware On-Policy Distillation (OPD) framework that selectively corrects vulnerable states encountered during quantized-student rollouts using the frozen full-precision policy as a teacher. PIVOT-Q identifies vulnerable states using discounted accumulated discrepancies over a short horizon, applies phase-balanced sparse supervision, and uses a Behavioral Anchor to prevent unnecessary changes. Experiments under seven LIBERO-Plus environmental variations demonstrate consistent recovery across multiple VLA backbones and quantization methods. Notably, PIVOT-Q consistently outperforms full-state distillation across all settings while using only 7.4% of its state-level distillation budget. Our code is available at this https URL.
| Subjects: | Machine Learning (cs.LG); Robotics (cs.RO) |
| Cite as: | arXiv:2610.05745 [cs.LG] |
| (or arXiv:2610.05745v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2610.05745 arXiv-issued DOI via DataCite |
Submission history
From: Siao Liu [view email]
[v1]
Mon, 5 Oct 2026 03:51:21 UTC (4,828 KB)
[v2]
Fri, 9 Oct 2026 10:33:09 UTC (4,827 KB)
来源:arXiv:cs.LG · arxiv.org