Density estimation in representation space to predict model uncertainty
Not on the map. Why?
Abstract (arXiv)
Deep learning models frequently make incorrect predictions with high confidence when presented with test examples that are not well represented in their training dataset. We propose a novel and straightforward approach to estimate prediction uncertainty in a pre-trained neural network model. Our method estimates the training data density in representation space for a novel input. A neural network model then uses this information to determine whether we expect the pre-trained model to make a correct prediction. This uncertainty model is trained by predicting in-distribution errors, but can detect out-of-distribution data without having seen any such example. We test our method for a state-of-the art image classification model in the settings of both in-distribution uncertainty estimation as well as out-of-distribution detection.
Key facts. Cited by 27 papers in the map (data through 7 Oct 2026). First posted on arXiv 20 Aug 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 3; 2023: 7; 2024: 5; 2025: 8; 2026: 1.
27 papers.
Sort:
By area:
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2023Learning to Intervene on Concept Bottleneckscited by 29
- 2024A Comprehensive Survey on Evidential Deep Learning and Its Applicationscited by 25
- 2021Gradient-Based Quantification of Epistemic Uncertainty for Deep Object Detectorscited by 16
- 2023Better with Less: A Data-Active Perspective on Pre-Training Graph Neural Networkscited by 14
- 2023Evidential Uncertainty Quantification: A Variance-Based Perspectivecited by 14
- 2024A Structured Review of Literature on Uncertainty in Machine Learning & Deep Learningcited by 9
- 2023Concurrent Misclassification and Out-of-Distribution Detection for Semantic Segmentation via Energy-Based Normalizing Flowcited by 8
- 2025Probabilistic Machine Learning for Noisy Labels in Earth Observationcited by 6
- 2022What is Flagged in Uncertainty Quantification? Latent Density Models for Uncertainty Categorizationcited by 5
- 2025Uncertainty-Aware Deep Learning for Wildfire Danger Forecastingcited by 4
- 2022TEDL: A Two-stage Evidential Deep Learning Method for Classification Uncertainty Quantificationcited by 3
15 more
- 2022An Exploratory Study of AI System Risk Assessment from the Lens of Data Distribution and Uncertaintycited by 3
- 2024Accuracy-Preserving Calibration via Statistical Modeling on Probability Simplexcited by 3
- 2024Understanding the Dependence of Perception Model Competency on Regions in an Imagecited by 3
- 2025Automated Real-time Assessment of Intracranial Hemorrhage Detection AI Using an Ensembled Monitoring Model (EMM)cited by 3
- 2025Mimir: Hierarchical Goal-Driven Diffusion with Uncertainty Propagation for End-to-End Autonomous Drivingcited by 3
- 2023Generalized Uncertainty of Deep Neural Networks: Taxonomy and Applicationscited by 2
- 2023Delay-SDE-net: A deep learning approach for time series modelling with memory and uncertainty estimatescited by 2
- 2025Quantifying Uncertainty in the Presence of Distribution Shiftscited by 2
- 2025Towards Reliable Time Series Forecasting under Future Uncertainty: Ambiguity and Novelty Rejection Mechanismscited by 1
- 2021Graceful Degradation and Related Fieldscited by 0
- 2023Efficient IoT Inference via Context-Awarenesscited by 0
- 2024Expert-aware uncertainty estimation for quality control of neural-based blood typingcited by 0
- 2025A Framework for Uncertainty Quantification Based on Nearest Neighbors Across Layerscited by 0
- 2025A Machine Learning Perspective on Automated Driving Corner Casescited by 0
- 2026Revisiting Neural Activation Coverage for Uncertainty Estimationcited by 0
- 2026Revisiting Neural Activation Coverage for Uncertainty Estimation24 Apr 2026
- 2025Mimir: Hierarchical Goal-Driven Diffusion with Uncertainty Propagation for End-to-End Autonomous Driving8 Dec 2025
- 2025A Machine Learning Perspective on Automated Driving Corner Cases12 Oct 2025
- 2025Uncertainty-Aware Deep Learning for Wildfire Danger Forecasting29 Sep 2025
- 2025A Framework for Uncertainty Quantification Based on Nearest Neighbors Across Layers24 Jun 2025
- 2025Quantifying Uncertainty in the Presence of Distribution Shifts23 Jun 2025
- 2025Automated Real-time Assessment of Intracranial Hemorrhage Detection AI Using an Ensembled Monitoring Model (EMM)16 May 2025
- 2025Probabilistic Machine Learning for Noisy Labels in Earth Observation4 Apr 2025
- 2025Towards Reliable Time Series Forecasting under Future Uncertainty: Ambiguity and Novelty Rejection Mechanisms25 Mar 2025
- 2024A Comprehensive Survey on Evidential Deep Learning and Its Applications7 Sep 2024
- 2024Expert-aware uncertainty estimation for quality control of neural-based blood typing15 Jul 2024
- 2024Understanding the Dependence of Perception Model Competency on Regions in an Image15 Jul 2024
15 more
- 2024A Structured Review of Literature on Uncertainty in Machine Learning & Deep Learning1 Jun 2024
- 2024Accuracy-Preserving Calibration via Statistical Modeling on Probability Simplex21 Feb 2024
- 2023Evidential Uncertainty Quantification: A Variance-Based Perspective19 Nov 2023
- 2023Better with Less: A Data-Active Perspective on Pre-Training Graph Neural Networks2 Nov 2023
- 2023Efficient IoT Inference via Context-Awareness29 Oct 2023
- 2023Learning to Intervene on Concept Bottlenecks25 Aug 2023
- 2023Concurrent Misclassification and Out-of-Distribution Detection for Semantic Segmentation via Energy-Based Normalizing Flow16 May 2023
- 2023Delay-SDE-net: A deep learning approach for time series modelling with memory and uncertainty estimates14 Mar 2023
- 2023Generalized Uncertainty of Deep Neural Networks: Taxonomy and Applications2 Feb 2023
- 2022An Exploratory Study of AI System Risk Assessment from the Lens of Data Distribution and Uncertainty13 Dec 2022
- 2022TEDL: A Two-stage Evidential Deep Learning Method for Classification Uncertainty Quantification12 Sep 2022
- 2022What is Flagged in Uncertainty Quantification? Latent Density Models for Uncertainty Categorization11 Jul 2022
- 2021Gradient-Based Quantification of Epistemic Uncertainty for Deep Object Detectors9 Jul 2021
- 2021A Survey of Uncertainty in Deep Neural Networks7 Jul 2021
- 2021Graceful Degradation and Related Fields21 Jun 2021
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2023Learning to Intervene on Concept Bottleneckscited by 29
- 2024A Comprehensive Survey on Evidential Deep Learning and Its Applicationscited by 25
- 2021Gradient-Based Quantification of Epistemic Uncertainty for Deep Object Detectorscited by 16
- 2023Better with Less: A Data-Active Perspective on Pre-Training Graph Neural Networkscited by 14
- 2023Evidential Uncertainty Quantification: A Variance-Based Perspectivecited by 14
- 2024A Structured Review of Literature on Uncertainty in Machine Learning & Deep Learningcited by 9
- 2023Concurrent Misclassification and Out-of-Distribution Detection for Semantic Segmentation via Energy-Based Normalizing Flowcited by 8
- 2025Probabilistic Machine Learning for Noisy Labels in Earth Observationcited by 6
- 2022What is Flagged in Uncertainty Quantification? Latent Density Models for Uncertainty Categorizationcited by 5
- 2025Uncertainty-Aware Deep Learning for Wildfire Danger Forecastingcited by 4
- 2022TEDL: A Two-stage Evidential Deep Learning Method for Classification Uncertainty Quantificationcited by 3
12 most cited of 26 All 26 (JSON)
- 2026Revisiting Neural Activation Coverage for Uncertainty Estimation24 Apr 2026
- 2025A Machine Learning Perspective on Automated Driving Corner Cases12 Oct 2025
- 2025Uncertainty-Aware Deep Learning for Wildfire Danger Forecasting29 Sep 2025
- 2025A Framework for Uncertainty Quantification Based on Nearest Neighbors Across Layers24 Jun 2025
- 2025Quantifying Uncertainty in the Presence of Distribution Shifts23 Jun 2025
- 2025Automated Real-time Assessment of Intracranial Hemorrhage Detection AI Using an Ensembled Monitoring Model (EMM)16 May 2025
- 2025Probabilistic Machine Learning for Noisy Labels in Earth Observation4 Apr 2025
- 2025Towards Reliable Time Series Forecasting under Future Uncertainty: Ambiguity and Novelty Rejection Mechanisms25 Mar 2025
- 2024A Comprehensive Survey on Evidential Deep Learning and Its Applications7 Sep 2024
- 2024Expert-aware uncertainty estimation for quality control of neural-based blood typing15 Jul 2024
- 2024Understanding the Dependence of Perception Model Competency on Regions in an Image15 Jul 2024
- 2024A Structured Review of Literature on Uncertainty in Machine Learning & Deep Learning1 Jun 2024
12 newest of 26 All 26 (JSON)