Results of the NeurIPS'21 Challenge on Billion-Scale Approximate Nearest Neighbor Search

被引:0
作者
Simhadri, Harsha Vardhan [1 ]
Williams, George [2 ]
Aumueller, Martin [3 ]
Douze, Matthijs [4 ]
Babenko, Artem [5 ]
Baranchuk, Dmitry [5 ]
Chen, Qi [1 ]
Hosseini, Lucas [4 ]
Krishnaswamy, Ravishankar [1 ]
Srinivasa, Gopal [1 ]
Subramanya, Suhas Jayaram [6 ]
Wang, Jingdong [7 ]
机构
[1] Microsoft Res, Redmond, WA 98052 USA
[2] GSI Technol, Sunnyvale, CA USA
[3] IT Univ Copenhagen, Copenhagen, Denmark
[4] Meta AI Res, Menlo Pk, CA USA
[5] Yandex, Moscow, Russia
[6] Carnegie Mellon Univ, Pittsburgh, PA 15213 USA
[7] Baidu, Beijing, Peoples R China
来源
NEURIPS 2021 COMPETITIONS AND DEMONSTRATIONS TRACK, VOL 176 | 2021年 / 176卷
关键词
Approximate nearest neighbor search; large-scale search;
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Despite the broad range of algorithms for Approximate Nearest Neighbor Search, most empirical evaluations of algorithms have focused on smaller datasets, typically of 1 million points (Aumuller et al., 2020). However, deploying recent advances in embedding based techniques for search, recommendation and ranking at scale require ANNS indices at billion, trillion or larger scale. Barring a few recent papers, there is limited consensus on which algorithms are effective at this scale vis-a-vis their hardware cost. This competition(1) compares ANNS algorithms at billion-scale by hardware cost, accuracy and performance. We set up an open source evaluation framework(2)and leaderboards for both standardized and specialized hardware. The competition involves three tracks. The standard hardware track T1 evaluates algorithms on an Azure VM with limited DRAM, often the bottleneck in serving billion-scale indices, where the embedding data can be hundreds of GigaBytes in size. It uses FAISS (Johnson et al., 2017) as the baseline. The standard hardware track T2 additional allows inexpensive SSDs in addition to the limited DRAM and uses DiskANN (Subramanya et al., 2019) as the baseline. The specialized hardware track T3 allows any hardware configuration, and again uses FAISS as the baseline. We compiled six diverse billion-scale datasets, four newly released for this competition, that span a variety of modalities, data types, dimensions, deep learning models, distance functions and sources. The outcome of the competition was ranked leaderboards of algorithms in each track based on recall at a query throughput threshold. Additionally, for track T3, separate leaderboards were created based on recall as well as cost-normalized and power-normalized query throughput.
引用
收藏
页码:177 / 189
页数:13
相关论文
共 29 条
[1]  
Malkov YA, 2018, Arxiv, DOI arXiv:1603.09320
[2]   Near-optimal hashing algorithms for approximate nearest neighbor in high dimensions [J].
Andoni, Alexandr ;
Indyk, Piotr .
COMMUNICATIONS OF THE ACM, 2008, 51 (01) :117-122
[3]   HD-Index: Pushing the Scalability-Accuracy Boundary for Approximate kNN Search in High-Dimensional Spaces [J].
Arora, Akhil ;
Sinha, Sakshi ;
Kumar, Piyush ;
Bhattacharya, Arnab .
PROCEEDINGS OF THE VLDB ENDOWMENT, 2018, 11 (08) :906-919
[4]  
ARYA S, 1993, PROCEEDINGS OF THE FOURTH ANNUAL ACM-SIAM SYMPOSIUM ON DISCRETE ALGORITHMS, P271
[5]   PUFFINN: Parameterless and Universally Fast FInding of Nearest Neighbors [J].
Aumuller, Martin ;
Christiani, Tobias ;
Pagh, Rasmus ;
Vesterli, Michael .
27TH ANNUAL EUROPEAN SYMPOSIUM ON ALGORITHMS (ESA 2019), 2019, 144
[6]   ANN-Benchmarks: A benchmarking tool for approximate nearest neighbor algorithms [J].
Aumuller, Martin ;
Bernhardsson, Erik ;
Faithfull, Alexander .
INFORMATION SYSTEMS, 2020, 87
[7]   Efficient Indexing of Billion-Scale datasets of deep descriptors [J].
Babenko, Artem ;
Lempitsky, Victor .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :2055-2063
[8]   Additive Quantization for Extreme Vector Compression [J].
Babenko, Artem ;
Lempitsky, Victor .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :931-938
[9]   Revisiting the Inverted Indices for Billion-Scale Approximate Nearest Neighbors [J].
Baranchuk, Dmitry ;
Babenko, Artem ;
Malkov, Yury .
COMPUTER VISION - ECCV 2018, PT XII, 2018, 11216 :209-224
[10]  
Beygelzimer A., 2006, ICML