Evolution-based Region Adversarial Prompt Learning for Robustness Enhancement in Vision-Language Models
Xiaojun Jia, Sensen Gao, Simeng Qin, Ke Ma, Xinfeng Li, Yihao Huang, Wei Dong, Yang Liu, Xiaochun Cao
TL;DR
This work tackles the vulnerability of large vision-language models to adversarial examples by introducing ER-APT, an evolution-based region adversarial prompt learning framework. It merges gradient-based adversarial generation with a genetic-evolution refinement to produce diverse, region-focused perturbations for few-shot prompt tuning, complemented by a dynamic loss weighting scheme that adaptively balances clean accuracy and robustness. The approach achieves state-of-the-art robustness and generalization across multiple datasets and settings, including cross-dataset and AutoAttack evaluations, with ablations confirming the critical roles of the evolutionary region and adaptive loss weighting. Although it incurs additional training cost, ER-APT provides a practical and scalable route to robust VLM deployment and transferability across tasks and data distributions.
Abstract
Large pre-trained vision-language models (VLMs), such as CLIP, demonstrate impressive generalization but remain highly vulnerable to adversarial examples (AEs). Previous work has explored robust text prompts through adversarial training, achieving some improvement in both robustness and generalization. However, they primarily rely on singlegradient direction perturbations (e.g., PGD) to generate AEs, which lack diversity, resulting in limited improvement in adversarial robustness. To address these limitations, we propose an evolution-based region adversarial prompt tuning method called ER-APT, which combines gradient methods with genetic evolution to generate more diverse and challenging AEs. In each training iteration, we first generate AEs using traditional gradient-based methods. Subsequently, a genetic evolution mechanism incorporating selection, mutation, and crossover is applied to optimize the AEs, ensuring a broader and more aggressive perturbation distribution.The final evolved AEs are used for prompt tuning, achieving region-based adversarial optimization instead of conventional single-point adversarial prompt tuning. We also propose a dynamic loss weighting method to adjust prompt learning efficiency for accuracy and robustness. Experimental evaluations on various benchmark datasets demonstrate the superiority of our proposed method, outperforming stateof-the-art APT methods. The code is released at https://github.com/jiaxiaojunQAQ/ER-APT.
