- Conference Article
- 10.1109/icauc68182.2026.11441046
Efficient Image Retrieval with Vision Transformer (ViT) and FAISS
- Jan 19, 2026
- Alekya Himabindu + 5 more +5
The image data has grown fast in fields like social media, healthcare, and e-commerce and this has enhanced the demand of quick and precise image retrieval systems. Conventional methods of retrieval that have relied on handcrafted methods of feature extraction and pixel-wise comparison are costly in computer time and cannot withstand changes in image quality, lighting, and view point. To resolve these drawbacks, this project adopts an effective content-based image retrieval framework with a Vision Transformer (ViT) to extract features and FAISS (Facebook AI Similarity Search) to conducted fast similarity search. The system gets compact numerical representations of features of images as a result of a deep learning model and constructs a scale index to search similarity. The FAISS IndexIVFFlat architecture allows to search the nearest-neighbor approximationally with poor scale and low query time by grouping image features into clusters and lessening the search space at query time. Metadata of the indexed images are stored in the system and the top-k most similar images are retrieved according to the distance scores. This method is accurate and fast thus appropriate when dealing with large scale image databases. The system relies on the IndexIVFFlat structure of FAISS to cluster image features into regions which is a significant reduction of the search space that can support meaningful approximate nearest-neighbor queries. Results indicate that the system can be effective in separating useful and useless images that have high similarity (up to 0.819) and low distances (0.220) that ensure the accuracy and scalability of the system to image databases of large scales.
Read more