arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.10084cs.IR
收藏

上下文感知的自我注意学习排序

Context-Aware Learning to Rank with Self-Attention

Przemysław Pobrotyn, Tomasz Bartczak, Mikołaj Synowiec, Radosław Białobrzeski, Jarosław Bojar

摘要:

学习排序是许多电子商务搜索引擎的关键组成部分。在学习排序中,人们感兴趣的是根据项目对用户的效用来优化项目列表的全局排序。流行的方法通过优化逐点、逐对或逐列损失来学习一个单独对项目进行评分的评分函数(即,不考虑列表中其他项目的上下文)。然后按分数的降序对列表进行排序。在训练阶段,在损失水平上考虑了同一列表中存在的项目之间可能的相互作用。然而,在推理过程中,项目是单独评分的,并且不考虑它们之间可能的相互作用。在本文中,我们提出了一个上下文感知的神经网络模型,学习项目分数应用自我注意力机制。因此,在训练和推理中,给定项目的相关性在列表中存在的所有其他项目的上下文中确定。我们通过经验证明了基于自我注意力的神经架构在多层感知器基线上的显着性能增益,特别是在来自大型电子商务市场Allegro.pl的搜索日志的数据集上。最后,我们报告了在MSLR-WEB 30 K(学习排序基准)上的最新结果。

英文摘要:

Learning to rank is a key component of many e-commerce search engines. In learning to rank, one is interested in optimising the global ordering of a list of items according to their utility for users.Popular approaches learn a scoring function that scores items individually (i.e. without the context of other items in the list) by optimising a pointwise, pairwise or listwise loss. The list is then sorted in the descending order of the scores. Possible interactions between items present in the same list are taken into account in the training phase at the loss level. However, during inference, items are scored individually, and possible interactions between them are not considered. In this paper, we propose a context-aware neural network model that learns item scores by applying a self-attention mechanism. The relevance of a given item is thus determined in the context of all other items present in the list, both in training and in inference. We empirically demonstrate significant performance gains of self-attention based neural architecture over Multi-LayerPerceptron baselines, in particular on a dataset coming from search logs of a large scale e-commerce marketplace, Allegro.pl. This effect is consistent across popular pointwise, pairwise and listwise losses.Finally, we report new state-of-the-art results on MSLR-WEB30K, the learning to rank benchmark.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