GCMT:基于特征增强的图卷积多任务推荐模型
李国轩1
雒伟群2
罗丽锦1
张子健1
陆敬蔚1
1.西藏民族大学信息工程学院,咸阳 7120822.西藏民族大学信息工程学院,咸阳 712082;西藏光信息处理与可视化技术重点实验室,咸阳 712082
摘要：图卷积神经网络在消息聚合和传播方面的先进能力,极大地提升了基于知识图谱的推荐系统的准确性.尽管如此,目前许多知识图谱卷积方法并未深入挖掘用户兴趣,也未能对卷积后的实体特征进行有效的三元组建模.针对这些局限性,文章提出一种基于特征增强的图卷积多任务推荐模型.该模型首先利用BERT预训练模型对实体的初始表示进行建模,随后通过嵌入注意力机制的图卷积神经网络对节点特征进行传播与更新.进一步地,引入三元组关系建模策略对更新后的实体向量进行表示学习.在预测阶段,利用DNN网络融合用户和实体两个特征向量,以实现更精准的推荐.为了验证模型的有效性,笔者在推荐领域的常用公共数据集上进行了广泛的实验.实验结果表明,与KGCN模型相比,在Movielens-20M数据集上,模型在AUC和F1上分别实现了 1.5％和1.3％的提升;在Book-Crossing数据集上,AUC和F1值分别提升了 1.8％和1.0％.在自建数据集上,模型的AUC和F1值,分别提升了 4.8％和3.6％.这些显著的性能提升证明了我们的模型在推荐系统领域的卓越性能和应用潜力.
关键词：知识图谱多任务推荐算法预训练模型图卷积神经网络
资助基金：西藏民族大学校内项目(24MDY07)西藏光信息处理与可视化技术重点实验室开放基金项目(KLFPXZMU2303)西藏自治区科技重大专项(XZ202201ZD0002G03)
论文发表日期：2025-02-28
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：9( 72-80 )
英文信息