HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Neural Graph Collaborative Filtering

Xiang Wang; Xiangnan He; Meng Wang; Fuli Feng; Tat-Seng Chua

Neural Graph Collaborative Filtering

Abstract

Learning vector representations (aka. embeddings) of users and items lies at the core of modern recommender systems. Ranging from early matrix factorization to recently emerged deep learning based methods, existing efforts typically obtain a user's (or an item's) embedding by mapping from pre-existing features that describe the user (or the item), such as ID and attributes. We argue that an inherent drawback of such methods is that, the collaborative signal, which is latent in user-item interactions, is not encoded in the embedding process. As such, the resultant embeddings may not be sufficient to capture the collaborative filtering effect. In this work, we propose to integrate the user-item interactions -- more specifically the bipartite graph structure -- into the embedding process. We develop a new recommendation framework Neural Graph Collaborative Filtering (NGCF), which exploits the user-item graph structure by propagating embeddings on it. This leads to the expressive modeling of high-order connectivity in user-item graph, effectively injecting the collaborative signal into the embedding process in an explicit manner. We conduct extensive experiments on three public benchmarks, demonstrating significant improvements over several state-of-the-art models like HOP-Rec and Collaborative Memory Network. Further analysis verifies the importance of embedding propagation for learning better user and item representations, justifying the rationality and effectiveness of NGCF. Codes are available at https://github.com/xiangwang1223/neural_graph_collaborative_filtering.

Code Repositories

zzz2010/ngcf-paddle2
paddle
Mentioned in GitHub
weiyinwei/mmgcn
pytorch
Mentioned in GitHub
immabe/NGCF_pytorch
pytorch
Mentioned in GitHub
madansinghal/graphranko
tf
Mentioned in GitHub
huangtinglin/NGCF-PyTorch
pytorch
Mentioned in GitHub
weiyinwei/huign
pytorch
Mentioned in GitHub
CHFsky/ngcf-paddle
paddle
Mentioned in GitHub
jinfeng-xu/fkan-gcf
pytorch
Mentioned in GitHub
florianeBhz/NGCF_Pytorch
pytorch
Mentioned in GitHub
chfhf/ngcf-paddle
paddle
Mentioned in GitHub
talkingwallace/NGCF-pytorch
pytorch
Mentioned in GitHub
lt610/DAGNN
pytorch
Mentioned in GitHub
xiaoleiHou214/BGCF-pytorch-master
pytorch
Mentioned in GitHub
MiloudBi/NGCF-pytorch
pytorch
Mentioned in GitHub
PreferredAI/cornac
tf
Mentioned in GitHub
massquantity/LibRecommender
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-gowallaNGCF
Recall@20: 0.1570
link-prediction-on-movielens-25mNGCF
Hits@10: 0.7807
nDCG@10: 0.4866
recommendation-systems-on-amazon-bookNGCF
Recall@20: 0.0344
nDCG@20: 0.0263
recommendation-systems-on-gowallaNGCF
Recall@20: 0.1570
nDCG@20: 0.1327

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Neural Graph Collaborative Filtering | Papers | HyperAI