基于评论构建边的图嵌入模型
首发时间:2021-04-07
摘要:针对在推荐系统中训练用户和商品的图嵌入时,信息利用不充分的问题,提出了利用推荐系统中的评论和交互信息来训练用户、商品图嵌入的模型RHG,该模型以基于边的图神经网络为基础,首先利用评论文本构建图结构中的边用于对图节点信息进行聚合,其次设计不同特征空间用于提高聚合节点信息的效率,然后根据不同节点的重要性设计带有权重的损失函数从而优化训练。最终,在亚马逊的三个真实数据集上,通过推荐系统中的评分任务对图嵌入效果进行评判,RHG的均方误差相比目前基于文本的最优图神经网络模型CGAT降低2.75%,证实了RHG模型的有效性。
For information in English, please click here
Graph Embedding Model over Review-Based Edges
Abstract:In order to solve the problem of insufficient information utilization in user and item graph embedding training of recommendation system, The model RHG is proposed to train user and item graph embeddings by using reviews and interactive information in the recommendation system. Based on the edge based graph neural network, the model first constructs the edge in the graph by using the review text to aggregate the graph node information. Next, the model designs different feature spaces to improve the efficiency of the node information aggregation, and then a weighted loss function is proposed according to the importance of different nodes to optimize the training. Finally, on Amazon\'s three real datasets, the graph embedding effect is evaluated by the scoring task in the recommendation system. The mean square error of RHG is reduced by 2.75% compared with the current state-of-the-art model CGAT, which proves the effectiveness of the model.
Keywords: Pattern recognition and intelligent system graph embedding review text feature spaces
引用
No.****
同行评议
共计0人参与
勘误表
基于评论构建边的图嵌入模型
评论
全部评论0/1000