Scalable Graph Indexing using GPUs for Approximate Nearest Neighbor Search

TL;DR

This paper presents Tagore, a GPU-accelerated library for scalable graph indexing in high-dimensional ANNS, introducing GNN-Descent and CFS algorithms to significantly improve speed and efficiency while maintaining quality.

Contribution

The paper introduces Tagore, a novel GPU-based graph indexing library with new algorithms GNN-Descent and CFS for efficient large-scale ANNS.

Findings

Achieves 1.32x-112.79x speedup on real datasets

Supports various k-NN graph pruning strategies

Maintains high index quality with improved efficiency

Abstract

Approximate nearest neighbor search (ANNS) in high-dimensional vector spaces has a wide range of real-world applications. Numerous methods have been proposed to handle ANNS efficiently, while graph-based indexes have gained prominence due to their high accuracy and efficiency. However, the indexing overhead of graph-based indexes remains substantial. With exponential growth in data volume and increasing demands for dynamic index adjustments, this overhead continues to escalate, posing a critical challenge. In this paper, we introduce Tagore, a fast library accelerated by GPUs for graph indexing, which has powerful capabilities of constructing refinement-based graph indexes such as NSG and Vamana. We first introduce GNN-Descent, a GPU-specific algorithm for efficient k-Nearest Neighbor (k-NN) graph initialization. GNN-Descent speeds up the similarity comparison by a two-phase descent procedure and enables highly parallelized neighbor updates. Next, aiming to support various k-NN graph pruning strategies, we formulate a universal computing procedure termed CFS and devise two generalized GPU kernels for parallel processing complex dependencies in neighbor relationships. For large-scale datasets exceeding GPU memory capacity, we propose an asynchronous GPU-CPU-disk indexing framework with a cluster-aware caching mechanism to minimize the I/O pressure on the disk. Extensive experiments on 7 real-world datasets exhibit that Tagore achieves 1.32x-112.79x speedup while maintaining the index quality.