mlsanityData to 14 Oct 2026

Uncertainties of Latent Representations in Computer Vision

arXiv 2408.14281 ↗First posted 26 Aug 2024cs.LG cs.AI cs.CVcited by 1

Not on the map. Why?

Abstract (arXiv)

Uncertainty quantification is a key pillar of trustworthy machine learning. It enables safe reactions under unsafe inputs, like predicting only when the machine learning model detects sufficient evidence, discarding anomalous data, or emitting warnings when an error is likely to be inbound. This is particularly crucial in safety-critical areas like medical image classification or self-driving cars. Despite the plethora of proposed uncertainty quantification methods achieving increasingly higher scores on performance benchmarks, uncertainty estimates are often shied away from in practice. Many machine learning projects start from pretrained latent representations that come without uncertainty estimates. Uncertainties would need to be trained by practitioners on their own, which is notoriously difficult and resource-intense. This thesis makes uncertainty estimates easily accessible by adding them to the latent representation vectors of pretrained computer vision models. Besides proposing approaches rooted in probability and decision theory, such as Monte-Carlo InfoNCE (MCInfoNCE) and loss prediction, we delve into both theoretical and empirical questions. We show that these unobservable uncertainties about unobservable latent representations are indeed provably correct. We also provide an uncertainty-aware representation learning (URL) benchmark to compare these unobservables against observable ground-truths. Finally, we compile our findings to pretrain lightweight representation uncertainties on large-scale computer vision models that transfer to unseen datasets in a zero-shot manner. Our findings do not only advance the current theoretical understanding of uncertainties over latent variables, but also facilitate the access to uncertainty quantification for future researchers inside and outside the field, enabling straightforward but trustworthy machine learning.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 54 of its 116 references that are in the map. First posted on arXiv 26 Aug 2024.

Builds on

54 of 116 references.

References · 54

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  4. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  5. 2017On Calibration of Modern Neural Networkscited by 4,960
  6. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  7. 2021Zero-Shot Text-to-Image Generationcited by 4,288
  8. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
  9. 2021LAION-400M: Open Dataset of CLIP-Filtered 400 Million Image-Text Pairscited by 1,555
  10. 2018A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attackscited by 1,485
  11. 2020Gradient Surgery for Multi-Task Learningcited by 1,256
  12. 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 792
38 more
  1. 2023Scaling Vision Transformers to 22 Billion Parameterscited by 694
  2. 2019A Large-scale Study of Representation Learning with the Visual Task Adaptation Benchmarkcited by 425
  3. 2020Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awarenesscited by 389
  4. 2024The Faiss librarycited by 376
  5. 2019Deep Learning for Anomaly Detection: A Surveycited by 358
  6. 2020Are we done with ImageNet?cited by 333
  7. 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 236
  8. 2021Contrastive Learning Inverts the Data Generating Processcited by 221
  9. 2022Quantifying Aleatoric and Epistemic Uncertainty in Machine Learning: Are Conditional Entropy and Mutual Information Appropriate Measures?cited by 125
  10. 2021DEUP: Direct Epistemic Uncertainty Predictioncited by 104
  11. 2022Plex: Towards Reliability using Pretrained Large Model Extensionscited by 102
  12. 2023Probabilistic Concept Bottleneck Modelscited by 101
  13. 2020Revisiting Training Strategies and Generalization Performance in Deep Metric Learningcited by 100
  14. 2017L2-constrained Softmax Loss for Discriminative Face Verificationcited by 96
  15. 2024How to Train Data-Efficient LLMscited by 93
  16. 2024Benchmarking Uncertainty Disentanglement: Specialized Uncertainties for Specialized Taskscited by 78
  17. 2021On the Practicality of Deterministic Epistemic Uncertaintycited by 62
  18. 2018Modeling Uncertainty with Hedged Instance Embeddingcited by 52
  19. 2022Is one annotation enough? A data-centric image classification benchmark for noisy and ambiguous label estimationcited by 43
  20. 2023On Second-Order Scoring Rules for Epistemic Uncertainty Quantificationcited by 38
  21. 2020Uncertainty Estimation Using a Single Deep Deterministic Neural Networkcited by 37
  22. 2023What Can We Learn From The Selective Prediction And Uncertainty Estimation Performance Of 523 Imagenet Classifierscited by 35
  23. 2024Is Epistemic Uncertainty Faithfully Represented by Evidential Deep Learning Methods?cited by 35
  24. 2023A framework for benchmarking class-out-of-distribution detection and its application to ImageNetcited by 31
  25. 2023Probabilistic Contrastive Learning Recovers the Correct Aleatoric Uncertainty of Ambiguous Inputscited by 29
  26. 2022Accelerated Linearized Laplace Approximation for Bayesian Deep Learningcited by 28
  27. 2023Sources of Uncertainty in Supervised Machine Learning -- A Statisticians' Viewcited by 28
  28. 2023Bad Students Make Great Teachers: Active Learning Accelerates Large-Scale Visual Understandingcited by 28
  29. 2022MUAD: Multiple Uncertainties for Autonomous Driving, a benchmark for multiple uncertainty types and taskscited by 24
  30. 2023URL: A Representation Learning Benchmark for Transferable Uncertainty Estimatescited by 17
  31. 2023Learning Sample Difficulty from Pre-trained Models for Reliable Predictioncited by 15
  32. 2023Massively Scaling Heteroscedastic Classifierscited by 11
  33. 2019Stochastic Prototype Embeddingscited by 10
  34. 2023When does Privileged Information Explain Away Label Noise?cited by 10
  35. 2023When Less is More: Investigating Data Pruning for Pretraining LLMs at Scalecited by 10
  36. 2024Pretrained Visual Uncertaintiescited by 10
  37. 2022Probabilistic Embeddings Revisitedcited by 8
  38. 2023Medical Image Retrieval Using Pretrained Embeddingscited by 8

50 most cited of 54 All 54 (JSON)

Built on by

Cited by year

Citing papers per year: 2024: 1.

1 paper.