CCD-Level and Load-Aware Thread Orchestration for In-Memory Vector ANNS on Multi-Core CPUs
Yuchen Huang, Baiteng Ma, Yiping Sun, Yang Shi, Xiao Chen + 4 more
TLDR
This paper proposes a CCD-level and load-aware thread orchestration framework to boost in-memory vector ANNS performance on multi-core CPUs.
Key contributions
- Provides a uniform interface for inter-query parallel HNSW and intra-query parallel IVF search.
- Achieves cache-friendly and workload-adaptive mapping for task dispatching.
- Employs CCD-aware task stealing to effectively address load imbalance.
Why it matters
Vector ANNS is crucial for modern search and recommendation systems, but scaling on multi-core CPUs is challenging due to memory access and cache issues. This work proposes a novel thread orchestration framework that specifically optimizes for CCD-based architectures and real-world workloads. It achieves substantial throughput gains and latency reductions, offering a practical solution for high-performance ANNS.
Original Abstract
Vector approximate nearest neighbor search (ANNS) underpins search engines, recommendation systems, and advertising services. Recent advances in ANNS indexes make CPU a cost-effective choice for serving million-scale, in-memory vector search, yet per-core throughput remains constrained by memory access latency of vector reading and the compute intensity of distance evaluations in production deployments. With the growing scale of the business and advances in hardware, modern CCD-based multi-core CPUs have been widely deployed for high throughput in our services. However, we find that simply increasing core counts does not yield optimal performance scaling. To improve the efficiency of more cores from the CCD-based architecture, we analyze the distributions of real-world requests in our production environments. We observe high access locality in vector search in our online services and low cache utilization, resulting from overlooking the multi-chiplet nature of CCD based CPUs. Hence, we propose a workload- and hardware-aware thread orchestration framework at CCD-level that (i) provides a uniform interface for both inter-query parallel HNSW search and intra-query parallel IVF search, (ii) achieves cache-friendly and workload-adaptive mapping of task dispatching, and (iii) employs CCD-aware task stealing to address load imbalance. Applied to real production workloads from search, recommendation, and advertising services of Xiaohongshu (RedNote), our approach delivers up to 3.7x higher throughput and 30-90% reductions in P50 and P999 latency. In detail, compared with the original framework, the cache-miss ratio decreases by 6-30%, and the total CPU stall is reduced by 20-80%.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.