预训练 源码

图神经预训练,用于带有辅助信息的推荐 匿名提交的代码:带有辅助信息的推荐图神经网络预训练 介绍 利用与实体(即用户和项目)关联的辅助信息来增强推荐系统的性能已被广泛认为是重要的建模维度。 尽管许多现有方法都集中于通过整合建议损失功能和额外的辅助信息感知损失来整合实体辅助信息的集成方案,但在本文中,我们还是提出了一种利用辅助信息的新型预训练方案。 特别是,我们首先使用实体​​的边信息对表示模型进行预训练,然后使用现有的基于通用表示的推荐模型对其进行微调。 具体来说,我们通过考虑从边信息分别构造为两种不同类型的图的实体及其关系,提出两种模型,分别称为GCN-P和COM-P,以对实体嵌入进行预训练。