Abstract:Vector quantization (VQ) based ANN indexes, such as Inverted File System (IVF) and Product Quantization (PQ), have been widely applied to embedding based document retrieval thanks to the competitive time and memory efficiency. Originally, VQ is learned to minimize the reconstruction loss, i.e., the distortions between the original dense embeddings and the reconstructed embeddings after quantization. Unfortunately, such an objective is inconsistent with the goal of selecting ground-truth documents for the input query, which may cause severe loss of retrieval quality. Recent works identify such a defect, and propose to minimize the retrieval loss through contrastive learning. However, these methods intensively rely on queries with ground-truth documents, whose performance is limited by the insufficiency of labeled data. In this paper, we propose Distill-VQ, which unifies the learning of IVF and PQ within a knowledge distillation framework. In DistillVQ, the dense embeddings are leveraged as "teachers", which predict the query's relevance to the sampled documents. The VQ modules are treated as the "students", which are learned to reproduce the predicted relevance, such that the reconstructed embeddings may fully preserve the retrieval result of the dense embeddings. By doing so, Distill-VQ is able to derive substantial training signals from the massive unlabeled data, which significantly contributes to the retrieval quality. We perform comprehensive explorations for the optimal conduct of knowledge distillation, which may provide useful insights for the learning of VQ based ANN index. We also experimentally show that the labeled data is no longer a necessity for high-quality vector quantization, which indicates Distill-VQ's strong applicability in practice. The evaluations are performed on MS MARCO and Natural Questions benchmarks, where Distill-VQ notably outperforms the SOTA VQ methods in Recall and MRR. Our code is avaliable at https://github.com/staoxiao/LibVQ.

Logit Variated Product Quantization Based on Parts Interaction and Metric Learning with Knowledge Distillation for Fine-Grained Image Retrieval

Deep Product Quantization Module for Efficient Image Retrieval

Learning Local Feature Descriptors Through Ranking Losses Improved by Variance Shrinkage

Information Gain Product Quantization for Image Retrieval

Orthonormal Product Quantization Network for Scalable Face Image Retrieval

Beyond Product Quantization: Deep Progressive Quantization for Image Retrieval.

Deep Progressive Asymmetric Quantization Based on Causal Intervention for Fine-Grained Image Retrieval

Semi-supervised product quantization for approximate nearest neighbor search

Mean-Removed Product Quantization for Large-scale Image Retrieval

HiHPQ: Hierarchical Hyperbolic Product Quantization for Unsupervised Image Retrieval

Distill-VQ: Learning Retrieval Oriented Vector Quantization by Distilling Knowledge from Dense Embeddings

Large Scale Product Search with Spatial Quantization and Deep Ranking

Scalable Image Retrieval by Sparse Product Quantization

Matching-oriented Embedding Quantization for Ad-hoc Retrieval.

Entropy-Optimized Deep Weighted Product Quantization for Image Retrieval

Deep Fourier Ranking Quantization for Semi-Supervised Image Retrieval

Bilinear Optimized Product Quantization for Scalable Visual Content Analysis.

Large-similarity image similarity retrieval method and system based on deep product quantization

Unleashing the Full Potential of Product Quantization for Large-Scale Image Retrieval.

Matching-oriented Product Quantization for Ad-hoc Retrieval

Towards Codebook-Free Deep Probabilistic Quantization for Image Retrieval.