English

Evidential Transformers for Improved Image Retrieval

Computer Vision and Pattern Recognition 2025-09-09 v2 Information Retrieval Machine Learning

Abstract

We introduce the Evidential Transformer, an uncertainty-driven transformer model for improved and robust image retrieval. In this paper, we make several contributions to content-based image retrieval (CBIR). We incorporate probabilistic methods into image retrieval, achieving robust and reliable results, with evidential classification surpassing traditional training based on multiclass classification as a baseline for deep metric learning. Furthermore, we improve the state-of-the-art retrieval results on several datasets by leveraging the Global Context Vision Transformer (GC ViT) architecture. Our experimental results consistently demonstrate the reliability of our approach, setting a new benchmark in CBIR in all test settings on the Stanford Online Products (SOP) and CUB-200-2011 datasets.

Keywords

Cite

@article{arxiv.2409.01082,
  title  = {Evidential Transformers for Improved Image Retrieval},
  author = {Danilo Dordevic and Suryansh Kumar},
  journal= {arXiv preprint arXiv:2409.01082},
  year   = {2025}
}

Comments

6 pages, 6 figures, presented at the 3rd Workshop on Uncertainty Quantification for Computer Vision, at the ECCV 2024 conference in Milan, Italy

R2 v1 2026-06-28T18:31:12.279Z