How (and How Not) to Use Data Augmentation in VLA Post-Training

arXiv:2610.05994v1 Announce Type: cross Abstract: Vision-language-action (VLA) models currently demonstrate strong performance in a wide range of real-world robotics tasks. However, they often still lack the generalization ability to handle large visual out-of-distribution shifts. Post-training of VLAs with reinforcement learning (RL) has been shown to benefit robustness, but significant room for improvement remains. In this work, we systematically study the effect of image augmentation on VLA post-training. We find that it is crucial to augment only the critic module during RL updates, while leaving the actor's input clean during both rollouts and updates. For pi_{0.5} and GR00T N1.5 this raises out-of-distribution success on LIBERO-Plus by 7.8 and 10.0 points respectively, while augmenting the actor collapses training entirely. We investigate a range of augmentation types and strengths, and provide practical recommendations for improving generalization in VLA post-training.

Related

• Acting in Meters: Learning Metric Interactions for Precise Robotic Manipulation
• AhaRobot: A Low-Cost Open-Source Bimanual Mobile Manipulator for Embodied AI
• AtomicVLA: Unlocking the Potential of Atomic Skill Learning in Robots
• NAC: Neural Action Codec for Vision-Language-Action Models

Source: arXiv cs.LG | 2026-10-06