HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

UltraGCN: Ultra Simplification of Graph Convolutional Networks for Recommendation

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

UltraGCN: Ultra Simplification of Graph Convolutional Networks for Recommendation

Abstract

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.

Code Repositories

shuyao-wang/dsl
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-gowallaEmb-GCN
Recall@20: 0.1862
collaborative-filtering-on-gowallaUltraGCN
NDCG@20: 0.1580
Recall@20: 0.1862
collaborative-filtering-on-yelp2018UltraGCN
NDCG@20: 0.0561
Recall@20: 0.0683
recommendation-systems-on-amazon-bookEmb-GCN
Recall@20: 0.0681
nDCG@20: 0.0556
recommendation-systems-on-gowallaEmb-GCN
Recall@20: 0.1862
nDCG@20: 0.1580

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
UltraGCN: Ultra Simplification of Graph Convolutional Networks for Recommendation | Papers | HyperAI