用专家混合模型提升视觉语言模型的抗干扰能力
TAME: Test-Time Adversarial Prompt Tuning via Mixture-of-Experts for Vision-Language Models

- 测试时动态组合多个提示专家,自适应防御对抗扰动
- 在11个数据集上使CLIP抗攻击能力提升49.1%以上
- 无需微调,适合部署在开放场景的模型安全增强
大规模预训练视觉语言模型(如CLIP)具备强大的零样本泛化能力,但对难以察觉的对抗扰动极为敏感,威胁其在开放世界中的应用安全。为在不进行下游任务微调的前提下提升鲁棒性,我们提出TAME,一种新型测试时防御方法。基于先前的TAPT,TAME通过引入输入相关的专家混合(MoE)架构,将单个可调提示替换为可动态组合的专家提示集合,实现更强表达力与自适应性。具体而言,TAME维护一组可学习的专家提示,并利用依赖输入的路由机制,在推理时为每个未标记测试样本聚合定制化的提示组合。该防御机制由三个无监督目标驱动:(1) 多视图预测熵最小化;(2) 视觉令牌统计量与预先计算的干净/对抗参考分布的分层对齐;(3) 对MoE结构的正则化以平衡专家使用并保持提示多样性。我们在11个基准数据集(包括ImageNet和10个零样本数据集)上评估TAME,结果表明,TAME在AutoAttack下使原始CLIP的零样本抗攻击能力提升至少49.1%,同时基本保留了在干净样本上的泛化性能。TAME在多种提示设计下均优于现有对抗提示调优方法,平均鲁棒性提升达30.2%以上。
原文摘要 · Abstract (English)
Large-scale pre-trained Vision-Language models (VLMs), such as CLIP, exhibit strong zero-shot generalization, yet remain highly vulnerable to imperceptible adversarial perturbations, raising serious safety concerns for open-world deployment. To enhance robustness without requiring downstream task-specific retraining, we propose TAME, a novel test-time defense. Building upon our prior Test-Time Adversarial Prompt Tuning (TAPT), TAME introduces an architectural reformulation by replacing TAPT's single adaptive prompt with an input-conditioned Mixture-of-Experts (MoE) framework, enabling more expressive and adaptive defense. Specifically, TAME maintains a bank of learnable expert prompts and employs an input-dependent routing mechanism to aggregate a customized prompt mixture for each unlabeled test sample at inference time. This test-time defense mechanism is driven by three unsupervised objectives: (1) multi-view prediction entropy minimization, (2) layer-wise alignment of visual token statistics to precomputed clean and adversarial reference distributions, and (3) MoE regularization for balanced expert utilization and prompt diversity. We evaluated TAME on 11 benchmark datasets, including ImageNet and 10 additional zero-shot datasets. The results show that TAME improves the zero-shot adversarial robustness of the original CLIP by at least 49.1% under AutoAttack while largely preserving generalization on clean samples. TAME also consistently outperforms existing adversarial prompt tuning methods across multiple prompt designs, yielding an average robustness gain of at least 30.2%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。