HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

LightGCN: Simplifying and Powering Graph Convolution Network for Recommendation

Xiangnan He Kuan Deng Xiang Wang Yan Li Yongdong Zhang Meng Wang

LightGCN: Simplifying and Powering Graph Convolution Network for Recommendation

Abstract

Graph Convolution Network (GCN) has become new state-of-the-art for collaborative filtering. Nevertheless, the reasons of its effectiveness for recommendation are not well understood. Existing work that adapts GCN to recommendation lacks thorough ablation analyses on GCN, which is originally designed for graph classification tasks and equipped with many neural network operations. However, we empirically find that the two most common designs in GCNs -- feature transformation and nonlinear activation -- contribute little to the performance of collaborative filtering. Even worse, including them adds to the difficulty of training and degrades recommendation performance. In this work, we aim to simplify the design of GCN to make it more concise and appropriate for recommendation. We propose a new model named LightGCN, including only the most essential component in GCN -- neighborhood aggregation -- for collaborative filtering. Specifically, LightGCN learns user and item embeddings by linearly propagating them on the user-item interaction graph, and uses the weighted sum of the embeddings learned at all layers as the final embedding. Such simple, linear, and neat model is much easier to implement and train, exhibiting substantial improvements (about 16.0\% relative improvement on average) over Neural Graph Collaborative Filtering (NGCF) -- a state-of-the-art GCN-based recommender model -- under exactly the same experimental setting. Further analyses are provided towards the rationality of the simple LightGCN from both analytical and empirical perspectives.

Code Repositories

lucapantea/LightGCN
pytorch
Mentioned in GitHub
LehengTHU/Agent4Rec
pytorch
Mentioned in GitHub
JiahaoWuGit/DcRec
pytorch
Mentioned in GitHub
apat1n/LightGCN-Pytorch
pytorch
Mentioned in GitHub
Trantin84/NGCF-Tin
tf
Mentioned in GitHub
yshenaw/GF_CF
pytorch
Mentioned in GitHub
jinfeng-xu/fkan-gcf
pytorch
Mentioned in GitHub
gusye1234/LightGCN-PyTorch
pytorch
Mentioned in GitHub
tanya525625/LightGCN-PyTorch
pytorch
Mentioned in GitHub
sayamsingla2000/LightGCN_MovieLens
pytorch
Mentioned in GitHub
shuyao-wang/dsl
pytorch
Mentioned in GitHub
PreferredAI/cornac
tf
Mentioned in GitHub
kuandeng/LightGCN
Official
tf
Mentioned in GitHub
gusye1234/pytorch-light-gcn
Official
pytorch
Mentioned in GitHub
massquantity/LibRecommender
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-amazon-bookLightGCN
NDCG@20: 0.0315
Recall@20: 0.0411
collaborative-filtering-on-gowallaLightGCN
NDCG@20: 0.1554
Recall@20: 0.1830
collaborative-filtering-on-yelp2018LightGCN
NDCG@20: 0.0530
Recall@20: 0.0649
recommendation-systems-on-amazon-bookLightGCN
Recall@20: 0.0411
nDCG@20: 0.0315
recommendation-systems-on-gowallaLightGCN
Recall@20: 0.1830
nDCG@20: 0.1554
recommendation-systems-on-yelp2018LightGCN
NDCG@20: 0.0530
Recall@20: 0.0649

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