arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2201.05499cs.IRcs.LG
收藏

基于注意力的图形推荐

Attention-Based Recommendation On Graphs

Taher Hekmatfar, Saman Haratizadeh, Parsa Razban, Sama Goliaei

摘要:

图神经网络(GNN)在不同的任务中表现出了卓越的性能。然而,目前关于GNN在推荐系统上的研究还很少。GCN作为GNN的一种类型,可以为图中的不同实体提取高质量的嵌入。在协同过滤任务中,核心问题是找出实体对预测目标用户未来行为的信息量。使用注意力机制,当底层数据被建模为图时,我们可以使GCN能够进行这样的分析。在这项研究中,我们提出了GARec作为一个基于模型的推荐系统,它应用了一个注意力机制以及一个空间GCN的推荐图,以提取嵌入的用户和项目。注意力机制告诉GCN一个相关的用户或项目应该在多大程度上影响目标实体的最终表示。我们比较的性能GARec对一些基线算法的RMSE。该方法在不同的MovieLens数据集上优于现有的基于模型的非图神经网络和图神经网络。

英文摘要:

Graph Neural Networks (GNN) have shown remarkable performance in different tasks. However, there are a few studies about GNN on recommender systems. GCN as a type of GNNs can extract high-quality embeddings for different entities in a graph. In a collaborative filtering task, the core problem is to find out how informative an entity would be for predicting the future behavior of a target user. Using an attention mechanism, we can enable GCNs to do such an analysis when the underlying data is modeled as a graph. In this study, we proposed GARec as a model-based recommender system that applies an attention mechanism along with a spatial GCN on a recommender graph to extract embeddings for users and items. The attention mechanism tells GCN how much a related user or item should affect the final representation of the target entity. We compared the performance of GARec against some baseline algorithms in terms of RMSE. The presented method outperforms existing model-based, non-graph neural networks and graph neural networks in different MovieLens datasets.

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

↑