Combining Multi-Head Attention and Sparse Multi-Head Attention Networks for Session-Based Recommendation

被引:1
作者
Zhao, Zhiwei [1 ]
Wang, Xiaoye [1 ]
Xiao, Yingyuan [1 ]
机构
[1] Tianjin Univ Technol, Sch Comp Sci & Engn, Tianjin, Peoples R China
来源
2023 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, IJCNN | 2023年
关键词
session-based recommendation; multi-head attention; sparse multi-head attention;
D O I
10.1109/IJCNN54540.2023.10191924
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The core of session-based recommendation is to predict the next interactive item based on a set of anonymous user temporal or specified behavior sequences (e.g., click, browse or purchase item sequence), which is a key task of many online services today. Recently, self-attention networks have achieved remarkable success in the task of session-based recommendation. However, in session-based recommendation, some items may be clicked by mistake, and most of the current attention mechanisms assign weights to these items, resulting in the disadvantage of distraction. Although sparse attention networks can address the aforementioned issues, solely relying on sparse attention may in turn reduce the weight of some real-intent clicked items. Therefore, this paper proposes a model that combines multi-headed attention network and sparse multi-headed attention network, referred to as CMAN, which solves the drawback of assigning weights to items clicked by mistake in the traditional attention mechanism. And also prevents the drawback of reducing the weights of items that are truly clicked by some users brought by using sparse attention mechanism alone to some extent. Experiments on two real datasets show that the model outperforms some state-of-the-art models.
引用
收藏
页数:8
相关论文
共 24 条
[1]  
[Anonymous], 1990, Neurocomputing:Algorithms, architectures and applications, DOI DOI 10.1007/978-3-642-76153-9_28
[2]  
Cho K., 2014, P 2014 C EMP METH NA, P1724, DOI 10.3115/v1/d14-1179
[3]  
Fang J., 2021, ARXIV210201922
[4]  
Hidasi Balazs, 2015, CoRR
[5]   Self-Attentive Sequential Recommendation [J].
Kang, Wang-Cheng ;
McAuley, Julian .
2018 IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM), 2018, :197-206
[6]   MATRIX FACTORIZATION TECHNIQUES FOR RECOMMENDER SYSTEMS [J].
Koren, Yehuda ;
Bell, Robert ;
Volinsky, Chris .
COMPUTER, 2009, 42 (08) :30-37
[7]   Neural Attentive Session-based Recommendation [J].
Li, Jing ;
Ren, Pengjie ;
Chen, Zhumin ;
Ren, Zhaochun ;
Lian, Tao ;
Ma, Jun .
CIKM'17: PROCEEDINGS OF THE 2017 ACM CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, 2017, :1419-1428
[8]   STAMP: Short-Term Attention/Memory Priority Model for Session-based Recommendation [J].
Liu, Qiao ;
Zeng, Yifu ;
Mokhosi, Refuoe ;
Zhang, Haibin .
KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, :1831-1839
[9]   Evaluation of session-based recommendation algorithms [J].
Ludewig, Malte ;
Jannach, Dietmar .
USER MODELING AND USER-ADAPTED INTERACTION, 2018, 28 (4-5) :331-390
[10]  
Martins AFT, 2016, PR MACH LEARN RES, V48