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

Detecting Errors and Estimating Accuracy on Unlabeled Data with Self-training Ensembles

About

When a deep learning model is deployed in the wild, it can encounter test data drawn from distributions different from the training data distribution and suffer drop in performance. For safe deployment, it is essential to estimate the accuracy of the pre-trained model on the test data. However, the labels for the test inputs are usually not immediately available in practice, and obtaining them can be expensive. This observation leads to two challenging tasks: (1) unsupervised accuracy estimation, which aims to estimate the accuracy of a pre-trained classifier on a set of unlabeled test inputs; (2) error detection, which aims to identify mis-classified test inputs. In this paper, we propose a principled and practically effective framework that simultaneously addresses the two tasks. The proposed framework iteratively learns an ensemble of models to identify mis-classified data points and performs self-training to improve the ensemble with the identified points. Theoretical analysis demonstrates that our framework enjoys provable guarantees for both accuracy estimation and error detection under mild conditions readily satisfied by practical deep learning models. Along with the framework, we proposed and experimented with two instantiations and achieved state-of-the-art results on 59 tasks. For example, on iWildCam, one instantiation reduces the estimation error for unsupervised accuracy estimation by at least 70% and improves the F1 score for error detection by at least 4.7% compared to existing methods.

Jiefeng Chen, Frederick Liu, Besim Avci, Xi Wu, Yingyu Liang, Somesh Jha• 2021

Related benchmarks

TaskDatasetResultRank
OOD accuracy predictioniWildCam-WILDS
MAE0.7
16
Accuracy EstimationDigits
Absolute Estimation Error0.023
10
Accuracy EstimationOffice-31
Absolute Estimation Error0.018
10
Accuracy EstimationCIFAR10-C
Absolute Estimation Error2.2
10
Accuracy EstimationAmazon Review
Absolute Estimation Error0.018
10
Error detectionDigits
F1 Score88.1
10
Error detectionOffice-31
F1 Score79
10
Error detectionCIFAR10-C
F1 Score0.872
10
Error detectioniWILDCam
F1 Score80.9
10
Error detectionAmazon Review
F1 Score0.453
10
Showing 10 of 10 rows

Other info

Code

Follow for update