Training-Free Uncertainty Estimation for Dense Regression: Sensitivity as a Surrogate
About
Uncertainty estimation is an essential step in the evaluation of the robustness for deep learning models in computer vision, especially when applied in risk-sensitive areas. However, most state-of-the-art deep learning models either fail to obtain uncertainty estimation or need significant modification (e.g., formulating a proper Bayesian treatment) to obtain it. Most previous methods are not able to take an arbitrary model off the shelf and generate uncertainty estimation without retraining or redesigning it. To address this gap, we perform a systematic exploration into training-free uncertainty estimation for dense regression, an unrecognized yet important problem, and provide a theoretical construction justifying such estimations. We propose three simple and scalable methods to analyze the variance of outputs from a trained network under tolerable perturbations: infer-transformation, infer-noise, and infer-dropout. They operate solely during the inference, without the need to re-train, re-design, or fine-tune the models, as typically required by state-of-the-art uncertainty estimation methods. Surprisingly, even without involving such perturbations in training, our methods produce comparable or even better uncertainty estimation when compared to training-required state-of-the-art methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 3D Object Detection | aiMotive (test) | Mean AP (all-point)64.59 | 12 | |
| Uncertainty Estimation in Image Super-Resolution | Set5 (test) | Pearson Correlation49.3 | 7 | |
| Uncertainty Estimation in Image Super-Resolution | Set14 (test) | Pearson Correlation Coefficient0.49 | 7 | |
| Uncertainty Estimation in Image Super-Resolution | BSDS100 (test) | Pearson Correlation0.465 | 7 | |
| Uncertainty Estimation in Image Super-Resolution | IXI T1-weighted MRI (test) | Pearson Correlation Coefficient0.598 | 7 | |
| 3D Object Detection | aiMotive distant region >75m (test) | Highway (all-point AP)45.57 | 6 | |
| 3D Object Detection | aiMotive Motion Blur | AP (All-Point)62.61 | 6 | |
| 3D Object Detection | aiMotive Over-Exposure | AP (all-point)64.94 | 6 | |
| 3D Object Detection | aiMotive Under-Exposure | AP (All-Point)63.25 | 6 | |
| 3D Object Detection | aiMotive LiDAR-Fog | AP (All-Point)65.19 | 6 |