arXiv:2410.21582cs.CVcs.AI2024-10

大模型微调后反而更脆弱,越大的预训练数据未必越好。

Large Pre-Training Datasets Don't Always Guarantee Robustness after Fine-Tuning

  • 用新任务微调会破坏大模型的泛化能力
  • 最大数据集训练的模型微调后鲁棒性下降最严重
  • 适合需要稳定性能的下游应用开发者关注

大规模预训练模型通过微调被用于学习特定任务,目标是在保持通用性能的同时获得新技能。一个关键目标是鲁棒性:在分布外(OOD)任务上的表现能力。我们评估了微调是否保留预训练模型的整体鲁棒性,发现基于大规模数据集训练的模型在微调后出现严重的灾难性遗忘和OOD泛化能力丧失。为此,我们提出了鲁棒性继承基准(ImageNet-RIB),该基准可应用于任意预训练模型,包含一组相关但不同的OOD下游任务,通过在一个任务上微调,再测试其余任务的表现。结果表明,尽管持续学习方法有一定帮助,但微调仍普遍降低鲁棒性。令人意外的是,基于最大且最多样化数据集(如LAION-2B)训练的模型,在小数据集上微调后表现出更大的鲁棒性损失和更低的绝对鲁棒性,相比在较小数据集上预训练的模型。这说明选择最强基础模型并非在专业任务上表现最优的必然路径。

原文摘要 · Abstract (English)

Large-scale pretrained models are widely leveraged as foundations for learning new specialized tasks via fine-tuning, with the goal of maintaining the general performance of the model while allowing it to gain new skills. A valuable goal for all such models is robustness: the ability to perform well on out-of-distribution (OOD) tasks. We assess whether fine-tuning preserves the overall robustness of the pretrained model, and observed that models pretrained on large datasets exhibited strong catastrophic forgetting and loss of OOD generalization. To systematically assess robustness preservation in fine-tuned models, we propose the Robustness Inheritance Benchmark (ImageNet-RIB). The benchmark, which can be applied to any pretrained model, consists of a set of related but distinct OOD (downstream) tasks and involves fine-tuning on one of the OOD tasks in the set then testing on the rest. We find that though continual learning methods help, fine-tuning reduces robustness across pretrained models. Surprisingly, models pretrained on the largest and most diverse datasets (e.g., LAION-2B) exhibit both larger robustness losses and lower absolute robustness after fine-tuning on small datasets, relative to models pretrained on smaller datasets. These findings suggest that starting with the strongest foundation model is not necessarily the best approach for performance on specialist tasks. https://jd730.github.io/projects/ImageNet-RIB

模型鲁棒性微调大模型

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