2409.01082

Total: 1

#1 Evidential Transformers for Improved Image Retrieval [PDF3] [Copy] [Kimi5] [REL]

Authors: Danilo Dordevic, Suryansh Kumar

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.

Subjects: Computer Vision and Pattern Recognition , Information Retrieval , Machine Learning

Publish: 2024-09-02 09:10:47 UTC