Ensemble-Based Dirichlet Modeling for Predictive Uncertainty and Selective Classification
Not on the map. Why?
Abstract (arXiv)
Neural network classifiers trained with cross-entropy loss achieve strong predictive accuracy but lack the capability to provide inherent predictive uncertainty estimates, thus requiring external techniques to obtain these estimates. In addition, softmax scores for the true class can vary substantially across independent training runs, which limits the reliability of uncertainty-based decisions in downstream tasks. Evidential Deep Learning aims to address these limitations by producing uncertainty estimates in a single pass, but evidential training is highly sensitive to design choices including loss formulation, prior regularization, and activation functions. Therefore, this work introduces an alternative Dirichlet parameter estimation strategy by applying a method of moments estimator to ensembles of softmax outputs, with an optional maximum-likelihood refinement step. This ensemble-based construction decouples uncertainty estimation from the fragile evidential loss design while also mitigating the variability of single-run cross-entropy training, producing explicit Dirichlet predictive distributions. Across multiple datasets, we show that the improved stability and predictive uncertainty behavior of these ensemble-derived Dirichlet estimates translate into stronger performance in downstream uncertainty-guided applications such as prediction confidence scoring and selective classification.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 46 of its 64 references that are in the map. First posted on arXiv 7 Apr 2026.
Builds on
46 of 64 references.
References · 46
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,274
- 2017Selective Classification for Deep Neural Networkscited by 799
- 2018Evidential Deep Learning to Quantify Classification Uncertaintycited by 728
- 2021MedMNIST v2 -- A large-scale lightweight benchmark for 2D and 3D biomedical image classificationcited by 682
- 2018Predictive Uncertainty Estimation via Prior Networkscited by 555
- 2015Variational Dropout and the Local Reparameterization Trickcited by 489
- 2019Measuring Calibration in Deep Learningcited by 443
34 more
- 2016Ladder Variational Autoencoderscited by 315
- 2019Verified Uncertainty Calibrationcited by 307
- 2017Variational Dropout Sparsifies Deep Neural Networkscited by 265
- 2020Pitfalls of In-Domain Uncertainty Estimation and Ensembling in Deep Learningcited by 208
- 2019Cyclical Annealing Schedule: A Simple Approach to Mitigating KL Vanishingcited by 191
- 2019Evaluating model calibration in classificationcited by 165
- 2020Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-Countscited by 163
- 2019Lagging Inference Networks and Posterior Collapse in Variational Autoencoderscited by 124
- 2021Prior and Posterior Networks: A Survey on Evidential Deep Learning Methods For Uncertainty Estimationcited by 71
- 2020Towards Trustworthy Predictions from Deep Neural Networks with Fast Adversarial Calibrationcited by 34
- 2023Learn to Accumulate Evidence from All Training Samples: Theory and Practicecited by 28
- 2022OpenTAL: Towards Open Set Temporal Action Localizationcited by 23
- 2020Amortized Bayesian model comparison with evidential deep learningcited by 22
- 2024Uncertainty Estimation by Density Aware Evidential Deep Learningcited by 22
- 2022Region-Based Evidential Deep Learning to Quantify Uncertainty and Improve Robustness of Brain Tumor Segmentationcited by 20
- 2024Are Uncertainty Quantification Capabilities of Evidential Deep Learning a Mirage?cited by 17
- 2024Revisiting Essential and Nonessential Settings of Evidential Deep Learningcited by 15
- 2021Improving evidential deep learning via multi-task learningcited by 13
- 2022Towards Confident Detection of Prostate Cancer using High Resolution Micro-ultrasoundcited by 9
- 2025Credal and Interval Deep Evidential Classificationscited by 8
- 2020Learning Calibrated Uncertainties for Domain Shift: A Distributionally Robust Learning Approachcited by 6
- 2022TEDL: A Two-stage Evidential Deep Learning Method for Classification Uncertainty Quantificationcited by 3
- 2024Risk-aware Classification via Uncertainty Quantificationcited by 3
- 2021Leveraging Graph and Deep Learning Uncertainties to Detect Anomalous Trajectoriescited by 2
- 2023ProPanDL: A Modular Architecture for Uncertainty-Aware Panoptic Segmentationcited by 2
- 2023Knowledge from Uncertainty in Evidential Deep Learningcited by 2
- 2024Predictive Uncertainty Quantification for Bird's Eye View Segmentation: A Benchmark and Novel Loss Functioncited by 2
- 2024Evidential Federated Learning for Skin Lesion Image Classificationcited by 2
- 2024Evidential Deep Learning for Probabilistic Modelling of Extreme Storm Eventscited by 2
- 2025E2-BKI: Evidential Ellipsoidal Bayesian Kernel Inference for Uncertainty-aware Gaussian Semantic Mappingcited by 2
- 2025Generalized Regularized Evidential Deep Learning Models: Theory and Comprehensive Evaluationcited by 2
- 2025Calibrated and uncertain? Evaluating uncertainty estimates in binary classification modelscited by 1
- 2025Inferring Cosmological Parameters with Evidential Physics-Informed Neural Networkscited by 1
- 2025Bi-level Meta-Policy Control for Dynamic Uncertainty Calibration in Evidential Deep Learningcited by 1
Built on by
Not cited yet.