當前位置:首頁 > 新聞中心 > 科研動態

科研動態

智能感知與計算研究中心提出基于圖神經網絡的會話序列推薦模型

  • 发表日期:2019-07-08 【 【打印】【關閉】
  •   我所智能感知與計算研究中心提出並開源了基于圖神經網絡的會話序列 (session-based) 推薦模型SR-GNN 

      會話 (session) 是服务器端用来记录识别用户的一种机制。在推荐系统中,基于會話的推荐 (session-based recommendation) 是依據一個時間窗口內用戶連續的行爲進行推薦。比如,一位用戶在登錄淘寶的這段時間裏連續點擊了10個商品,那麽這10个商品就构成了一个短序列。由于會話数据具有海量、匿名的特点,基于會話的推荐近来受到了广泛的关注。 

      针对该问题,智能感知与计算研究中心吴书、朱彦樵等人提出一种基于图神经网络的會話序列推薦模型,目前成为该问题的最优模型,相关成果已经发表在人工智能顶级会议AAAI-19上。 

      之前的方法通过循环神经网络等来对會話进行序列化建模,较难得到准确的用户表达向量,并且忽略了用户点击项中可能存在的复杂的转换特性。因此,为了更好地捕获會話的结构并考虑节点间的转换,吴书等人提出了一种基于图神经网络 (graph neural network, GNN) 的會話推荐方法。该方法将所有會話序列聚合在一起并建模为图结构,基于该图,使用图神经网络来捕获项目的转换关系,之后使用注意力机制 (attention mechanism) 将每个會話表示为全局偏好和會話当前兴趣的组合并据此进行推荐。该模型在多个公开数据集上进行了实验,效果均好于当前的state-of-the-art方法。 

      該模型已在百度PaddlePaddle平台上開源。

       

    會話模型

      論文: Shu Wu, Yuyuan Tang, Yanqiao Zhu, Liang Wang, Xing Xie, and Tieniu Tan, Session-based Recommendation with Graph Neural Networks, in AAAI 2019.    

      開源項目地址:https://github.com/PaddlePaddle/models/tree/develop/PaddleRec/gnn