Reliability-aware Prediction Via Uncertainty Learning For Person Image Retrieval

Abstract

Current person image retrieval methods have achieved great improvements in accuracy metrics. However, they rarely describe the reliability of the prediction. In this paper, we propose an Uncertainty-Aware Learning (UAL) method to remedy this issue. UAL aims at providing reliability-aware predictions by considering data uncertainty and model uncertainty simultaneously. Data uncertainty captures the ``noise" inherent in the sample, while model uncertainty depicts the model's confidence in the sample's prediction. Specifically, in UAL, (1) we propose a sampling-free data uncertainty learning method to adaptively assign weights to different samples during training, down-weighting the low-quality ambiguous samples. (2) we leverage the Bayesian framework to model the model uncertainty by assuming the parameters of the network follow a Bernoulli distribution. (3) the data uncertainty and the model uncertainty are jointly learned in a unified network, and they serve as two fundamental criteria

Stats

• citations12
• S2 citations—
• github stars0
• HF likes0
• heat score8.35
• arxiv keydou2022reliability

Related papers

• Unsupervised Data Uncertainty Learning In Visual Retrieval Systems (2019)0.00
• Bayesian Triplet Loss: Uncertainty Quantification In Image Retrieval (2020)11.49
• Exploring Uncertainty In Conditional Multi-modal Retrieval Systems (2019)0.00
• Dynamic Uncertainty Learning With Noisy Correspondence For Text-based Person Search (2025)7.50
• Ranking-aware Uncertainty For Text-guided Image Retrieval (2023)0.00
• Exploring Uncertainty Measures For Image-caption Embedding-and-retrieval Task (2019)2.26
• Risk Controlled Image Retrieval (2023)0.00
• Hyp-uml: Hyperbolic Image Retrieval With Uncertainty-aware Metric Learning (2023)0.00