PilotANN: Memory-Bounded GPU Acceleration for Vector Search
提出 PilotANN,一个面向图索引近似最近邻搜索的混合 CPU-GPU 系统。同时利用 CPU 的大内存和 GPU 的并行计算能力,在显存受限条件下实现大规模向量搜索的 GPU 加速。
INSIGHT传统 GPU 向量搜索受限于显存容量,PilotANN 通过 CPU-GPU 协同调度突破了这一瓶颈。对于搜索系统中的向量召回层,这意味着可以在不购买更多 GPU 显存的情况下处理更大规模的向量库。
业务关联社区搜索中语义召回依赖大规模向量检索,PilotANN 的内存受限 GPU 加速方案可以在现有硬件条件下显著降低向量搜索延迟,提升 p99 响应时间。
证据在大规模向量搜索基准上测试,实现了内存受限条件下的 GPU 加速。利用 CPU 大容量 RAM 与 GPU 并行能力的混合架构,适用于图索引的 ANN 检索场景。