Training a learning-to-rank model using a linear difference vector

Abstract

The disclosure herein describes training a document recommendation model using loss data generated from a linear score difference vector. A training data entry is provided including a query and a set of candidate documents. A document recommendation model generates a set of document prediction scores indicative of a likelihood that the candidate documents are responses to the query and a pairwise score difference matrix is generated using the set of document prediction scores. The pairwise score difference matrix is transformed into a score difference vector using a correct document vector that indicates a correct document among the set of candidate documents. Loss data of the document recommendation model is generated using the score difference vector and the document recommendation model is adjusted using the calculated loss data. Training the document recommendation model based on the linear score difference vector reduces resource usage when compared to training with a difference matrix.