arXiv:2604.12780cs.CVcs.AI2026-04被引 1

只微调6%参数,实现接近全模型对抗训练的鲁棒性。

Efficient Adversarial Training via Criticality-Aware Fine-Tuning

论文配图:Efficient Adversarial Training via Criticality-Aware Fine-Tuning
图 1 · 摘自论文原文
  • 根据关键性动态选择需微调的模块,聚焦提升鲁棒性的核心参数。
  • 在大ViT模型上仅调6%参数,对抗鲁棒性损失仅4.3%。
  • 适合资源受限场景,尤其适用于大规模视觉Transformer的高效防御。

视觉变换器(ViT)在多种视觉任务中表现优异,其可扩展性使其在大数据集上具备强泛化能力。然而,随着参数量增加,其对对抗样本的鲁棒性并未同比提升。对抗训练(AT)虽是增强鲁棒性的有效方法,但通常需微调整个模型,计算成本高昂,尤其对大型ViT架构。本文提出关键性感知对抗训练(CAAT),仅微调一小部分关键参数即可达到与标准AT相当的鲁棒性。CAAT通过自适应识别对鲁棒性贡献最大的参数,结合参数高效微调(PEFT)技术,仅对超过阈值的关键权重矩阵进行调整。实验表明,与纯对抗训练相比,CAAT仅损失4.3%的对抗鲁棒性,却仅需微调约6%的参数。在三个主流对抗学习数据集上的广泛实验显示,CAAT以更少可训练参数超越现有轻量化对抗训练方法。

原文摘要 · Abstract (English)

Vision Transformer (ViT) models have achieved remarkable performance across various vision tasks, with scalability being a key advantage when applied to large datasets. This scalability enables ViT models to exhibit strong generalization capabilities. However, as the number of parameters increases, the robustness of ViT models to adversarial examples does not scale proportionally. Adversarial training (AT), one of the most effective methods for enhancing robustness, typically requires fine-tuning the entire model, leading to prohibitively high computational costs, especially for large ViT architectures. In this paper, we aim to robustly fine-tune only a small subset of parameters to achieve robustness comparable to standard AT. To accomplish this, we introduce Criticality-Aware Adversarial Training (CAAT), a novel method that adaptively allocates resources to the most robustness-critical parameters, fine-tuning only selected modules. Specifically, CAAT efficiently identifies parameters that contribute most to adversarial robustness. It then leverages parameter-efficient fine-tuning (PEFT) to robustly adjust weight matrices where the number of critical parameters exceeds a predefined threshold. CAAT exhibits favorable generalization when scaled to larger vision transformer architectures, potentially paving the way for adversarial training at scale, e.g, compared with plain adversarial training, CAAT incurs only a 4.3% decrease in adversarial robustness while tuning approximately 6% of its parameters. Extensive experiments on three widely used adversarial learning datasets demonstrate that CAAT outperforms state-of-the-art lightweight AT methods with fewer trainable parameters.

对抗训练视觉Transformer参数高效

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