用渐进扰动策略让普通模型也能稳定练出抗干扰能力
Robust Fine-Tuning from Non-Robust Pretrained Models: Mitigating Suboptimal Transfer With Epsilon-Scheduling
- 训练时逐步增加扰动强度,避免初始阶段过度追求抗干扰导致任务学不好
- 在多个数据集上验证,新方法显著提升模型在各种干扰下的平均表现
- 适合想用现成预训练模型快速实现鲁棒性的开发者或研究者
微调预训练模型是现代机器学习的标准流程。然而,同时实现下游任务适应性和对抗样本鲁棒性的鲁棒微调(RFT)仍具挑战性。尽管开源中存在大量非鲁棒预训练模型,其在鲁棒微调中的潜力尚未被充分理解。我们系统研究了从这些非鲁棒模型出发的鲁棒微调,发现:即使使用小扰动进行鲁棒目标微调,也可能导致性能严重下降,称为次优迁移。在困难任务或高扰动场景下,性能可能低至可视为迁移失败。我们发现,直接使用鲁棒目标会阻碍训练初期的任务适配,最终影响最优迁移。为此提出新方法Epsilon-Scheduling——在训练过程中逐步增加扰动强度,有效促进最优迁移。此外引入期望鲁棒性作为评估指标,全面衡量模型在多种扰动下的表现。在六种预训练模型和五个数据集上的大量实验表明,Epsilon-Scheduling能有效防止次优迁移,并持续提升期望鲁棒性。
原文摘要 · Abstract (English)
Fine-tuning pretrained models is a standard and effective workflow in modern machine learning. However, robust fine-tuning (RFT), which aims to simultaneously achieve adaptation to a downstream task and robustness to adversarial examples, remains challenging. Despite the abundance of non-robust pretrained models in open-source repositories, their potential for RFT is less understood. We address this knowledge gap by systematically examining RFT from such non-robust models. Our experiments reveal that fine-tuning non-robust models with a robust objective, even under small perturbations, can lead to poor performance, a phenomenon that we dub suboptimal transfer. In challenging scenarios (eg, difficult tasks, high perturbation), the resulting performance can be so low that it may be considered a transfer failure. We find that fine-tuning using a robust objective impedes task adaptation at the beginning of training and eventually prevents optimal transfer. However, we propose a novel heuristic, Epsilon-Scheduling, a schedule over perturbation strength used during training that promotes optimal transfer. Additionally, we introduce expected robustness, a metric that captures performance across a range of perturbations, providing a more comprehensive evaluation of the accuracy-robustness trade-off for diverse models at test time. Extensive experiments on a wide range of configurations (six pretrained models and five datasets) show that Epsilon-Scheduling successfully prevents suboptimal transfer and consistently improves expected robustness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。