Learning to rank

 

Ranking is a central part of many information retrieval problems, such as document retrieval, collaborative filtering, sentiment analysis, and online advertising.

A possible architecture of a machine-learned search engine is shown in the accompanying figure.

Training data consists of queries and documents matching them together with the relevance degree of each match. It may be prepared manually by human assessors (or raters, as Google calls them), who check results for some queries and determine relevance of each result. It is not feasible to check the relevance of all documents, and so typically a technique called pooling is used — only the top few documents, retrieved by some existing ranking models are checked. This technique may introduce selection bias. Alternatively, training data may be derived automatically by analyzing clickthrough logs (i.e. search results which got clicks from users),^[3] query chains,^[4] or such search engines' features as Google's (since-replaced) SearchWiki. Clickthrough logs can be biased by the tendency of users to click on the top search results on the assumption that they are already well-ranked.

Training data is used by a learning algorithm to produce a ranking model which computes the relevance of documents for actual queries.

Typically, users expect a search query to complete in a short time (such as a few hundred milliseconds for web search), which makes it impossible to evaluate a complex ranking model on each document in the corpus, and so a two-phase scheme is used.^[5] First, a small number of potentially relevant documents are identified using simpler retrieval models which permit fast query evaluation, such as the vector space model, Boolean model, weighted AND,^[6] or BM25. This phase is called top-${\displaystyle k}$  document retrieval and many heuristics were proposed in the literature to accelerate it, such as using a document's static quality score and tiered indexes.^[7] In the second phase, a more accurate but computationally expensive machine-learned model is used to re-rank these documents.

Learning to rank algorithms have been applied in areas other than information retrieval:

• In machine translation for ranking a set of hypothesized translations;^[8]
• In computational biology for ranking candidate 3-D structures in protein structure prediction problems;^[8]
• In recommender systems for identifying a ranked list of related news articles to recommend to a user after he or she has read a current news article.^[9]

For the convenience of MLR algorithms, query-document pairs are usually represented by numerical vectors, which are called feature vectors. Such an approach is sometimes called bag of features and is analogous to the bag of words model and vector space model used in information retrieval for representation of documents.

Components of such vectors are called features, factors or ranking signals. They may be divided into three groups (features from document retrieval are shown as examples):

• Query-independent or static features — those features, which depend only on the document, but not on the query. For example, PageRank or document's length. Such features can be precomputed in off-line mode during indexing. They may be used to compute document's static quality score (or static rank), which is often used to speed up search query evaluation.^[7][10]
• Query-dependent or dynamic features — those features, which depend both on the contents of the document and the query, such as TF-IDF score or other non-machine-learned ranking functions.
• Query-level features or query features, which depend only on the query. For example, the number of words in a query.

Some examples of features, which were used in the well-known LETOR dataset:

• TF, TF-IDF, BM25, and language modeling scores of document's zones (title, body, anchors text, URL) for a given query;
• Lengths and IDF sums of document's zones;
• Document's PageRank, HITS ranks and their variants.

Selecting and designing good features is an important area in machine learning, which is called feature engineering.

There are several measures (metrics) which are commonly used to judge how well an algorithm is doing on training data and to compare the performance of different MLR algorithms. Often a learning-to-rank problem is reformulated as an optimization problem with respect to one of these metrics.

Examples of ranking quality measures:

• Mean average precision (MAP);
• DCG and NDCG;
• Precision@n, NDCG@n, where "@n" denotes that the metrics are evaluated only on top n documents;
• Mean reciprocal rank;
• Kendall's tau;
• Spearman's rho.

DCG and its normalized variant NDCG are usually preferred in academic research when multiple levels of relevance are used.^[11] Other metrics such as MAP, MRR and precision, are defined only for binary judgments.

Recently, there have been proposed several new evaluation metrics which claim to model user's satisfaction with search results better than the DCG metric:

• Expected reciprocal rank (ERR);^[12]
• Yandex's pfound.^[13]

