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

TL;DR

This paper investigates how fine-tuning large pretrained models affects their robustness, revealing that larger datasets do not always lead to better robustness preservation, and introduces a benchmark for systematic assessment.

Contribution

The paper introduces the Robustness Inheritance Benchmark (ImageNet-RIB) to evaluate robustness transfer during fine-tuning and demonstrates that larger pretraining datasets can lead to greater robustness loss.

Findings

Fine-tuning often reduces model robustness, even with continual learning methods.

Models pretrained on larger datasets like LAION-2B suffer more robustness loss.

Starting with the largest pretrained models may not be optimal for specialized tasks.

Abstract

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