Lightweight ViT-Based Image Retrieval System with Qdrant for Efficient Similarity Search
摘要
This study presents a lightweight image retrieval system for a domain-specific dataset. The proposed system employs a Vision Transformer (ViT) with multi-head attention [1] to learn visual representations from a product image dataset collected from the Vietnamese e-commerce platforms. The model was trained entirely from scratch. The image feature vectors are stored and indexed in Qdrant to enable efficient similarity search. Experimental results indicate that the proposed approach provides improved clustering performance across several baseline models under our evaluation setting, while requiring fewer parameters.