Billion-Scale Similarity Search with GPUs
A GPU design for k-selection that accelerates exact, approximate, and compressed similarity search, enabling billion-scale nearest-neighbor graphs.
Similarity search over complex data such as images and videos needs high-dimensional features and specialized indexing. This paper improves GPU use: GPUs excel at parallel distance computation, but prior methods were limited by low-parallelism k-min selection or poor memory use. They propose a new k-selection design applied to brute-force, approximate, and product-quantization compressed search, beating prior art by wide margins. It builds a k-NN graph over 95M Yfcc100M images in 35 minutes and connects 1 billion vectors in under 12 hours on four Titan X GPUs.
Based on: Billion-Scale Similarity Search with GPUs · IEEE Transactions on Big Data