Both of these metrics are based on the assumption that the user is more likely to stop looking at search results after examining a more relevant document, than after a less relevant document.

Tie-Yan Liu of Microsoft Research Asia has analyzed existing algorithms for learning to rank problems in his book Learning to Rank for Information Retrieval.^[1] He categorized them into three groups by their input spaces, output spaces, hypothesis spaces (the core function of the model) and loss functions: the pointwise, pairwise, and listwise approach. In practice, listwise approaches often outperform pairwise approaches and pointwise approaches. This statement was further supported by a large scale experiment on the performance of different learning-to-rank methods on a large collection of benchmark data sets.^[14]

In this section, without further notice, ${\displaystyle x}$  denotes an object to be evaluated, for example, a document or an image, ${\displaystyle f(x)}$  denotes a single-value hypothesis, ${\displaystyle h(\cdot )}$  denotes a bi-variate or multi-variate function and ${\displaystyle L(\cdot )}$  denotes the loss function.

In this case, it is assumed that each query-document pair in the training data has a numerical or ordinal score. Then the learning-to-rank problem can be approximated by a regression problem — given a single query-document pair, predict its score. Formally speaking, the pointwise approach aims at learning a function ${\displaystyle f(x)}$  predicting the real-value or ordinal score of a document ${\displaystyle x}$  using the loss function ${\displaystyle L(f;x_{j},y_{j})}$ .

A number of existing supervised machine learning algorithms can be readily used for this purpose. Ordinal regression and classification algorithms can also be used in pointwise approach when they are used to predict the score of a single query-document pair, and it takes a small, finite number of values.

In this case, the learning-to-rank problem is approximated by a classification problem — learning a binary classifier ${\displaystyle h(x_{u},x_{v})}$  that can tell which document is better in a given pair of documents. The classifier shall take two documents as its input and the goal is to minimize a loss function ${\displaystyle L(h;x_{u},x_{v},y_{u,v})}$ . The loss function typically reflects the number and magnitude of inversions in the induced ranking.

In many cases, the binary classifier ${\displaystyle h(x_{u},x_{v})}$  is implemented with a scoring function ${\displaystyle f(x)}$ . As an example, RankNet ^[15] adapts a probability model and defines ${\displaystyle h(x_{u},x_{v})}$  as the estimated probability of the document ${\displaystyle x_{u}}$  has higher quality than ${\displaystyle x_{v}}$ :

${\displaystyle P_{u,v}(f)={\text{CDF}}(f(x_{u})-f(x_{v})),}$ 

where ${\displaystyle {\text{CDF}}(\cdot )}$  is a cumulative distribution function, for example, the standard logistic CDF, i.e.

${\displaystyle {\text{CDF}}(x)={\frac {1}{1+\exp \left[-x\right]}}.}$ 

These algorithms try to directly optimize the value of one of the above evaluation measures, averaged over all queries in the training data. This is often difficult in practice because most evaluation measures are not continuous functions with respect to ranking model's parameters, and so continuous approximations or bounds on evaluation measures have to be used. For example the SoftRank algorithm.^[16] LambdaMART is a pairwise algorithm which has been empirically shown to approximate listwise objective functions.^[17]

A partial list of published learning-to-rank algorithms is shown below with years of first publication of each method:

