arXiv:2506.22602cs.LGstat.ML2025-06

用快速方法提升抗攻击迁移学习效率,稳定且不损失性能。

Are Fast Methods Stable in Adversarially Robust Transfer Learning?

  • 用FGSM替代传统方法进行对抗微调,计算量大幅降低。
  • 在ε=4、8时无灾难性过拟合,ε=32下线性探测仍稳定。
  • 速度提升4倍,鲁棒性损失仅0.39%~1.39%,适合高效训练场景。

迁移学习常用于降低模型训练的计算成本,通过微调使下游任务利用预训练数据中学到的特征,并快速适应新任务。这在实现对抗鲁棒性方面尤其重要,因为从头训练对抗模型成本极高。然而,高鲁棒性仍需在微调阶段进行对抗训练,耗时比标准微调多一个数量级。本文重新考察快速梯度符号法(FGSM)在鲁棒迁移学习中的应用,以降低对抗微调的计算开销。结果发现,与从头训练相比,FGSM在对抗微调中表现更稳定:在标准扰动范围ε=4或ε=8下未出现灾难性过拟合。该稳定性在参数高效微调方法中进一步增强,线性探测下甚至可维持至ε=32。实验表明,相较于常用方法投影梯度下降(PGD),FGSM在ε=4和ε=8时平均仅损失0.39%和1.39%的测试鲁棒性,但训练时间减少4倍。

原文摘要 · Abstract (English)

Transfer learning is often used to decrease the computational cost of model training, as fine-tuning a model allows a downstream task to leverage the features learned from the pre-training dataset and quickly adapt them to a new task. This is particularly useful for achieving adversarial robustness, as adversarially training models from scratch is very computationally expensive. However, high robustness in transfer learning still requires adversarial training during the fine-tuning phase, which requires up to an order of magnitude more time than standard fine-tuning. In this work, we revisit the use of the fast gradient sign method (FGSM) in robust transfer learning to improve the computational cost of adversarial fine-tuning. We surprisingly find that FGSM is much more stable in adversarial fine-tuning than when training from scratch. In particular, FGSM fine-tuning does not suffer from any issues with catastrophic overfitting at standard perturbation budgets of $\varepsilon=4$ or $\varepsilon=8$. This stability is further enhanced with parameter-efficient fine-tuning methods, where FGSM remains stable even up to $\varepsilon=32$ for linear probing. We demonstrate how this stability translates into performance across multiple datasets. Compared to fine-tuning with the more commonly used method of projected gradient descent (PGD), on average, FGSM only loses 0.39% and 1.39% test robustness for $\varepsilon=4$ and $\varepsilon=8$ while using $4\times$ less training time. Surprisingly, FGSM may not only be a significantly more efficient alternative to PGD in adversarially robust transfer learning but also a well-performing one.

对抗鲁棒迁移学习FGSM高效训练

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