Lune

ICML2026Top-tier venue

Epistemic Uncertainty Quantification for Pre-trained VLMs via Riemannian Flow Matching

Li Ju, Mayank Nautiyal, Andreas Hellander, Ekta Vats, Prashant Singh

2026Year
2Citations

Abstract

Vision-Language Models (VLMs) are typically deterministic in nature and lack intrinsic mechanisms to quantify epistemic uncertainty, which reflects the model’s lack of knowledge or ignorance of its own representations. We theoretically motivate negative log-density of an embedding as a proxy for the epistemic uncertainty, where low-density regions signify model ignorance. The proposed method REPVLM computes the probability density on the hyperspherical manifold of the VLM embeddings using Riemannian Flow Matching. We empirically demonstrate that REPVLM achieves near-perfect correlation between uncertainty and prediction error, significantly outperforming existing baselines. Beyond classification, we also demonstrate that the model also provides a scalable metric for out-of-distribution detection and automated data curation.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 6fbcd67b-59e5-4ac9-bfb6-6235c4826419

Builds on20

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines