Extending ROC Analysis to Uncertainty-Aware Risk Prediction with an Interval-Based AUC (iAUC)
Organizations: Department of Biostatistics & Bioinformatics, Duke University
Abstract
In high-stakes risk prediction, interval-valued predictions provide a natural way to represent predictive uncertainty. However, standard evaluation tools such as the receiver operating characteristic (ROC) curve and the area under the curve (AUC) are designed for point-valued predictions and do not capture how uncertainty affects discrimination performance. To address this gap, we develop an uncertainty-aware ROC framework and corresponding interval-based AUC (iAUC) metrics for evaluating the performance of interval-valued risk predictions. The framework constructs two ROC-style curves with associated areas and , and shows that these quantities induce a three-region decomposition of positive-negative rankings into confidently correct, ambiguous, and confidently incorrect orderings. Under valid class-conditional coverage, and , together with the pairwise miscoverage rate, provide theoretical bounds on the optimal AUC, linking interval coverage to achievable discrimination. The framework is compatible with a range of methods for quantifying epistemic uncertainty in risk prediction, including Bayesian, bootstrap, and ensemble-based procedures, and provides a practical way to compare predictive models and interval-generation methods. Experiments on clinical benchmark data and MIMIC-IV mortality prediction show that models with similar point AUC can have different uncertainty-aware ranking profiles, and that the proposed decomposition can support risk-aware model selection and threshold-based abstention.
Figures & tables
| Model | Generator | Point AUC | |||
|---|---|---|---|---|---|
| LR | Bootstrap | 0.821 | 0.591 | 0.352 | 0.057 |
| RF | Deep Ensemble | 0.810 | 0.728 | 0.147 | 0.125 |
| HistGBM | Bootstrap | 0.826 | 0.305 | 0.685 | 0.009 |
| HistGBM | Deep Ensemble | 0.824 | 0.566 | 0.385 | 0.049 |
| EBM | Bootstrap | 0.830 | 0.265 | 0.728 | 0.007 |
| EBM | Deep Ensemble | 0.831 | 0.675 | 0.249 | 0.076 |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Quantity | Value | Difference |
|---|---|---|
| vs. | 0.6074 / 0.6072 | 0.0002 |
| vs. | 0.9427 / 0.9439 | 0.0013 |
| 0.6072 | — | |
| 0.3367 | — | |
| 0.0561 | — | |
| Sum of probabilities | 1.0000 | — |
| Model | Generator | Point AUC | Sum | |||||
|---|---|---|---|---|---|---|---|---|
| LR | Bootstrap | 0.821 | 0.591 | 0.352 | 0.057 | 0.591 | 0.943 | 1.000 |
| LR | Bayesian Bootstrap | 0.822 | 0.604 | 0.337 | 0.059 | 0.604 | 0.941 | 1.000 |
| LR | Naive | 0.821 | 0.573 | 0.377 | 0.051 | 0.573 | 0.950 | 1.000 |
| RF | Bootstrap | 0.812 | 0.493 | 0.471 | 0.036 | 0.493 | 0.964 | 1.000 |
| RF | Deep Ensemble | 0.810 | 0.728 | 0.147 | 0.125 | 0.728 | 0.875 | 1.000 |
| RF | Bayesian Bootstrap | 0.811 | 0.674 | 0.234 | 0.092 | 0.674 | 0.907 | 1.000 |
| Model | AUC | ||||
|---|---|---|---|---|---|
| LR | 0.772 | 0.678 | 0.848 | 0.170 | 0.152 |
| RF | 0.814 | 0.637 | 0.925 | 0.288 | 0.075 |
| HistGB | 0.835 | 0.567 | 0.962 | 0.395 | 0.038 |
| MLP | 0.827 | 0.405 | 0.985 | 0.581 | 0.015 |
| Eligible models | Selected model | Selected | |
|---|---|---|---|
| 0.05 | HistGB, MLP | HistGB | 0.567 |
| 0.10 | RF, HistGB, MLP | RF | 0.637 |
| 0.15 | RF, HistGB, MLP | RF | 0.637 |
| Model | Threshold | Point TPR | Point FPR | Abstain | Retained | Pos. abstain | Neg. abstain | Retained TPR/FPR |
|---|---|---|---|---|---|---|---|---|
| LR | Train prev. | 0.664 | 0.263 | 0.137 | 0.863 | 0.155 | 0.135 | 0.683 / 0.228 |
| RF | Train prev. | 0.722 | 0.261 | 0.248 | 0.752 | 0.265 | 0.246 | 0.777 / 0.195 |
| HistGB | Train prev. | 0.741 | 0.241 | 0.317 | 0.683 | 0.405 | 0.306 | 0.801 / 0.141 |
| MLP | Train prev. | 0.757 | 0.274 | 0.476 | 0.524 | 0.606 | 0.461 | 0.824 / 0.093 |
| LR | 0.343 | 0.078 | 0.057 | 0.943 | 0.120 | 0.049 | 0.333 / 0.058 | |
| RF | 0.437 | 0.082 | 0.134 | 0.866 | 0.291 | 0.115 | 0.410 / 0.044 |
Uncertainty Quantification
Uncertainty quantification estimates how reliable predictions and scientific inferences are, and how uncertainty propagates into decisions.
See all 649 papers