Abstract:In recent years, Approximate Nearest Neighbor Search (ANNS) has played a pivotal role in modern search and recommendation systems, especially in emerging LLM applications like Retrieval-Augmented Generation. There is a growing exploration into harnessing the parallel computing capabilities of GPUs to meet the substantial demands of ANNS. However, existing systems primarily focus on offline scenarios, overlooking the distinct requirements of online applications that necessitate real-time insertion of new vectors. This limitation renders such systems inefficient for real-world scenarios. Moreover, previous architectures struggled to effectively support real-time insertion due to their reliance on serial execution streams. In this paper, we introduce a novel Real-Time Adaptive Multi-Stream GPU ANNS System (RTAMS-GANNS). Our architecture achieves its objectives through three key advancements: 1) We initially examined the real-time insertion mechanisms in existing GPU ANNS systems and discovered their reliance on repetitive copying and memory allocation, which significantly hinders real-time effectiveness on GPUs. As a solution, we introduce a dynamic vector insertion algorithm based on memory blocks, which includes in-place rearrangement. 2) To enable real-time vector insertion in parallel, we introduce a multi-stream parallel execution mode, which differs from existing systems that operate serially within a single stream. Our system utilizes a dynamic resource pool, allowing multiple streams to execute concurrently without additional execution blocking. 3) Through extensive experiments and comparisons, our approach effectively handles varying QPS levels across different datasets, reducing latency by up to 40%-80%. The proposed system has also been deployed in real-world industrial search and recommendation systems, serving hundreds of millions of users daily, and has achieved good results.

Efficient Selection Algorithm for Fast k-NN Search on GPUs

A social network-aware top-N recommender system using GPU.

Fast $k$-NNG construction with GPU-based quick multi-select

Constrained All-k-Nearest-Neighbor Search

Performance Optimization of Top-k Queries on GPU.

Improving GPU-accelerated adaptive IDW interpolation algorithm using fast kNN search

Efficient Parallel Processing of High-Dimensional Spatial K NN Queries

Billion-Scale Similarity Search with GPUs

A Nearest Neighbor Searches(NNS) Algorithm for Fast Registration of 3D Point Clouds Based on GPGPU

RTop-K: Ultra-Fast Row-Wise Top-K Algorithm and GPU Implementation for Neural Networks

Semi-Convex Hull Tree: Fast Nearest Neighbor Queries for Large Scale Data on GPUs

A Real-Time Spike Sorting Method Based on the Embedded GPU

A Parallel Implementation of Nearest Neighbor Analysis Based on GPGPU

Efficient parallel processing for K-nearest-neighbor search in spatial databases

Efficient Large-scale Approximate Nearest Neighbor Search on the GPU

A Real-Time Adaptive Multi-Stream GPU System for Online Approximate Nearest Neighborhood Search

JUNO: Optimizing High-Dimensional Approximate Nearest Neighbour Search with Sparsity-Aware Algorithm and Ray-Tracing Core Mapping

L-FNNG: Accelerating Large-Scale KNN Graph Construction on CPU-FPGA Heterogeneous Platform

BANG: Billion-Scale Approximate Nearest Neighbor Search using a Single GPU

Efficient Parallel K Best Connected Trajectory (K-BCT) Query with GPGPU: A Combinatorial Min-Distance and Progressive Bounding Box Approach

CAGRA: Highly Parallel Graph Construction and Approximate Nearest Neighbor Search for GPUs