QuickScorer: A Fast Algorithm to Rank Documents with Additive Ensembles of Regression Trees

被引:53
作者
Lucchese, Claudio [1 ]
Nardini, Franco Maria [1 ]
Orlando, Salvatore [2 ]
Perego, Raffaele [1 ]
Tonellotto, Nicola [1 ]
Venturini, Rossano [3 ]
机构
[1] ISTI CNR, Pisa, Italy
[2] Univ Venice, I-30123 Venice, Italy
[3] Univ Pisa, I-56100 Pisa, Italy
来源
SIGIR 2015: PROCEEDINGS OF THE 38TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL | 2015年
关键词
Learning to Rank; Efficiency; Cache-aware Algorithms;
D O I
10.1145/2766462.2767733
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
Learning-to-Rank models based on additive ensembles of regression trees have proven to be very effective for ranking query results returned by Web search engines, a scenario where quality and efficiency requirements are very demanding. Unfortunately, the computational cost of these ranking models is high. Thus, several works already proposed solutions aiming at improving the efficiency of the scoring process by dealing with features and peculiarities of modern CPUs and memory hierarchies. In this paper, we present QUICKSCORER, a new algorithm that adopts a novel bitvector representation of the tree-based ranking model, and performs an interleaved traversal of the ensemble by means of simple logical bitwise operations. The performance of the proposed algorithm are unprecedented, due to its cache-aware approach, both in terms of data layout and access patterns, and to a control flow that entails very low branch mis-prediction rates. The experiments on real Learning-to-Rank datasets show that QuickScorer is able to achieve speedups over the best state-of-the-art baseline ranging from 2x to 6.5x.
引用
收藏
页码:73 / 82
页数:10
相关论文
共 19 条
[1]  
[Anonymous], 2010, Proceedings of the ACM International Conference on Information and Knowledge Management
[2]   Runtime Optimizations for Tree-based Machine Learning Models [J].
Asadi, Nima ;
Lin, Jimmy ;
de Vries, Arjen P. .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2014, 26 (09) :2281-2292
[3]  
Burges Christopher J.C., 2010, Learning, DOI DOI 10.1111/J.1467-8535
[4]  
Degenhardt J., 2010, P 3 ACM INT C WEB SE, P411
[5]   Greedy function approximation: A gradient boosting machine [J].
Friedman, JH .
ANNALS OF STATISTICS, 2001, 29 (05) :1189-1232
[6]  
Ganjisaffar Y, 2011, PROCEEDINGS OF THE 34TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR'11), P85
[7]   Cumulated gain-based evaluation of IR techniques [J].
Järvelin, K ;
Kekäläinen, J .
ACM TRANSACTIONS ON INFORMATION SYSTEMS, 2002, 20 (04) :422-446
[8]  
Patterson D. A., 2014, COMPUTER ORG DESIGN, V5th
[9]   The probabilistic relevance framework: BM25 and beyond [J].
Robertson, Stephen ;
Zaragoza, Hugo .
Foundations and Trends in Information Retrieval, 2009, 3 (04) :333-389
[10]  
Segalovich I., 2010, MACHINE LEARNING SEA