Red-Teaming of Vision-Language-Action Models via Quality Diversity Environment Generation
Organizations: Thomas Lord Department of Computer Science, University of Southern California. · Columbia University. · Adobe Research · Cornell University.
Abstract
A long-standing goal in robotics is to develop generalist robots that can perform tasks across a wide range of settings, flexibly specified with images and text. Vision-Language-Action (VLA) models offer a promising path toward this goal by training on large collections of robot demonstrations. However, while VLAs perform well in-distribution, they often fail under minor test-time shifts, such as changes in object positions, lighting, textures, and camera poses. To address this, we propose Quality Diversity for Diverse Environment Generation (Q-DEG), a method for improving VLA robustness by generating adversarial environments and fine-tuning the VLA on its own successful rollouts. Q-DEG uses Quality Diversity (QD) optimization to find a set of high-failure environments, while promoting diversity using policy embeddings extracted from the VLA. These environments expose vulnerabilities that can be mitigated through self-imitation fine-tuning, thereby improving the VLA's robustness. In simulation, we conduct experiments across three representative VLA models, and our results indicate that Q-DEG improves robustness to out-of-distribution (OOD) adversarial environments compared to the base VLA and VLAs fine-tuned using data collected from two red-teaming baselines. Furthermore, we demonstrate that improving VLA robustness through Q-DEG facilitates sim-to-real transfer. Our anonymous project website is at: https://qd-env.github.io/qdeg/.
Figures & tables
| -FAST | OpenVLA-OFT | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Eval. set | base | rand | CMA-ES | Q-DEG | base | rand | CMA-ES | Q-DEG | base | rand | CMA-ES | Q-DEG |
| default | 96.80 | 68.60 | 82.60 | 89.60 | 96.20 | 94.80 | 86.70 | 93.80 | 98.40 | 85.60 | 93.60 | 95.60 |
| rand | 29.15 | 39.40 | 37.79 | 48.73 | 56.25 | 82.29 | 68.75 | 78.13 | 77.38 | 77.98 | 83.33 | 88.09 |
| CMA-ES | 27.90 | 38.46 | 24.81 | 29.82 | 45.75 | 52.52 | 48.27 | 75.79 | 54.98 | 49.39 | 69.17 | 58.00 |
| Q-DEG | 54.83 | 51.28 | 36.89 | 56.13 | 47.19 | 50.77 | 55.61 | 69.13 | 38.03 | 61.82 | 70.45 | 81.81 |
| average | 52.17 | 49.44 | 45.52 | 56.07 | 61.35 | 70.10 | 64.83 | 79.21 | 67.20 | 68.70 | 79.14 | 80.88 |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Robustness of VLA Models
Robustness of vision-language-action (VLA) models concerns reliable robot behavior under changes to language, vision, sensors, tasks, and hardware.
See all 221 papers