Evaluating and Calibrating Uncertainty Prediction in Regression Tasks
Not on the map. Why?
Abstract (arXiv)
Predicting not only the target but also an accurate measure of uncertainty is important for many machine learning applications and in particular safety-critical ones. In this work we study the calibration of uncertainty prediction for regression tasks which often arise in real-world systems. We show that the existing definition for calibration of a regression uncertainty [Kuleshov et al. 2018] has severe limitations in distinguishing informative from non-informative uncertainty predictions. We propose a new definition that escapes this caveat and an evaluation method using a simple histogram-based approach. Our method clusters examples with similar uncertainty prediction and compares the prediction with the empirical uncertainty on these examples. We also propose a simple, scaling-based calibration method that preforms as well as much more complex ones. We show results on both a synthetic, controlled problem and on the object detection bounding-box regression task using the COCO and KITTI datasets.
Key facts. Cited by 87 papers in the map (data through 14 Oct 2026). First posted on arXiv 28 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 5; 2023: 27; 2024: 12; 2025: 11; 2026: 20.
87 papers.
Sort:
By area:
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 680
- 2021How to Certify Machine Learning Based Safety-critical Systems? A Systematic Literature Reviewcited by 39
- 2022NeuralUQ: A comprehensive library for uncertainty quantification in neural differential equations and operatorscited by 33
- 2021Deep Ensembles from a Bayesian Perspectivecited by 27
- 2022Sample Efficient Deep Reinforcement Learning via Uncertainty Estimationcited by 27
- 2022BayesCap: Bayesian Identity Cap for Calibrated Uncertainty in Frozen Neural Networkscited by 23
- 2021Calibrated Uncertainty for Molecular Property Prediction using Ensembles of Message Passing Neural Networkscited by 21
- 2023Towards Building Self-Aware Object Detectors via Reliable Uncertainty Quantification and Calibrationcited by 19
- 2021Recalibration of Aleatoric and Epistemic Regression Uncertainty in Medical Imagingcited by 18
- 2023URL: A Representation Learning Benchmark for Transferable Uncertainty Estimatescited by 17
- 2023How Reliable is Your Regression Model's Uncertainty Under Real-World Distribution Shifts?cited by 14
- 2023Calibration in Machine Learning Uncertainty Quantification: beyond consistency to target adaptivitycited by 13
38 more
- 2026BAGEN: Are LLM Agents Budget-Aware?cited by 13
- 2024Kermut: Composite kernel regression for protein variant effectscited by 12
- 2021Robust Monocular Localization in Sparse HD Maps Leveraging Multi-Task Uncertainty Estimationcited by 11
- 2023Evidential Deep Learning: Enhancing Predictive Uncertainty Estimation for Earth System Science Applicationscited by 11
- 2024Diffusion MRI with Machine Learningcited by 11
- 2023Graph Neural Network Interatomic Potential Ensembles with Calibrated Aleatoric and Epistemic Uncertainty on Energy and Forcescited by 10
- 2023Understanding Pathologies of Deep Heteroskedastic Regressioncited by 10
- 2022Parametric and Multivariate Uncertainty Calibration for Regression and Object Detectioncited by 9
- 2023A Holistic Assessment of the Reliability of Machine Learning Systemscited by 9
- 2023Clarifying Trust of Materials Property Predictions using Neural Networks with Distribution-Specific Uncertainty Quantificationcited by 8
- 2023Properties of the ENCE and other MAD-based calibration metricscited by 7
- 2023Uncertainty Quantification in Machine Learning for Biosignal Applications -- A Reviewcited by 7
- 2021Reliable Neural Networks for Regression Uncertainty Estimationcited by 6
- 2023Stratification of uncertainties recalibrated by isotonic regression and its impact on calibration error statisticscited by 6
- 2023Validation of uncertainty quantification metrics: a primer based on the consistency and adaptivity conceptscited by 5
- 2023Conformalizing Machine Translation Evaluationcited by 5
- 2023Tailoring Mixup to Data for Calibrationcited by 5
- 2024A Survey on Lexical Ambiguity Detection and Word Sense Disambiguationcited by 5
- 2024Enhancing Uncertainty Quantification in Drug Discovery with Censored Regression Labelscited by 5
- 2024SAUC: Sparsity-Aware Uncertainty Calibration for Spatiotemporal Prediction with Graph Neural Networkscited by 5
- 2021A framework for benchmarking uncertainty in deep regressioncited by 4
- 2023Can bin-wise scaling improve consistency and adaptivity of prediction uncertainty for machine learning regression ?cited by 4
- 2024Multi-fidelity Gaussian process surrogate modeling for regression problems in physicscited by 4
- 2024Navigation under uncertainty: Trajectory prediction and occlusion reasoning with switching dynamical systemscited by 4
- 2021f-Cal: Calibrated aleatoric uncertainty estimation from neural networks for robot perceptioncited by 3
- 2023Likelihood Annealing: Fast Calibrated Uncertainty for Regressioncited by 3
- 2023Towards Reliable Uncertainty Quantification via Deep Ensembles in Multi-output Regression Taskcited by 3
- 2024denoiSplit: a method for joint microscopy image splitting and unsupervised denoisingcited by 3
- 2024Improved Uncertainty Estimation of Graph Neural Network Potentials Using Engineered Latent Space Distancescited by 3
- 2025Uncertainty quantification with approximate variational learning for wearable photoplethysmography prediction taskscited by 3
- 2021Evaluating Predictive Uncertainty and Robustness to Distributional Shift Using Real World Datacited by 2
- 2022A Robust Learning Methodology for Uncertainty-aware Scientific Machine Learning modelscited by 2
- 2023Surrogate uncertainty estimation for your time series forecasting black-box: learn when to trustcited by 2
- 2023A General Framework for Uncertainty Quantification via Neural SDE-RNNcited by 2
- 2023Machine-Learning Solutions for the Analysis of Single-Particle Diffusion Trajectoriescited by 2
- 2023HRTF Interpolation using a Spherical Neural Process Meta-Learnercited by 2
- 2025CLUE: Neural Networks Calibration via Learning Uncertainty-Error alignmentcited by 2
- 2025LFaB: Low fidelity as Bias for Active Learning in the chemical configuration spacecited by 2
50 most cited of 87 All 87 (JSON)
- 2026Observation Quality in DSM-Assisted Aerial Localization: A Benchmark and Base-Anchored Uncertainty Calibration13 Oct 2026
- 2026UQ-LOB: Uncertainty-Aware Limit Order Book Mid-Price Forecasting25 Sep 2026
- 2026Faithfulness Without Anchors: A Position Paper on the Structural Misalignments of Explainable AI Faithfulness Metrics for Feature Importance24 Sep 2026
- 2026A Calibration Audit of Confidence in Feed-Forward 3D Reconstruction Models30 Aug 2026
- 2026Information-Based Calibration of Uncertainty Quantification in Product-of-Experts Gaussian Process Models29 Aug 2026
- 2026Structure-preserving uncertainty quantification for GENERIC dynamics12 Aug 2026
- 2026From Keypoints to Predictive Distributions: Post-Hoc Uncertainty for YOLO-Pose Models29 Jul 2026
- 2026Probabilistic Physics-Aware Machine Learning Predictions of Electric Truck Energy Consumption with Field Data21 Jul 2026
- 2026When Variance Is Not an Error Map: Calibrated Uncertainty for Radiative Gaussian Splatting in Sparse-View CT15 Jul 2026
- 2026Exact and Calibrated Diffusion Reconstruction for Digital Breast Tomosynthesis14 Jul 2026
- 2026From Pixels to Newtons: Predicting In Vivo Joint Contact Forces from Monocular Video4 Jun 2026
- 2026BAGEN: Are LLM Agents Budget-Aware?29 May 2026
38 more
- 2026Reinforcement Learning over Predictive Distributions for LLM Regression20 May 2026
- 2026Uncertainty Reliability Under Domain Shift: An Investigation for Data-Driven Blood Pressure Estimation in Photoplethysmography18 May 2026
- 2026CredibleDFGO: Differentiable Factor Graph Optimization with Credibility Supervision7 May 2026
- 2026Recipes for Calibration Checks in Safety-Critical Applications29 Apr 2026
- 2026One-Shot Localisation of the Global Minimum of a Noisy One-Dimensional Function: An Iterative Neural Minimizer Compared with Set Transformers and Classical Estimators4 Apr 2026
- 2026Aligning Recommendations with User Popularity Preferences1 Apr 2026
- 2026Practical Deep Heteroskedastic Regression2 Mar 2026
- 2026How Uncertain Is the Grade? A Benchmark of Uncertainty Metrics for LLM-Based Automatic Assessment17 Feb 2026
- 2025On the Effect of Regularization on Nonparametric Mean-Variance Regression27 Nov 2025
- 2025Solar-GECO: Perovskite Solar Cell Property Prediction with Geometric-Aware Co-Attention24 Nov 2025
- 2025A systematic evaluation of uncertainty quantification techniques in deep learning: a case study in photoplethysmography signal analysis31 Oct 2025
- 2025On Uncertainty Calibration for Equivariant Functions24 Oct 2025
- 2025Post-Hoc Split-Point Self-Consistency Verification for Efficient, Unified Quantification of Aleatoric and Epistemic Uncertainty in Deep Learning16 Sep 2025
- 2025Evaluating the Quality of the Quantified Uncertainty for (Re)Calibration of Data-Driven Regression Models25 Aug 2025
- 2025LFaB: Low fidelity as Bias for Active Learning in the chemical configuration space21 Aug 2025
- 2025A Simple Approximate Bayesian Inference Neural Surrogate for Stochastic Petri Net Models14 Jul 2025
- 2025CLUE: Neural Networks Calibration via Learning Uncertainty-Error alignment28 May 2025
- 2025Uncertainty quantification with approximate variational learning for wearable photoplethysmography prediction tasks16 May 2025
- 2025Do you understand epistemic uncertainty? Think again! Rigorous frequentist epistemic uncertainty estimation in regression17 Mar 2025
- 2024Navigation under uncertainty: Trajectory prediction and occlusion reasoning with switching dynamical systems14 Oct 2024
- 2024SAUC: Sparsity-Aware Uncertainty Calibration for Spatiotemporal Prediction with Graph Neural Networks13 Sep 2024
- 2024Enhancing Uncertainty Quantification in Drug Discovery with Censored Regression Labels6 Sep 2024
- 2024Improved Uncertainty Estimation of Graph Neural Network Potentials Using Engineered Latent Space Distances15 Jul 2024
- 2024Multi-fidelity Gaussian process surrogate modeling for regression problems in physics18 Apr 2024
- 2024Kermut: Composite kernel regression for protein variant effects9 Apr 2024
- 2024A Survey on Lexical Ambiguity Detection and Word Sense Disambiguation24 Mar 2024
- 2024denoiSplit: a method for joint microscopy image splitting and unsupervised denoising18 Mar 2024
- 2024Validation of ML-UQ calibration statistics using simulated reference values: a sensitivity analysis1 Mar 2024
- 2024Negative impact of heavy-tailed uncertainty and error distributions on the reliability of calibration statistics for machine learning regression tasks15 Feb 2024
- 2024Forecasting VIX using Bayesian Deep Learning30 Jan 2024
- 2024Diffusion MRI with Machine Learning1 Jan 2024
- 2023Coherent energy and force uncertainty in deep learning force fields7 Dec 2023
- 2023Uncertainty Quantification in Machine Learning for Biosignal Applications -- A Review15 Nov 2023
- 2023Tailoring Mixup to Data for Calibration2 Nov 2023
- 2023HRTF Interpolation using a Spherical Neural Process Meta-Learner20 Oct 2023
- 2023Can bin-wise scaling improve consistency and adaptivity of prediction uncertainty for machine learning regression ?18 Oct 2023
- 2023Evidential Deep Learning: Enhancing Predictive Uncertainty Estimation for Earth System Science Applications22 Sep 2023
- 2023Calibration in Machine Learning Uncertainty Quantification: beyond consistency to target adaptivity12 Sep 2023
50 newest of 87 All 87 (JSON)
- 2026BAGEN: Are LLM Agents Budget-Aware?cited by 13
- 2026BAGEN: Are LLM Agents Budget-Aware?29 May 2026
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 680
- 2021How to Certify Machine Learning Based Safety-critical Systems? A Systematic Literature Reviewcited by 39
- 2022NeuralUQ: A comprehensive library for uncertainty quantification in neural differential equations and operatorscited by 33
- 2021Deep Ensembles from a Bayesian Perspectivecited by 27
- 2022Sample Efficient Deep Reinforcement Learning via Uncertainty Estimationcited by 27
- 2022BayesCap: Bayesian Identity Cap for Calibrated Uncertainty in Frozen Neural Networkscited by 23
- 2021Calibrated Uncertainty for Molecular Property Prediction using Ensembles of Message Passing Neural Networkscited by 21
- 2023Towards Building Self-Aware Object Detectors via Reliable Uncertainty Quantification and Calibrationcited by 19
- 2021Recalibration of Aleatoric and Epistemic Regression Uncertainty in Medical Imagingcited by 18
- 2023URL: A Representation Learning Benchmark for Transferable Uncertainty Estimatescited by 17
- 2023How Reliable is Your Regression Model's Uncertainty Under Real-World Distribution Shifts?cited by 14
- 2023Calibration in Machine Learning Uncertainty Quantification: beyond consistency to target adaptivitycited by 13
12 most cited of 84 All 84 (JSON)
- 2026Observation Quality in DSM-Assisted Aerial Localization: A Benchmark and Base-Anchored Uncertainty Calibration13 Oct 2026
- 2026UQ-LOB: Uncertainty-Aware Limit Order Book Mid-Price Forecasting25 Sep 2026
- 2026Faithfulness Without Anchors: A Position Paper on the Structural Misalignments of Explainable AI Faithfulness Metrics for Feature Importance24 Sep 2026
- 2026A Calibration Audit of Confidence in Feed-Forward 3D Reconstruction Models30 Aug 2026
- 2026Information-Based Calibration of Uncertainty Quantification in Product-of-Experts Gaussian Process Models29 Aug 2026
- 2026Structure-preserving uncertainty quantification for GENERIC dynamics12 Aug 2026
- 2026From Keypoints to Predictive Distributions: Post-Hoc Uncertainty for YOLO-Pose Models29 Jul 2026
- 2026Probabilistic Physics-Aware Machine Learning Predictions of Electric Truck Energy Consumption with Field Data21 Jul 2026
- 2026When Variance Is Not an Error Map: Calibrated Uncertainty for Radiative Gaussian Splatting in Sparse-View CT15 Jul 2026
- 2026From Pixels to Newtons: Predicting In Vivo Joint Contact Forces from Monocular Video4 Jun 2026
- 2026Reinforcement Learning over Predictive Distributions for LLM Regression20 May 2026
- 2026Uncertainty Reliability Under Domain Shift: An Investigation for Data-Driven Blood Pressure Estimation in Photoplethysmography18 May 2026
12 newest of 84 All 84 (JSON)