Bayesian Triplet Loss: Uncertainty Quantification in Image Retrieval
Uncertainty quantification in image retrieval is crucial for downstream decisions, yet it remains a challenging and largely unexplored problem. Current methods for estimating uncertainties are poorly calibrated, computationally expensive, or based on heuristics. We present a new method that views image embeddings as stochastic features rather than deterministic features. Our two main contributions are (1) a likelihood that matches the triplet constraint and that evaluates the probability of an anchor being closer to a positive than a negative; and (2) a prior over the feature space that justifies the conventional l2 normalization. To ensure computational efficiency, we derive a variational approximation of the posterior, called the Bayesian triplet loss, that produces state-of-the-art uncertainty estimates and matches the predictive performance of current state-of-the-art methods.
Code (0)
등록된 구현이 없습니다.
Tasks
Computational EfficiencyImage RetrievalRetrievalTripletUncertainty QuantificationSimilar Papers 제목 키워드 기반
Bayesian Deep Convolutional Encoder-Decoder Networks for Surrogate Modeling and Uncertainty Quantification
We are interested in the development of surrogate models for uncertainty quantification and propagation in problems governed by stochastic PDEs using a deep convolutional encoder-decoder network in a similar fashion to a…
Bayesian InferenceDecoderGaussian ProcessesImage-to-Image Regression+2A generalized Bayes framework for probabilistic clustering
Loss-based clustering methods, such as k-means and its variants, are standard tools for finding groups in data. However, the lack of quantification of uncertainty in the estimated clusters is a disadvantage. Model-based …
ClusteringUncertainty QuantificationFully Bayesian VIB-DeepSSM
Statistical shape modeling (SSM) enables population-based quantitative analysis of anatomical shapes, informing clinical diagnosis. Deep learning approaches predict correspondence-based SSM directly from unsegmented 3D i…
AnatomyUncertainty QuantificationVariational InferenceUncertainty quantification in non-rigid image registration via stochastic gradient Markov chain Monte Carlo
We develop a new Bayesian model for non-rigid registration of three-dimensional medical images, with a focus on uncertainty quantification. Probabilistic registration of large images with calibrated uncertainty estimates…
Image RegistrationUncertainty QuantificationVariational InferenceBayesian Scattering: A Principled Baseline for Uncertainty on Image Data
Uncertainty quantification for image data is dominated by complex deep learning methods, yet the field lacks an interpretable, mathematically grounded baseline. We propose Bayesian scattering to fill this gap, serving as…