Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

UltraGCN: Ultra Simplification of Graph Convolutional Networks for Recommendation

About

With the recent success of graph convolutional networks (GCNs), they have been widely applied for recommendation, and achieved impressive performance gains. The core of GCNs lies in its message passing mechanism to aggregate neighborhood information. However, we observed that message passing largely slows down the convergence of GCNs during training, especially for large-scale recommender systems, which hinders their wide adoption. LightGCN makes an early attempt to simplify GCNs for collaborative filtering by omitting feature transformations and nonlinear activations. In this paper, we take one step further to propose an ultra-simplified formulation of GCNs (dubbed UltraGCN), which skips infinite layers of message passing for efficient recommendation. Instead of explicit message passing, UltraGCN resorts to directly approximate the limit of infinite-layer graph convolutions via a constraint loss. Meanwhile, UltraGCN allows for more appropriate edge weight assignments and flexible adjustment of the relative importances among different types of relationships. This finally yields a simple yet effective UltraGCN model, which is easy to implement and efficient to train. Experimental results on four benchmark datasets show that UltraGCN not only outperforms the state-of-the-art GCN models but also achieves more than 10x speedup over LightGCN. Our source code will be available at https://reczoo.github.io/UltraGCN.

Kelong Mao, Jieming Zhu, Xi Xiao, Biao Lu, Zhaowei Wang, Xiuqiang He• 2021

Related benchmarks

TaskDatasetResultRank
RecommendationGowalla (test)
Recall@200.1862
126
RecommendationAmazon-Book (test)
Recall@200.0681
101
RecommendationGowalla
Recall@200.0683
100
RecommendationYelp 2018 (test)
Recall@206.83
90
Collaborative FilteringYelp 2018
NDCG@205.61
42
Collaborative FilteringGowalla
NDCG@200.1589
40
Collaborative FilteringAmazon Books
NDCG@205.56
39
Collaborative FilteringAmazon-Book (test)
Recall@206.81
35
Collaborative FilteringYelp 2018 (test)
Recall@206.83
35
RecommendationMovieLens
Recall@2027.87
14
Showing 10 of 17 rows

Other info

Code

Follow for update