Class-specific Variational Auto-encoder For Content-based Image Retrieval
2023 · Mehdi Rafiei, Alexandros Iosifidis
Abstract
Using a discriminative representation obtained by supervised deep learning methods showed promising results on diverse Content-Based Image Retrieval (CBIR) problems. However, existing methods exploiting labels during training try to discriminate all available classes, which is not ideal in cases where the retrieval problem focuses on a class of interest. In this paper, we propose a regularized loss for Variational Auto-Encoders (VAEs) forcing the model to focus on a given class of interest. As a result, the model learns to discriminate the data belonging to the class of interest from any other possibility, making the learnt latent space of the VAE suitable for class-specific retrieval tasks. The proposed Class-Specific Variational Auto-Encoder (CS-VAE) is evaluated on three public and one custom datasets, and its performance is compared with that of three related VAE-based methods. Experimental results show that the proposed method outperforms its competition in both in-domain and out-
Authors
(none)
Tags
Stats
Related papers
- Loc-vae: Learning Structurally Localized Representation From 3D Brain MR Images For Content-based Image Retrieval (2022)5.24
- Exploring Masked Autoencoders For Sensor-agnostic Image Retrieval In Remote Sensing (2024)10.74
- Variational Autoencoder With CCA For Audio-visual Cross-modal Retrieval (2021)9.92
- Fast And Efficient Scene Categorization For Autonomous Driving Using Vaes (2022)2.26
- A Dense-depth Representation For VLAD Descriptors In Content-based Image Retrieval (2018)5.24
- Exploiting Distribution Constraints For Scalable And Efficient Image Retrieval (2024)0.00
- Exploiting Latent Codes: Interactive Fashion Product Generation, Similar Image Retrieval, And Cross-category Recommendation Using Variational Autoencoders (2020)0.00
- A Clinically Motivated Self-supervised Approach For Content-based Image Retrieval Of CT Liver Images (2022)9.76