Graph neural network recommendation algorithm based on improved dual tower model

被引:0
|
作者
He, Qiang [1 ]
Li, Xinkai [2 ]
Cai, Biao [2 ,3 ]
机构
[1] Chengdu Univ Technol, Sch Mech & Elect Engn, Chengdu 610059, Peoples R China
[2] Chengdu Univ Technol, Sch Comp Sci & Cyber Secur, Chengdu 610059, Peoples R China
[3] Chengdu Univ Technol, Coll Ind Technol, Yibin 644000, Peoples R China
基金
中国国家自然科学基金;
关键词
Recommendation; Dual tower model; Graph neural network; Collaborative filtering; MATRIX FACTORIZATION;
D O I
10.1038/s41598-024-54376-3
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
In this era of information explosion, recommendation systems play a key role in helping users to uncover content of interest among massive amounts of information. Pursuing a breadth of recall while maintaining accuracy is a core challenge for current recommendation systems. In this paper, we propose a new recommendation algorithm model, the interactive higher-order dual tower (IHDT), which improves current models by adding interactivity and higher-order feature learning between the dual tower neural networks. A heterogeneous graph is constructed containing different types of nodes, such as users, items, and attributes, extracting richer feature representations through meta-paths. To achieve feature interaction, an interactive learning mechanism is introduced to inject relevant features between the user and project towers. Additionally, this method utilizes graph convolutional networks for higher-order feature learning, pooling the node embeddings of the twin towers to obtain enhanced end-user and item representations. IHDT was evaluated on the MovieLens dataset and outperformed multiple baseline methods. Ablation experiments verified the contribution of interactive learning and high-order GCN components.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] A Social Recommendation Algorithm Based on Graph Neural Network
    Lyu Y.-X.
    Hao S.
    Qiao G.-T.
    Xing Y.
    Dongbei Daxue Xuebao/Journal of Northeastern University, 2024, 45 (01): : 10 - 17
  • [2] Research on Recommendation Algorithm Based on Heterogeneous Graph neural Network
    Chen Z.
    Li H.
    Du J.
    Hunan Daxue Xuebao/Journal of Hunan University Natural Sciences, 2021, 48 (10): : 137 - 144
  • [3] Session-Based Recommendation Model with Self Contrastive Graph Neural Network and Dual Predictor
    Zhang Y.
    Xia H.
    Liu Y.
    Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2024, 37 (03): : 242 - 252
  • [4] A Session Recommendation Model Based on Heterogeneous Graph Neural Network
    An, Zhiwei
    Tan, Yirui
    Zhang, Jinli
    Jiang, Zongli
    Li, Chen
    KNOWLEDGE SCIENCE, ENGINEERING AND MANAGEMENT, PT III, KSEM 2023, 2023, 14119 : 160 - 171
  • [5] Hierarchical Social Recommendation Model Based on a Graph Neural Network
    Bi, Zhongqin
    Jing, Lina
    Shan, Meijing
    Dou, Shuming
    Wang, Shiyang
    WIRELESS COMMUNICATIONS & MOBILE COMPUTING, 2021, 2021
  • [6] Multi-relation Neural Network Recommendation Model Based on Knowledge Graph Embedding Algorithm
    Liu, Hongpu
    Jiang, Jingfei
    Wang, Kaixin
    Kong, Lingshu
    Wang, Jingshu
    KNOWLEDGE SCIENCE, ENGINEERING AND MANAGEMENT, PT I, KSEM 2024, 2024, 14884 : 228 - 239
  • [7] DualGNN: Dual Graph Neural Network for Multimedia Recommendation
    Wang, Qifan
    Wei, Yinwei
    Yin, Jianhua
    Wu, Jianlong
    Song, Xuemeng
    Nie, Liqiang
    IEEE TRANSACTIONS ON MULTIMEDIA, 2023, 25 : 1074 - 1084
  • [8] Research on Graph Network Recommendation Algorithm Based on Random Walk and Convolutional Neural Network
    Huang, Meng
    2021 INTERNATIONAL CONFERENCE ON HIGH PERFORMANCE BIG DATA AND INTELLIGENT SYSTEMS (HPBD&IS), 2021, : 57 - 64
  • [9] An Improved Visual SLAM Algorithm Based on Graph Neural Network
    Wang, Wei
    Xu, Tao
    Xing, Kaisheng
    Liu, Jinhui
    Chen, Mengyuan
    IEEE ACCESS, 2023, 11 : 102366 - 102380
  • [10] Knowledge Graph Double Interaction Graph Neural Network for Recommendation Algorithm
    Kang, Shuang
    Shi, Lin
    Zhang, Zhenyou
    APPLIED SCIENCES-BASEL, 2022, 12 (24):