RoarGraph: A Projected Bipartite Graph for Efficient Cross-Modal Approximate Nearest Neighbor Search
Chen, Meng, Zhang, Kai, He, Zhenying, Jing, Yinan, Wang, X. Sean
–arXiv.org Artificial Intelligence
Approximate Nearest Neighbor Search (ANNS) is a fundamental and critical component in many applications, including recommendation systems and large language model-based applications. With the advancement of multimodal neural models, which transform data from different modalities into a shared high-dimensional space as feature vectors, cross-modal ANNS aims to use the data vector from one modality (e.g., texts) as the query to retrieve the most similar items from another (e.g., images or videos). However, there is an inherent distribution gap between embeddings from different modalities, and cross-modal queries become Out-of-Distribution (OOD) to the base data. Consequently, state-of-the-art ANNS approaches suffer poor performance for OOD workloads. In this paper, we quantitatively analyze the properties of the OOD workloads to gain an understanding of their ANNS efficiency. Unlike single-modal workloads, we reveal OOD queries spatially deviate from base data, and the k-nearest neighbors of an OOD query are distant from each other in the embedding space. The property breaks the assumptions of existing ANNS approaches and mismatches their design for efficient search. With insights from the OOD workloads, we propose pRojected bipartite Graph (RoarGraph), an efficient ANNS graph index built under the guidance of query distribution. Extensive experiments show that RoarGraph significantly outperforms state-of-the-art approaches on modern cross-modal datasets, achieving up to 3.56x faster search speed at a 90% recall rate for OOD queries.
arXiv.org Artificial Intelligence
Aug-16-2024
- Country:
- South America > Chile
- North America
- United States
- Texas
- Travis County > Austin (0.04)
- Dallas County > Dallas (0.04)
- Oregon > Multnomah County
- Portland (0.04)
- Ohio > Franklin County
- Columbus (0.04)
- New York
- New York County > New York City (0.04)
- Kings County > New York City (0.04)
- Massachusetts > Suffolk County
- Boston (0.04)
- California > Los Angeles County
- Long Beach (0.04)
- Arizona > Maricopa County
- Scottsdale (0.04)
- Alaska > Anchorage Municipality
- Anchorage (0.04)
- Texas
- Canada
- Ontario > Toronto (0.04)
- British Columbia > Metro Vancouver Regional District
- Vancouver (0.04)
- United States
- Europe
- Austria (0.04)
- Spain > Galicia
- Madrid (0.04)
- Romania > București - Ilfov Development Region
- Municipality of Bucharest > Bucharest (0.04)
- Italy > Calabria
- Catanzaro Province > Catanzaro (0.04)
- Asia > Afghanistan
- Parwan Province > Charikar (0.04)
- Africa > Rwanda
- Genre:
- Research Report > New Finding (0.46)
- Industry:
- Information Technology (0.46)
- Technology: