基于注意力的交互式学习排序模型用于文档检索

An Attention-Based Interactive Learning-to-Rank Model for Document Retrieval

IEEE Transactions on Systems, Man, and Cybernetics: Systems · 2021
被引 6
ABS 3

中文导读

提出一个基于注意力的交互式学习排序模型AIRank,将排序视为用户与系统交互的动态过程,通过自注意力方法捕捉文档间关系以丰富用户意图表示,并用改进的策略梯度方法训练,实验表明其效果优于现有方法。

Abstract

The core issue of learning-to-rank (LTR) for document retrieval lies in finding an optimal ranking policy to meet the search intent of the user. The majority of proposed LTR approaches treat the ranking as a static process, employing a fixed ranking policy to immediately assign scores to documents. By contrast, ranking is not a static but an interactive process where the user continues interacting with the document retrieval system through information exchange such as search intent (e.g., rating or clicking for the retrieved items). We model the interactive ranking process (IRP), and propose an Attention-Based Interactive LTR model (AIRank) to constitute an intent-aware flexible ranking policy to gratify the user’s need. To enhance the ranking quality, the inherent relations among documents are procured by the self-attention method to contribute to an enriched user intent representation. Furthermore, we mend the policy gradient learning method to train the AIRank in the IRP. Experiments demonstrate the effectiveness of AIRank compared to the state-of-the-art methods in terms of normalized discounted cumulative gain and expected reciprocal rank.

信息检索学习排序机器学习文档检索