Graph neural network recommendation algorithm based on improved dual tower model

Qiang He,Xinkai Li,Biao Cai
DOI: https://doi.org/10.1038/s41598-024-54376-3
IF: 4.6
2024-02-17
Scientific Reports
Abstract: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.
multidisciplinary sciences
What problem does this paper attempt to address?
### Problems the Paper Attempts to Solve The paper aims to address the challenges faced by recommendation systems in the era of information explosion, specifically the pursuit of broad recall while maintaining accuracy. Although the current dual-tower neural network models in recommendation systems are efficient in large-scale candidate record screening, they fail to effectively model the interactions between features due to the independent training of user and item towers, resulting in poor recommendation accuracy. To solve this problem, the authors propose a new recommendation algorithm model—the Interactive High-Order Dual-Tower (IHDT) model. ### Main Contributions and Methods 1. **Model Design**: - Proposed an IHDT model based on heterogeneous graph construction, extracting richer feature representations through meta-paths. - Introduced an interaction learning mechanism, injecting relevant features between the user and item towers. - Utilized graph convolutional networks for high-order feature learning, aggregating node embeddings to obtain enhanced final representations. 2. **Experimental Validation**: - Evaluated on the MovieLens dataset, showing superior performance across multiple benchmark methods. - Validated the effectiveness of the interaction learning and high-order graph convolutional network components through ablation experiments. ### Analysis of Experimental Results - **Performance Comparison**: The IHDT algorithm outperforms benchmark algorithms on all metrics, particularly excelling on sparse datasets. - **Impact of Recommendation List Length**: As the length of the recommendation list increases, recall and NDCG metrics show an upward trend, while precision metrics smoothly decline. - **Parameter Sensitivity Study**: Investigated the impact of key parameters such as the number of model layers, representation vector dimensions, learning rate, and L2 regularization coefficient on model performance. In summary, the paper proposes a novel recommendation system model, IHDT, aimed at improving the accuracy and diversity of recommendation systems, especially suitable for large-scale recommendation scenarios.