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

SURE: SUrvey REcipes for building reliable and robust deep networks

About

In this paper, we revisit techniques for uncertainty estimation within deep neural networks and consolidate a suite of techniques to enhance their reliability. Our investigation reveals that an integrated application of diverse techniques--spanning model regularization, classifier and optimization--substantially improves the accuracy of uncertainty predictions in image classification tasks. The synergistic effect of these techniques culminates in our novel SURE approach. We rigorously evaluate SURE against the benchmark of failure prediction, a critical testbed for uncertainty estimation efficacy. Our results showcase a consistently better performance than models that individually deploy each technique, across various datasets and model architectures. When applied to real-world challenges, such as data corruption, label noise, and long-tailed class distribution, SURE exhibits remarkable robustness, delivering results that are superior or on par with current state-of-the-art specialized methods. Particularly on Animal-10N and Food-101N for learning with noisy labels, SURE achieves state-of-the-art performance without any task-specific adjustments. This work not only sets a new benchmark for robust uncertainty estimation but also paves the way for its application in diverse, real-world scenarios where reliability is paramount. Our code is available at \url{https://yutingli0606.github.io/SURE/}.

Yuting Li, Yingyi Chen, Xuanlong Yu, Dexiong Chen, Xi Shen• 2024

Related benchmarks

TaskDatasetResultRank
Failure PredictionCIFAR10-LT IF=10 (test)
Accuracy93.91
28
Failure PredictionCIFAR100-LT IF=10 (test)
Acc0.7092
28
Learning with noisy labelsFood-101N noise ratio ~20% (test)
Top-1 Test Accuracy88
9
Classification with noisy labelsAnimal-10N noise ratio ~8% (test)
Top-1 Acc89
8
Showing 4 of 4 rows

Other info

Code

Follow for update