arXiv:2504.10850cs.LGcs.CR2025-04

不修改预训练模型,用自编码器提升下游任务抗干扰能力

How to Enhance Downstream Adversarial Robustness (almost) without Touching the Pre-Trained Foundation Model?

  • 用鲁棒自编码器做数据预处理,不更新模型权重
  • 在ImageNet等数据集上使下游模型抗攻击能力提升10%-15%
  • 适合无法修改大模型的部署场景,如云服务接口

随着强大基础模型的兴起,预训练-微调范式日益流行:基础模型通过海量多源数据预训练,下游用户仅需微调适配具体任务。然而,由于对抗训练计算成本高,难以对基础模型进行微调以提升下游任务的鲁棒性。针对此挑战,本文提出在不访问或更新基础模型权重的前提下,提升下游鲁棒性的方法。受鲁棒性继承研究启发(Kim et al., 2020),通过理论分析发现鲁棒对比学习与监督学习中对抗鲁棒性存在紧密关联。为验证并利用这一洞察,设计了一种简单而有效的鲁棒自编码器作为数据预处理手段,在输入基础模型前净化特征。该方法训练阶段完全无需接触基础模型。大量实验表明,所提方法能显著提升下游任务的鲁棒性,验证了特征鲁棒性(由小的对抗对比损失体现)与下游任务鲁棒性之间的关联。

原文摘要 · Abstract (English)

With the rise of powerful foundation models, a pre-training-fine-tuning paradigm becomes increasingly popular these days: A foundation model is pre-trained using a huge amount of data from various sources, and then the downstream users only need to fine-tune and adapt it to specific downstream tasks. However, due to the high computation complexity of adversarial training, it is not feasible to fine-tune the foundation model to improve its robustness on the downstream task. Observing the above challenge, we want to improve the downstream robustness without updating/accessing the weights in the foundation model. Inspired from existing literature in robustness inheritance (Kim et al., 2020), through theoretical investigation, we identify a close relationship between robust contrastive learning with the adversarial robustness of supervised learning. To further validate and utilize this theoretical insight, we design a simple-yet-effective robust auto-encoder as a data pre-processing method before feeding the data into the foundation model. The proposed approach has zero access to the foundation model when training the robust auto-encoder. Extensive experiments demonstrate the effectiveness of the proposed method in improving the robustness of downstream tasks, verifying the connection between the feature robustness (implied by small adversarial contrastive loss) and the robustness of the downstream task.

对抗鲁棒性自编码器预训练模型数据预处理

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