Approaching Test Time Augmentation in the Context of Uncertainty Calibration for Deep Neural Networks
Authors: Pedro Conde, Tiago Barros, Rui L. Lopes, Cristiano Premebida, Urbano J. Nunes
AI-generated Key Points
- Deep Neural Networks have led to widespread use of machine learning systems in real-world applications
- Reliable models with accurate predictions and calibrated uncertainties are critical for decision making processes
- Researchers propose a novel technique based on test time augmentation to improve uncertainty calibration of deep models for image classification
- The technique comes in two variations named M-ATTA and V-ATTA and leverages an adaptive weighting system to improve uncertainty calibration without affecting the model's accuracy
- M/V-ATTA improves uncertainty calibration by adapting weights during testing rather than training, unlike other test time augmentation approaches
- The team evaluated the performance of the technique using different metrics of uncertainty calibration on CIFAR-10, CIFAR-100, and Aerial Image Dataset benchmarks
- Empirical results indicate that M/V-ATTA outperforms state-of-the-art calibration techniques while maintaining baseline classification performance
- The code for M/V-ATTA is available on GitHub at https://github.com/pedrormconde/MV-ATTA
Abstract: With the rise of Deep Neural Networks, machine learning systems are nowadays ubiquitous in a number of real-world applications, which bears the need for highly reliable models. This requires a thorough look not only at the accuracy of such systems, but also to their predictive uncertainty. Hence, we propose a novel technique (with two different variations, named M-ATTA and V-ATTA) based on test time augmentation, to improve the uncertainty calibration of deep models for image classification. Unlike other test time augmentation approaches, M/V-ATTA improves uncertainty calibration without affecting the model's accuracy, by leveraging an adaptive weighting system. We evaluate the performance of the technique with respect to different metrics of uncertainty calibration. Empirical results, obtained on CIFAR-10, CIFAR-100, as well as on the benchmark Aerial Image Dataset, indicate that the proposed approach outperforms state-of-the-art calibration techniques, while maintaining the baseline classification performance. Code for M/V-ATTA available at: https://github.com/pedrormconde/MV-ATTA.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.