通过函数空间正则化,同时提升模型在下游任务的性能和分布外鲁棒性。
Feed Two Birds with One Scone: Exploiting Function-Space Regularization for Both OOD Robustness and ID Fine-Tuning Performance
- 在函数空间中用模拟分布外样本约束微调过程,保持预训练模型的鲁棒性。
- 在多种CLIP骨干网络上,同时显著提升下游任务精度与分布外鲁棒性。
- 适合追求高鲁棒性和高性能兼得的模型微调场景。
鲁棒微调旨在实现下游任务中的良好分布内(ID)性能,同时保持预训练模型在分布外(OOD)情况下的鲁棒性。现有方法通常通过保留预训练权重、特征或输出来实现,但我们发现这些方法对不同模型架构并不总能提升OOD鲁棒性。这是因为OOD鲁棒性要求模型函数对下游任务输入产生稳定预测,而现有方法在函数空间优化上可能作为不良代理。为此,我们提出一种新正则化策略,在函数空间中利用模拟的OOD样本约束微调过程与预训练模型之间的距离,以保持预训练模型的OOD鲁棒性。此外,为进一步增强微调模型的OOD鲁棒性,引入一致性正则化,促使模型对扰动样本产生稳定预测。大量实验表明,该方法在多种CLIP骨干网络上可一致提升下游任务的ID微调性能与OOD鲁棒性,优于现有的基于正则化的鲁棒微调方法。
原文摘要 · Abstract (English)
Robust fine-tuning aims to achieve competitive in-distribution (ID) performance while maintaining the out-of-distribution (OOD) robustness of a pre-trained model when transferring it to a downstream task. To remedy this, most robust fine-tuning methods aim to preserve the pretrained weights, features, or logits. However, we find that these methods cannot always improve OOD robustness for different model architectures. This is due to the OOD robustness requiring the model function to produce stable prediction for input information of downstream tasks, while existing methods might serve as a poor proxy for the optimization in the function space. Based on this finding, we propose a novel regularization that constrains the distance of fine-tuning and pre-trained model in the function space with the simulated OOD samples, aiming to preserve the OOD robustness of the pre-trained model. Besides, to further enhance the OOD robustness capability of the fine-tuning model, we introduce an additional consistency regularization to promote stable predictions of perturbed samples. Extensive experiments demonstrate our approach could consistently improve both downstream task ID fine-tuning performance and OOD robustness across a variety of CLIP backbones, outperforming existing regularization-based robust fine-tuning methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。