High-Quality Prediction Intervals for Deep Learning: A Distribution-Free, Ensembled Approach
About
This paper considers the generation of prediction intervals (PIs) by neural networks for quantifying uncertainty in regression tasks. It is axiomatic that high-quality PIs should be as narrow as possible, whilst capturing a specified portion of data. We derive a loss function directly from this axiom that requires no distributional assumption. We show how its form derives from a likelihood principle, that it can be used with gradient descent, and that model uncertainty is accounted for in ensembled form. Benchmark experiments show the method outperforms current state-of-the-art uncertainty quantification methods, reducing average PI width by over 10%.
Tim Pearce, Mohamed Zaki, Alexandra Brintrup, Andy Neely• 2018
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Prediction Interval Estimation | Naval (test) | PICP91.16 | 6 | |
| Prediction Interval Estimation | Miami (test) | PICP90.26 | 6 | |
| Prediction Interval Estimation | kin8nm (test) | PICP89.52 | 6 | |
| Prediction Interval Estimation | Boston (test) | PICP90.29 | 6 | |
| Prediction Interval Estimation | sulfur (test) | PICP89.41 | 6 | |
| Prediction Interval Estimation | concrete (test) | PICP89.37 | 6 | |
| Prediction Interval Estimation | Yacht (test) | PICP90.32 | 6 | |
| Prediction Interval Estimation | Protein (test) | PICP90.38 | 6 | |
| Prediction Interval Estimation | cpu_act (test) | PICP90.32 | 6 | |
| Prediction Interval Estimation | Energy (test) | PICP89.42 | 6 |
Showing 10 of 12 rows