The impact of domain shift on the calibration of fine-tuned modelsDownload PDF

09 Oct 2021, 14:49 (modified: 09 Dec 2021, 20:40)NeurIPS 2021 Workshop DistShift PosterReaders: Everyone
Keywords: Transfer Learning, Calibration Error, Domain Shift
TL;DR: The impact of domain shift on the calibration of fine-tuned models
Abstract: Transfer learning has become a standard technique in computer vision and natural language processing thanks to the fact that it often substantially improves performance on downstream tasks. Recent work by Hendrycks et al. demonstrated that using a pre-trained model can also significantly improve a model's calibration, i.e. how well the model's confidence estimates correspond to the probability of its prediction being correct. In this paper, we provide some nuance to the claim that pre-training improves calibration by demonstrating that this beneficial effect diminishes when there is a domain shift between the pre-training and fine-tuning tasks.
1 Reply

Loading