Design and optimization of an ANNS framework based on GIDS
Approximate Nearest Neighbor Search (ANNS) systems face significant challenges when processing large-scale vector datasets that exceed GPU memory capacity, typically requiring dataset partitioning and distributed computation. We propose a novel ANNS framework leveraging GPU-Initiated Direct Storage (GIDS) access, enabling GPU threads to directly retrieve vector data from SSDs through user-space I/O mechanisms, bypassing CPU bottlenecks. Our system addresses memory constraints through a fine-grained page-level caching system optimized for high-concurrency GPU workloads. We introduce a two-tier graph search architecture where a lightweight upper-tier graph (1% of dataset) resides in GPU memory and provides query-specific entry points for main graph searches. This approach fundamentally redesigns the storage-compute pipeline by co-locating vectors with neighboring data in page-aligned storage and implementing GPU-controlled cache management. Experimental evaluation on SIFT1B demonstrates 20% improved query throughput compared to state-of-the-art DiskANN while maintaining superior recall rates. On clustered datasets, our hierarchical search achieves up to 48% recall improvement. The framework is modular, dataset-agnostic, and requires no vector quantization, indicating GPU-driven storage control as a promising direction for next-generation large-scale search engines.
Authors
- Zhaoyan Shen (ORCID: https://orcid.org/0000-0001-9526-6634)
- Qiang Li (ORCID: https://orcid.org/0000-0001-8947-1439)
- Zhiyong Zhang (ORCID: https://orcid.org/0000-0002-8782-2730)
- Jie Wang (ORCID: https://orcid.org/0009-0007-3073-7635)
- Xianwei Meng
Institutions
- Peng Cheng Laboratory (CN)
- Inspur (China) (CN)
- Shandong University of Science and Technology (CN)
Publication Details
- Journal
- High-Confidence Computing
- Published
- 2025-11-01
- DOI
- https://doi.org/10.1016/j.hcc.2025.100373
- Primary Topic
- Graph Theory and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 0.00