Self-Attention Network for Session-Based Recommendation With Streaming Data Input

被引:19
作者
Sun, Shiming [1 ,2 ,3 ]
Tang, Yuanhe [1 ,2 ]
Dai, Zemei [1 ,2 ]
Zhou, Fu [1 ,2 ]
机构
[1] NARI Grp Corp, State Grid Elect Power Res Inst, Nanjing 211106, Jiangsu, Peoples R China
[2] NARI Technol Dev Ltd Co, Nanjing 211106, Jiangsu, Peoples R China
[3] NARI Grp Corp, State Key Lab Smart Grid Protect & Control, Nanjing 211106, Jiangsu, Peoples R China
关键词
Session-based recommendation; self-attention network; streaming data;
D O I
10.1109/ACCESS.2019.2931945
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In the current era of the rapid development of big data, it has become increasingly critical and practical to study recommender systems with streaming data input. However, the recommender system is often faced with the challenge that the history records of new users or anonymous users are not available. Specifically, session-based recommendation, which aims to predict a user's next actions, is a typical task to overcome the challenge. To capture a user's long-term preference in session-based recommendations, recurrent neural networks (RNN)-based models have been widely applied with impressive results, but the inherent sequential nature of RNNs prevents parallelism within training examples, which is critical in long sessions because memory constraints limit batching across instances. In this paper, we propose a novel method, i.e., self-attention network for session-based recommendation (SANSR), which is based on only attention mechanisms, dispensing with recurrence, and supports parallelism in the session. The proposed model attempts to find items that are relevant based on previous time steps in the ongoing session and to assign them different weights to predict the next item. The extensive experiments are conducted on two real-world datasets, and the experimental results show that our proposed model is superior to the state-of-the-art methods.
引用
收藏
页码:110499 / 110509
页数:11
相关论文
共 23 条
  • [1] Anastasopoulos Antonios, 2018, P NAACL
  • [2] [Anonymous], 2015, ARXIV151106939
  • [3] [Anonymous], 2016, ARXIV160903976
  • [4] [Anonymous], 2017, NEURIPS
  • [5] An Attribute-aware Neural Attentive Model for Next Basket Recommendation
    Bai, Ting
    Nie, Jian-Yun
    Zhao, Wayne Xin
    Zhu, Yutao
    Du, Pan
    Wen, Ji-Rong
    [J]. ACM/SIGIR PROCEEDINGS 2018, 2018, : 1201 - 1204
  • [6] Attention-driven Factor Model for Explainable Personalized Recommendation
    Chen, Jingwu
    Zhuang, Fuzhen
    Hong, Xin
    Ao, Xiang
    Xie, Xing
    He, Qing
    [J]. ACM/SIGIR PROCEEDINGS 2018, 2018, : 909 - 912
  • [7] Describing Multimedia Content Using Attention-Based Encoder-Decoder Networks
    Cho, Kyunghyun
    Courville, Aaron
    Bengio, Yoshua
    [J]. IEEE TRANSACTIONS ON MULTIMEDIA, 2015, 17 (11) : 1875 - 1886
  • [8] Increasing recommended effectiveness with markov chains and purchase intervals
    Gu, Wanrong
    Dong, Shoubin
    Zeng, Zhizhao
    [J]. NEURAL COMPUTING & APPLICATIONS, 2014, 25 (05) : 1153 - 1162
  • [9] Neural Collaborative Filtering
    He, Xiangnan
    Liao, Lizi
    Zhang, Hanwang
    Nie, Liqiang
    Hu, Xia
    Chua, Tat-Seng
    [J]. PROCEEDINGS OF THE 26TH INTERNATIONAL CONFERENCE ON WORLD WIDE WEB (WWW'17), 2017, : 173 - 182
  • [10] Parallel Recurrent Neural Network Architectures for Feature-rich Session-based Recommendations
    Hidasi, Balazs
    Quadrana, Massimo
    Karatzoglou, Alexandros
    Tikk, Domonkos
    [J]. PROCEEDINGS OF THE 10TH ACM CONFERENCE ON RECOMMENDER SYSTEMS (RECSYS'16), 2016, : 241 - 248