基于异质信息网络和多任务学习的推荐模型

    Recommendation Models Based on Heterogeneous Information Networks and Multi-task Learning

    • 摘要: 针对基于异质信息网络的推荐系统难以充分捕捉节点的内容信息以及基于元路径的异质信息挖掘存在链接丢失的问题, 提出一个基于异质信息网络和多任务学习的推荐方法.该方法首先在各个元路径视图上计算不同邻居实例对节点的影响程度, 挖掘元路径内部信息; 接着使用注意力机制学习异质信息网络图的语义信息, 得到异质信息网络中节点的嵌入; 最后采用多任务学习方法同时优化推荐任务和链路预测任务来解决链接丢失问题.在3个公开的异质数据集上进行实验, 结果表明该模型能够充分挖掘异质信息网络的信息, 在推荐任务和链路预测任务上的性能皆优于对比模型.

       

      Abstract: To address the problem that it is difficult to adequately capture the content information of nodes in recommendation systems based on heterogeneous information networks, and to tackle the problem of link loss in information mining of heterogeneous networks based on meta-path, a recommendation method was proposed based on heterogeneous information networks and multi-task learning. The influence degree of different neighbor instances on nodes on each meta-path view was first calculated to mine the internal information of meta-path, then attention mechanism was used to learn the semantic information of heterogeneous graphs to get the embedding of nodes in heterogeneous networks, and finally multi-task learning method was adopted to optimize both recommendation and link prediction tasks to alleviate the link loss calculated problem. Experimental results on three public heterogeneous data sets show that the model can fully leverage the heterogeneous information network semantic and structure information, and its performance in the recommendation and link prediction task are both better than the baselines.

       

    /

    返回文章
    返回