Graph neural network recommendation algorithm based on improved dual tower model

被引:2
|
作者
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] GNNCL: A Graph Neural Network Recommendation Model Based on Contrastive Learning
    Chen, Jinguang
    Zhou, Jiahe
    Ma, Lili
    NEURAL PROCESSING LETTERS, 2024, 56 (02)
  • [4] A Collaborative Filtering Recommendation Algorithm Based on Community Detection and Graph Neural Network
    Sheng, Jinfang
    Liu, Qingqing
    Hou, Zhengang
    Wang, Bin
    NEURAL PROCESSING LETTERS, 2023, 55 (06) : 7095 - 7112
  • [5] A Collaborative Filtering Recommendation Algorithm Based on Community Detection and Graph Neural Network
    Jinfang Sheng
    Qingqing Liu
    Zhengang Hou
    Bin Wang
    Neural Processing Letters, 2023, 55 : 7095 - 7112
  • [6] GNNCL: A Graph Neural Network Recommendation Model Based on Contrastive Learning
    Jinguang Chen
    Jiahe Zhou
    Lili Ma
    Neural Processing Letters, 56
  • [7] 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
  • [8] 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
  • [9] 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
  • [10] Course Recommendation Based on Graph Convolutional Neural Network
    An Cong Tran
    Duc-Thien Tran
    Nguyen Thai-Nghe
    Tran Thanh Dien
    Hai Thanh Nguyen
    ADVANCES AND TRENDS IN ARTIFICIAL INTELLIGENCE. THEORY AND APPLICATIONS, IEA/AIE 2023, PT I, 2023, 13925 : 235 - 240