MGU-GNN: Minimal Gated Unit based Graph Neural Network for Session-based Recommendation

被引:8
作者
Kumar, Chhotelal [1 ]
Abuzar, Md [1 ]
Kumar, Mukesh [1 ]
机构
[1] Natl Inst Technol, Dept Comp Sci & Engn, Patna 800005, Bihar, India
关键词
Session-based recommender system; Next item recommendation; Minimal gated unit; Graph neural network; Gated recurrent unit;
D O I
10.1007/s10489-023-04679-1
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Session-based recommender systems (SBRS) play a crucial role in predicting the next click of a user from anonymous session data on numerous online platforms such as e-commerce, music, etc. However, predicting the next click is a very challenging task within the session, as it contains a very little amount of contextual information. Most of the existing techniques consider a session like a sequence of items to make recommendations and ignore the complex transition between items. In order to get accurate item embeddings and capture the complex transitions of items, we have proposed a Minimal Gated Unit based Graph Neural Network (MGU-GNN) for the session-based recommendation (SBR) tasks. We have also integrated a soft-attention network and target-based interest-aware network module, called MGU-GNN-TAR. The target-based interest-aware network module adapts to varying users' interests in terms of the items to be targeted. The soft-attention network module adapts long-term priorities and current session interest for better prediction of the user's next item or action. This model provides precise item embedding by incorporating the complex item transitions. The proposed model uses a gated mechanism called the Minimal Gated Unit, which has a single gate, and due to this reason, the parameters have been reduced to 67% as compared to the GRU cell. A GRU cell is the most basic of all gated hidden units. To demonstrate the efficacy of the proposed models, comprehensive experiments on four most commonly used publicly available real-world datasets have been performed, and they show that the proposed models routinely beat baseline methods and state-of-the-art SBR techniques on all four datasets.
引用
收藏
页码:23147 / 23165
页数:19
相关论文
共 37 条
  • [1] Dynamic Graph Attention-Aware Networks for Session-Based Recommendation
    Abugabah, Ahed
    Cheng, Xiaochun
    Wang, Jianfeng
    [J]. 2020 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2020,
  • [2] Handling Information Loss of Graph Neural Networks for Session-based Recommendation
    Chen, Tianwen
    Wong, Raymond Chi-Wing
    [J]. KDD '20: PROCEEDINGS OF THE 26TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2020, : 1172 - 1180
  • [3] Improving performance and efficiency of Graph Neural Networks by injective aggregation
    Dong, Wei
    Wu, Junsheng
    Zhang, Xinwan
    Bai, Zongwen
    Wang, Peng
    Wozniak, Marcin
    [J]. KNOWLEDGE-BASED SYSTEMS, 2022, 254
  • [4] Denoising Aggregation of Graph Neural Networks by Using Principal Component Analysis
    Dong, Wei
    Wozniak, Marcin
    Wu, Junsheng
    Li, Weigang
    Bai, Zongwen
    [J]. IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2023, 19 (03) : 2385 - 2394
  • [5] Sequence and Time Aware Neighborhood for Session-based Recommendations
    Garg, Diksha
    Gupta, Priyanka
    Malhotra, Pankaj
    Vig, Lovekesh
    Shroff, Gautam
    [J]. PROCEEDINGS OF THE 42ND INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '19), 2019, : 1069 - 1072
  • [6] Ghim-Eng Yap, 2012, Database Systems for Advanced Applications. Proceedings 17th International Conference, DASFAA 2012, P48, DOI 10.1007/978-3-642-29035-0_4
  • [7] 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
  • [8] User session interaction-based recommendation system using various machine learning techniques
    Kumar, Chhotelal
    Kumar, Mukesh
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (14) : 21279 - 21309
  • [9] Neural Attentive Session-based Recommendation
    Li, Jing
    Ren, Pengjie
    Chen, Zhumin
    Ren, Zhaochun
    Lian, Tao
    Ma, Jun
    [J]. CIKM'17: PROCEEDINGS OF THE 2017 ACM CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, 2017, : 1419 - 1428
  • [10] STAMP: Short-Term Attention/Memory Priority Model for Session-based Recommendation
    Liu, Qiao
    Zeng, Yifu
    Mokhosi, Refuoe
    Zhang, Haibin
    [J]. KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, : 1831 - 1839