Self-Attentive Sequential Recommendation

By Wang-Cheng Kang, Julian McAuley. Industrial Conference on Data Mining, 2018

3,858 citations recorded by the source corpus.

Sequential dynamics are a key feature of many modern recommender systems, which seek to capture the 'context' of users' activities on the basis of actions they have performed recently. To capture such patterns, two approaches have proliferated: Markov Chains (MCs) and Recurrent Neural Networks (RNNs). Markov Chains assume that a user's next action can be predicted on the basis of just their last (or last few) actions, while RNNs in principle allow for longer-term semantics to be uncovered. Generally speaking, MC-based methods perform best in extremely sparse datasets, where model parsimony is critical, while RNNs perform better in denser datasets where higher model complexity is affordable. The goal of our work is to balance these two goals, by proposing a self-attention based sequential model (SASRec) that allows us to capture long-term semantics (like an RNN), but, using an attention mechanism, makes its predictions based on relatively few actions (like an MC). At each time step, SASRec seeks to identify which items are 'relevant' from a user's action history, and use them to predict the next item. Extensive empirical studies show that our method outperforms various state-of-the-art sequential models (including MC/CNN/RNN-based approaches) on both sparse and dense datasets. Moreover, the model is an order of magnitude more efficient than comparable CNN/RNN-based models. Visualizations on attention weights also show how our model adaptively handles datasets with various density, and uncovers meaningful patterns in activity sequences.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Information Systems

View Paper: https://www.semanticscholar.org/paper/97faeefa771e8cc8e55159e2bd03e6f5eef249a8

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Xiangnan He
  2. Wayne Xin Zhao
  3. Xiang Wang
  4. Ji-rong Wen
  5. Tat-seng Chua
  6. Hongzhi Yin
  7. Yongfeng Zhang
  8. Meng Wang

In Computer Vision and Pattern Recognition

  1. Xiangnan He
  2. Xiang Wang
  3. Yongfeng Zhang
  4. Wayne Xin Zhao
  5. Meng Wang
  6. Ji-rong Wen
  7. Fuli Feng
  8. Tat-seng Chua

In Information Systems

  1. Xiangnan He
  2. Xiang Wang
  3. Wayne Xin Zhao
  4. Hongzhi Yin
  5. Yongfeng Zhang
  6. Meng Wang
  7. Ji-rong Wen
  8. Tat-seng Chua

Citations in the corpus