Scalable Overload-Aware Graph-Based Index Construction for 10-Billion-Scale Vector Similarity Search
Approximate Nearest Neighbor Search (ANNS) is essential for modern data-driven applications that require efficient retrieval of top-k results from massive vector databases. Although existing graph-based ANNS algorithms achieve a high recall rate on billion-scale datasets, their slow construction speed and limited scalability hinder their applicability to large-scale industrial scenarios. In this paper, we introduce SOGAIC, the first Scalable Overload-Aware Graph-Based ANNS Index Construction system tailored for ultra-large-scale vector databases: 1) We propose a dynamic data partitioning algorithm with overload constraints that adaptively introduces overlaps among subsets; 2) To enable efficient distributed subgraph construction, we employ an a load-balancing task scheduling framework combined with an agglomerative merging strategy; 3) Extensive experiments on various datasets demonstrate a reduction of 47.3% in average construction time compared to existing methods. The proposed method has also been successfully deployed in a real-world industrial search engine, managing over 10 billion daily updated vectors and serving hundreds of millions of users.
Authors
- Xueping Zhong
- Yang Shi
- Yao Hu (ORCID: https://orcid.org/0009-0006-1274-7111)
- Zhiyong Wang (ORCID: https://orcid.org/0009-0009-2335-4892)
- Jiaolong Du
- Yiping Sun (ORCID: https://orcid.org/0009-0000-9559-8268)
Institutions
- Xiaomi (China) (CN)
Publication Details
- Published
- 2025-05-08
- DOI
- https://doi.org/10.1145/3701716.3715576
- Citations
- 3
- Primary Topic
- Data Management and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 4.46