arXiv:2504.08906cs.CVcs.AI2025-04AAAI被引 8

提升视觉大模型SAM在对抗攻击下的鲁棒性,兼顾准确率与防御效果

Robust SAM: On the Adversarial Robustness of Vision Foundation Models

  • 设计跨提示攻击方法,增强攻击在不同提示类型间的迁移能力
  • 仅调整512个参数,使mIoU在对抗攻击下提升至少15%
  • 基于SVD的轻量级防御策略,平衡鲁棒性与原始性能

Segment Anything Model (SAM) 是一个广泛应用的视觉基础模型,涵盖图像分割、检测和跟踪等多种任务。鉴于其广泛的应用场景,研究其在对抗攻击下的鲁棒性对实际部署至关重要。然而,当前关于SAM鲁棒性的研究仍处于起步阶段。现有攻击方法常忽视提示(prompt)的作用,且缺乏有效的防御手段以兼顾鲁棒性与精度。为此,本文提出一个针对SAM的对抗鲁棒性评估与增强框架。具体而言,引入一种跨提示攻击方法,以提升攻击在不同提示类型间的迁移能力;同时提出一种少参数适应策略,用于防御多种对抗攻击。为平衡鲁棒性与精度,采用奇异值分解(SVD)约束可训练参数空间,仅允许奇异值可调。实验表明,所提跨提示攻击方法在SAM和SAM 2上均优于以往方法,攻击成功率更高。通过仅适配512个参数,该方法在各类对抗攻击下实现至少15%的平均交并比(mIoU)提升。相比已有防御方法,本方案在增强鲁棒性的同时最大限度保留了模型原有性能。

原文摘要 · Abstract (English)

The Segment Anything Model (SAM) is a widely used vision foundation model with diverse applications, including image segmentation, detection, and tracking. Given SAM's wide applications, understanding its robustness against adversarial attacks is crucial for real-world deployment. However, research on SAM's robustness is still in its early stages. Existing attacks often overlook the role of prompts in evaluating SAM's robustness, and there has been insufficient exploration of defense methods to balance the robustness and accuracy. To address these gaps, this paper proposes an adversarial robustness framework designed to evaluate and enhance the robustness of SAM. Specifically, we introduce a cross-prompt attack method to enhance the attack transferability across different prompt types. Besides attacking, we propose a few-parameter adaptation strategy to defend SAM against various adversarial attacks. To balance robustness and accuracy, we use the singular value decomposition (SVD) to constrain the space of trainable parameters, where only singular values are adaptable. Experiments demonstrate that our cross-prompt attack method outperforms previous approaches in terms of attack success rate on both SAM and SAM 2. By adapting only 512 parameters, we achieve at least a 15\% improvement in mean intersection over union (mIoU) against various adversarial attacks. Compared to previous defense methods, our approach enhances the robustness of SAM while maximally maintaining its original performance.

视觉模型对抗鲁棒性SAM轻量防御

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。