Success of Uncertainty-Aware Deep Models Depends on Data Manifold Geometry

Citation:

Penrod M, Termotto H, Reddy V, Yao J, Doshi-Velez F, Pan W. Success of Uncertainty-Aware Deep Models Depends on Data Manifold Geometry. Preprint. 2022.
Paper15.78 MB

Date Published:

aug

Abstract:

For responsible decision making in safety-critical settings, machine learning models must effectively detect and process edge-case data. Although existing works show that predictive uncertainty is useful for these tasks, it is not evident from literature which uncertainty-aware models are best suited for a given dataset. Thus, we compare six uncertainty-aware deep learning models on a set of edge-case tasks: robustness to adversarial attacks as well as out-of-distribution and adversarial detection. We find that the geometry of the data sub-manifold is an important factor in determining the success of various models. Our finding suggests an interesting direction in the study of uncertainty-aware deep learning models.

Notes:

arXiv:2208.01705 [cs]

Publisher's Version

Last updated on 06/27/2023