• 融入过滤算法的概率自注意力序列推荐模型

    Probabilistic self-attention sequence recommendation model with filtering algorithm

    • 序列推荐旨在通过对用户的历史交互序列建模,以预测该用户的下一项交互。目前,基于Transformer改进的序列推荐模型利用点积自注意力来衡量序列中的关系,拥有卓越的性能。然而,由于用户交互数据中夹杂一定的噪声,模型在训练时很容易拟合到噪声数据,导致过拟合。另外,传统自注意力机制中还天然存在高内存占用以及二次计算复杂度的问题。为了解决上述两个问题,针对性提出了一种融入过滤算法的概率自注意力序列推荐模型FPA-Rec(Filter ProbAttention Recommendation)。输入序列在经过嵌入矩阵映射之后,先经过一层可学习的过滤器结构来对序列中的噪声信号进行抑制,得到更加纯净的序列输入。之后使用概率自注意力代替传统的自注意力,来解决传统自注意力机制中存在的高内存占用和二次计算复杂度问题。实验验证结果表明,FPA-Rec模型达到了最优的推荐效果。在以Beauty、Yelp为代表的两个数据集上,评价指标Hit@10较最优模型提升1.2%、1.7%,评价指标NDCG@10较最优模型提升2.4%、2.0%。

       

      Abstract: Sequence recommendations are designed to predict a user's next interaction by modeling the user's historical interaction sequence. At present, the improved series recommendation model based on Transformer utilizes dot product self-attention to measure the relationships in the series and has excellent performance. However, due to the inclusion of some noise in the user interaction data, the model is easy to fit the noisy data during training, resulting in overfitting. In addition, the traditional self-attention mechanism also naturally has the problem of high memory consumption and secondary computing complexity. In order to solve the above two problems, a probabilistic self-attention sequence recommendation model FPA-Rec (Filter ProbAttention Sequence Recommendation) is proposed. After the input sequence goes through the embedding matrix mapping, it first passes through a layer of learning filter structure to suppress the noise signal in the sequence and get a more pure sequence input. Then probabilistic self-attention is used instead of traditional self-attention to solve the problems of high memory consumption and secondary computing complexity in the traditional self-attention mechanism. Experimental verification results show that FPA-Rec model achieves the optimal recommendation effect. On the two data sets represented by Beauty and Yelp, the evaluation index Hit@10 increased by 1.2% and 1.7% compared with the optimal model, the evaluation index NDCG@10 increased by 2.4% and 2.0% compared with the optimal model.

       

    /

    返回文章
    返回