Memory Augmented Graph Neural Networks for Sequential Recommendation

Authors

  • Chen Ma McGill University
  • Liheng Ma McGill University
  • Yingxue Zhang Huawei Noah's Ark Lab in Montreal
  • Jianing Sun Huawei Noah's Ark Lab in Montreal
  • Xue Liu McGill University
  • Mark Coates McGill University

DOI:

https://doi.org/10.1609/aaai.v34i04.5945

Abstract

The chronological order of user-item interactions can reveal time-evolving and sequential user behaviors in many recommender systems. The items that users will interact with may depend on the items accessed in the past. However, the substantial increase of users and items makes sequential recommender systems still face non-trivial challenges: (1) the hardness of modeling the short-term user interests; (2) the difficulty of capturing the long-term user interests; (3) the effective modeling of item co-occurrence patterns. To tackle these challenges, we propose a memory augmented graph neural network (MA-GNN) to capture both the long- and short-term user interests. Specifically, we apply a graph neural network to model the item contextual information within a short-term period and utilize a shared memory network to capture the long-range dependencies between items. In addition to the modeling of user interests, we employ a bilinear function to capture the co-occurrence patterns of related items. We extensively evaluate our model on five real-world datasets, comparing with several state-of-the-art methods and using a variety of performance metrics. The experimental results demonstrate the effectiveness of our model for the task of Top-K sequential recommendation.

Downloads

Published

2020-04-03

How to Cite

Ma, C., Ma, L., Zhang, Y., Sun, J., Liu, X., & Coates, M. (2020). Memory Augmented Graph Neural Networks for Sequential Recommendation. Proceedings of the AAAI Conference on Artificial Intelligence, 34(04), 5045-5052. https://doi.org/10.1609/aaai.v34i04.5945

Issue

Section

AAAI Technical Track: Machine Learning