Morse Neural Networks for Uncertainty Quantification
Abstract: We introduce a new deep generative model useful for uncertainty quantification: the Morse neural network, which generalizes the unnormalized Gaussian densities to have modes of high-dimensional submanifolds instead of just discrete points. Fitting the Morse neural network via a KL-divergence loss yields 1) a (unnormalized) generative density, 2) an OOD detector, 3) a calibration temperature, 4) a generative sampler, along with in the supervised case 5) a distance aware-classifier. The Morse network can be used on top of a pre-trained network to bring distance-aware calibration w.r.t the training data. Because of its versatility, the Morse neural networks unifies many techniques: e.g., the Entropic Out-of-Distribution Detector of (Mac^edo et al., 2021) in OOD detection, the one class Deep Support Vector Description method of (Ruff et al., 2018) in anomaly detection, or the Contrastive One Class classifier in continuous learning (Sun et al., 2021). The Morse neural network has connections to support vector machines, kernel methods, and Morse theory in topology.
- Amari, S.-i. Information Geometry and Its Applications. Applied mathematical sciences. Springer, 2016.
- Concrete problems in AI safety. https://arxiv.org/abs/1606.06565, 2016.
- Generalized energy based models. In International Conference on Learning Representations, 2021.
- Morse-Bott theory and equivariant cohomology. 1995.
- A proof of the Morse-Bott lemma. Expositiones Mathematicae, 2004.
- A connection between cut locus, Thom space and Morse-Bott functions. to appear in Algebraic & Geometric Topology., 2021.
- Anomalous example detection in deep learning: A survey. IEEE Access, 8, 2020.
- Your GAN is secretly an energy-based model and you should use discriminator driven latent sampling. In Larochelle, H., Ranzato, M., Hadsell, R., Balcan, M., and Lin, H. (eds.), NeurIPS, 2020.
- Learning energy-based models by diffusion recovery likelihood. In International Conference on Learning Representations, 2021.
- Explaining and harnessing adversarial examples. In ICLR, 2015.
- Deep Learning. MIT Press, 2016.
- On calibration of modern neural networks. In ICML, 2017.
- Reliable graph neural networks for drug discovery under distributional shift. In NeurIPS 2021 Workshop on Distribution Shifts: Connecting Methods and Applications, 2021.
- A baseline for detecting misclassified and out-of-distribution examples in neural networks. In International Conference on Learning Representations, 2017. URL https://openreview.net/forum?id=Hkg4TI9xl.
- Hirsch, M. W. Differential topology. Springer New York, 1997. ISBN 0387901485 3540901485.
- Continual learning by using information of each class holistically. AAAI, 2021.
- Highly accurate protein structure prediction with alphafold. Nature, 596:583 – 589, 2021.
- Measuring and Improving Model-Moderator Collaboration using Uncertainty Estimation, 2021.
- Simple and scalable predictive uncertainty estimation using deep ensembles. In NeurIPS, 2017.
- An evaluation dataset for intent classification and out-of-scope prediction. In EMNLP-IJCNLP 2019.
- A simple unified framework for detecting out-of-distribution samples and adversarial attacks. In NeurIPS, volume 31, 2018.
- Simple and principled uncertainty estimation with deterministic deep learning via distance awareness. In NeurIPS, 2020.
- A simple approach to improve single-model deep uncertainty via distance-awareness. JMLR, 2022.
- Entropic out-of-distribution detection. In International Joint Conference on Neural Networks (IJCNN), pp. 1–8, 2021.
- Density of states estimation for out of distribution detection. In AISTATS, 2021.
- Morse, A. P. The behavior of a function on its critical set. Annals of Mathematics, 40:62–70, 1939.
- Murphy, K. P. Probabilistic Machine Learning: Advanced Topics. MIT Press, 2023. URL http://probml.github.io/book2.
- Understanding the failure modes of out-of-distribution generalization. In ICLR, 2021.
- Do deep generative models know what they don’t know? In ICLR, 2019.
- Deep neural networks are easily fooled: High confidence predictions for unrecognizable images. In CVPR, 2015.
- Can you trust your model’s uncertainty? evaluating predictive uncertainty under dataset shift. In NeurIPS, 2019.
- Using mixup as a regularizer can surprisingly improve accuracy & out-of-distribution robustness. In NeurIPS, 2022.
- Prasad, S. Cut locus of submanifolds: A geometric and topological viewpoint. 2023.
- Likelihood ratios for out-of-distribution detection. In NeurIPS, volume 32, 2019.
- A simple fix to Mahalanobis distance for improving near-ood detection. arXiv preprint arXiv:2106.09022, 2021.
- Does your dermatology classifier know what it doesn’t know? detecting the long-tail of unseen conditions. Medical Imaging Analysis, 2021.
- Deep one-class classification. In ICML, 2018.
- A unified survey on anomaly, novelty, open-set, and out of-distribution detection: Solutions and future challenges. TMLR, 2022.
- Sard, A. The measure of the critical values of differentiable maps. Bulletin of the American Mathematical Society, 48:883–890, 1942.
- Deep learning-based object detection and scene perception under bad weather conditions. Electronics, 11(4), 2022.
- ILCOC: An incremental learning framework based on contrastive one-class classifiers. In CVPR, 2021.
- Energy-based open-world uncertainty modeling for confidence calibration. In ICCV 2021, 2021.
- Conjugate energy-based models. In Third Symposium on Advances in Approximate Bayesian Inference, 2021.
- Poisson flow generative models. In NeurIPS, 2022.
- Energy-based generative adversarial networks. In International Conference on Learning Representations, 2017.
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.