Year	Name	Type	Notes
1989	OPRF[18]	2pointwise	Polynomial regression (instead of machine learning, this work refers to pattern recognition, but the idea is the same).
1992	SLR[19]	2pointwise	Staged logistic regression.
1994	NMOpt[20]	2listwise	Non-Metric Optimization.
1999	MART (Multiple Additive Regression Trees)[21]	2pairwise
2000	Ranking SVM (RankSVM)	2pairwise	A more recent exposition is in,[3] which describes an application to ranking using clickthrough logs.
2001	Pranking	1pointwise	Ordinal regression.
2003	RankBoost	2pairwise
2005	RankNet	2pairwise
2006	IR-SVM[22]	2pairwise	Ranking SVM with query-level normalization in the loss function.
2006	LambdaRank	pairwise/listwise	RankNet in which pairwise loss function is multiplied by the change in the IR metric caused by a swap.
2007	AdaRank[23]	3listwise
2007	FRank	2pairwise	Based on RankNet, uses a different loss function - fidelity loss.
2007	GBRank	2pairwise
2007	ListNet	3listwise
2007	McRank	1pointwise
2007	QBRank	2pairwise
2007	RankCosine[24]	3listwise
2007	RankGP[25]	3listwise
2007	RankRLS	2pairwise	Regularized least-squares based ranking. The work is extended in [26] to learning to rank from general preference graphs.
2007	SVMmap	3listwise
2008	LambdaSMART/LambdaMART	pairwise/listwise	Winning entry in the Yahoo Learning to Rank competition in 2010, using an ensemble of LambdaMART models. Based on MART (1999)[27] “LambdaSMART”, for Lambda-submodel-MART, or LambdaMART for the case with no submodel.
2008	ListMLE[28]	3listwise	Based on ListNet.
2008	PermuRank[29]	3listwise
2008	SoftRank[30]	3listwise
2008	Ranking Refinement[31]	2pairwise	A semi-supervised approach to learning to rank that uses Boosting.
2008	SSRankBoost[32]	2pairwise	An extension of RankBoost to learn with partially labeled data (semi-supervised learning to rank).
2008	SortNet[33]	2pairwise	SortNet, an adaptive ranking algorithm which orders objects using a neural network as a comparator.
2009	MPBoost[34]	2pairwise	Magnitude-preserving variant of RankBoost. The idea is that the more unequal are labels of a pair of documents, the harder should the algorithm try to rank them.
2009	BoltzRank	3listwise	Unlike earlier methods, BoltzRank produces a ranking model that looks during query time not just at a single document, but also at pairs of documents.
2009	BayesRank	3listwise	A method combines Plackett-Luce Model and neural network to minimize the expected Bayes risk, related to NDCG, from the decision-making aspect.
2010	NDCG Boost[35]	3listwise	A boosting approach to optimize NDCG.
2010	GBlend	2pairwise	Extends GBRank to the learning-to-blend problem of jointly solving multiple learning-to-rank problems with some shared features.
2010	IntervalRank	2pairwise & listwise
2010	CRR[36]	2pointwise & pairwise	Combined Regression and Ranking. Uses stochastic gradient descent to optimize a linear combination of a pointwise quadratic loss and a pairwise hinge loss from Ranking SVM.
2014	LCR[37]	2pairwise	Applied local low-rank assumption on collaborative ranking. Received best student paper award at WWW'14.
2015	FaceNet	pairwise	Ranks face images with the triplet metric via deep convolutional network.
2016	XGBoost	pairwise	Supports various ranking objectives and evaluation metrics.
2017	ES-Rank[38]	listwise	Evolutionary Strategy Learning to Rank technique with 7 fitness evaluation metrics.
2018	DLCM[39]	2listwise	A multi-variate ranking function that encodes multiple items from an initial ranked list (local context) with a recurrent neural network and create result ranking accordingly.
2018	PolyRank[40]	pairwise	Learns simultaneously the ranking and the underlying generative model from pairwise comparisons.
2018	FATE-Net/FETA-Net[41]	listwise	End-to-end trainable architectures, which explicitly take all items into account to model context effects.
2019	FastAP[42]	listwise	Optimizes Average Precision to learn deep embeddings.
2019	Mulberry	listwise & hybrid	Learns ranking policies maximizing multiple metrics across the entire dataset.
2019	DirectRanker	pairwise	Generalisation of the RankNet architecture.
2019	GSF[43]	2listwise	A permutation-invariant multi-variate ranking function that encodes and ranks items with groupwise scoring functions built with deep neural networks.
2020	RaMBO[44]	listwise	Optimizes rank-based metrics using blackbox backpropagation.[45]
2020	PRM[46]	pairwise	Transformer network encoding both the dependencies among items and the interactions between the user and items.
2020	SetRank[47]	2listwise	A permutation-invariant multi-variate ranking function that encodes and ranks items with self-attention networks.
2021	PiRank[48]	listwise	Differentiable surrogates for ranking able to exactly recover the desired metrics and scales favourably to large list sizes, significantly improving internet-scale benchmarks.
2022	SAS-Rank	listwise	Combining Simulated Annealing with Evolutionary Strategy for implicit and explicit learning to rank from relevance labels.
2022	VNS-Rank	listwise	Variable Neighborhood Search in 2 Novel Methodologies in AI for Learning to Rank.
2022	VNA-Rank	listwise	Combining Simulated Annealing with Variable Neighbourhood Search for Learning to Rank.
2023	GVN-Rank	listwise	Combining Gradient Ascent with Variable Neighbourhood Search for Learning to Rank.

