Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Agreement-on-the-Line: Predicting the Performance of Neural Networks under Distribution Shift

About

Recently, Miller et al. showed that a model's in-distribution (ID) accuracy has a strong linear correlation with its out-of-distribution (OOD) accuracy on several OOD benchmarks -- a phenomenon they dubbed ''accuracy-on-the-line''. While a useful tool for model selection (i.e., the model most likely to perform the best OOD is the one with highest ID accuracy), this fact does not help estimate the actual OOD performance of models without access to a labeled OOD validation set. In this paper, we show a similar but surprising phenomenon also holds for the agreement between pairs of neural network classifiers: whenever accuracy-on-the-line holds, we observe that the OOD agreement between the predictions of any two pairs of neural networks (with potentially different architectures) also observes a strong linear correlation with their ID agreement. Furthermore, we observe that the slope and bias of OOD vs ID agreement closely matches that of OOD vs ID accuracy. This phenomenon, which we call agreement-on-the-line, has important practical applications: without any labeled data, we can predict the OOD accuracy of classifiers}, since OOD agreement can be estimated with just unlabeled data. Our prediction algorithm outperforms previous methods both in shifts where agreement-on-the-line holds and, surprisingly, when accuracy is not on the line. This phenomenon also provides new insights into deep neural networks: unlike accuracy-on-the-line, agreement-on-the-line appears to only hold for neural network classifiers.

Christina Baek, Yiding Jiang, Aditi Raghunathan, Zico Kolter• 2022

Related benchmarks

TaskDatasetResultRank
OOD accuracy predictioniWildCam-WILDS
MAE4.95
16
Model Performance PredictionWILDS-RxRx1 (OOD-domains)
MAE0.0027
12
OOD accuracy predictionCIFAR-10.1 (test)
MAE (OOD Accuracy)1.11
6
OOD accuracy predictionCIFAR-10.2 (test)
MAE (%)3.93
6
OOD accuracy predictionCIFAR-10C Fog
MAE (%)1.45
6
OOD accuracy predictionCIFAR-10C Saturate
MAE (%)41
6
OOD accuracy predictionFMOW-WILDS
MAE0.013
6
OOD accuracy predictionCamelyon17-WILDS
MAE (%)5.47
6
OOD accuracy predictionImageNet V2 (test)
MAE (%)2.06
6
OOD accuracy predictionCIFAR-10C Snow
MAE (%)1.32
6
Showing 10 of 10 rows

Other info

Code

Follow for update