Overcoming the Sync-Compute Dilemma in Parallel Graph-Based Vector Retrieval
Qiji Mo, Zhiyuan Hua, Zebin Yao, Lixiao Cui, Gang Wang, Xiaoguang Liu, Zijing Wei, Xinyu Liu, Tianxiao Tang, Shaozhi Liu, Lin Qu
摘要
Approximate Nearest Neighbor Search (ANNS) is fundamental to modern applications such as Retrieval-augmented Generation. Among various ANNS algorithms, graph-based methods have become the state-of-the-art due to their excellent search efficiency. The core of graph-based methods is to perform a Best-First Search (BFiS) on the proximity graph. However, with the continuous growth in data scale and dimensionality, the single-threaded BFiS algorithm is becoming increasingly inadequate to meet performance demands. In this paper, we first deconstruct the existing parallel graph-based ANNS algorithms (e.g., iQAN, Edge-wise). We find that they are constrained by the Bulk Synchronous Parallel model, which leads to a dilemma between reducing the synchronization overhead and reducing the computational overhead. To overcome this dilemma, we propose a decoupled search paradigm named ScatterSearch. In this paradigm, each thread independently conducts a full BFiS from a distinct entry point and maintains a private candidate queue, thereby eliminating explicit synchronization. Building on ScatterSearch, we further propose a leader-guided search pruning strategy to reduce computational overhead. It prunes search progress of some threads to avoid long-tail computation and redirects the pruned threads through a workstealing mechanism. Comprehensive experiments on real-world datasets show that with 16 threads and at Recall@100=0.99, our method reduces latency by up to 41.46% and increases throughput by up to compared to iQAN. Our method has been utilized on the Alibaba's taobao platform.
问问这篇 Paper
问问你的智能体。
Lune 读过与它相关的顶会 Paper,每个回答都会注明依据哪几篇。
相关 Paper
- FlashANNS: GPU-Driven Asynchronous I/O Pipelining for Eliminating Storage-Compute Bottlenecks in Billion-Scale Similarity SearchYang Xiao, Mo Sun, Ziyu Song, Bing Tian 等SIGMOD 2026 · 被引用 3 次
- iQAN: Fast and Accurate Vector Search with Efficient Intra-Query Parallelism on Multi-Core ArchitecturesZhen Peng, Minjia Zhang, Kai Li, Ruoming Jin 等PPoPP 2023 · 被引用 20 次
- A Comprehensive Survey and Experimental Comparison of Graph-Based Approximate Nearest Neighbor SearchMengzhao Wang, Xiaoliang Xu, Qiang Yue, Yuxiang WangVLDB 2021 · 被引用 354 次
- CSPG: Crossing Sparse Proximity Graphs for Approximate Nearest Neighbor SearchMing Yang, Yuzheng Cai, Weiguo ZhengNeurIPS 2024 · 被引用 14 次
- ParlayANN: Scalable and Deterministic Parallel Graph-Based Approximate Nearest Neighbor Search AlgorithmsMagdalen Dobson Manohar, Zheqi Shen, Guy E. Blelloch, Laxman Dhulipala 等PPoPP 2024 · 被引用 39 次