Note: as most supervised learning-to-rank algorithms can be applied to pointwise, pairwise and listwise case, only those methods which are specifically designed with ranking in mind are shown above.

Norbert Fuhr introduced the general idea of MLR in 1992, describing learning approaches in information retrieval as a generalization of parameter estimation;^[49] a specific variant of this approach (using polynomial regression) had been published by him three years earlier.^[18] Bill Cooper proposed logistic regression for the same purpose in 1992 ^[19] and used it with his Berkeley research group to train a successful ranking function for TREC. Manning et al.^[50] suggest that these early works achieved limited results in their time due to little available training data and poor machine learning techniques.

Several conferences, such as NeurIPS, SIGIR and ICML have had workshops devoted to the learning-to-rank problem since the mid-2000s (decade).

Commercial web search engines began using machine-learned ranking systems since the 2000s (decade). One of the first search engines to start using it was AltaVista (later its technology was acquired by Overture, and then Yahoo), which launched a gradient boosting-trained ranking function in April 2003.^[51][52]

Bing's search is said to be powered by RankNet algorithm,^[53][when?] which was invented at Microsoft Research in 2005.

In November 2009 a Russian search engine Yandex announced^[54] that it had significantly increased its search quality due to deployment of a new proprietary MatrixNet algorithm, a variant of gradient boosting method which uses oblivious decision trees.^[55] Recently they have also sponsored a machine-learned ranking competition "Internet Mathematics 2009"^[56] based on their own search engine's production data. Yahoo has announced a similar competition in 2010.^[57]

As of 2008, Google's Peter Norvig denied that their search engine exclusively relies on machine-learned ranking.^[58] Cuil's CEO, Tom Costello, suggests that they prefer hand-built models because they can outperform machine-learned models when measured against metrics like click-through rate or time on landing page, which is because machine-learned models "learn what people say they like, not what people actually like".^[59]

In January 2017, the technology was included in the open source search engine Apache Solr.^[60] It is also available in the open source OpenSearch and the source-available Elasticsearch.^[61][62] These implementations make learning to rank widely accessible for enterprise search.

Similar to recognition applications in computer vision, recent neural network based ranking algorithms are also found to be susceptible to covert adversarial attacks, both on the candidates and the queries.^[63] With small perturbations imperceptible to human beings, ranking order could be arbitrarily altered. In addition, model-agnostic transferable adversarial examples are found to be possible, which enables black-box adversarial attacks on deep ranking systems without requiring access to their underlying implementations.^[63][64]

Conversely, the robustness of such ranking systems can be improved via adversarial defenses such as the Madry defense.^[65]

• Content-based image retrieval
• Multimedia information retrieval
• Image retrieval
• Triplet loss

Competitions and public datasets

• LETOR: A Benchmark Collection for Research on Learning to Rank for Information Retrieval
• Yandex's Internet Mathematics 2009
• Yahoo! Learning to Rank Challenge
• Microsoft Learning to Rank Datasets