HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SimpleX: A Simple and Strong Baseline for Collaborative Filtering

Kelong Mao Jieming Zhu Jinpeng Wang Quanyu Dai Zhenhua Dong Xi Xiao Xiuqiang He

SimpleX: A Simple and Strong Baseline for Collaborative Filtering

Abstract

Collaborative filtering (CF) is a widely studied research topic in recommender systems. The learning of a CF model generally depends on three major components, namely interaction encoder, loss function, and negative sampling. While many existing studies focus on the design of more powerful interaction encoders, the impacts of loss functions and negative sampling ratios have not yet been well explored. In this work, we show that the choice of loss function as well as negative sampling ratio is equivalently important. More specifically, we propose the cosine contrastive loss (CCL) and further incorporate it to a simple unified CF model, dubbed SimpleX. Extensive experiments have been conducted on 11 benchmark datasets and compared with 29 existing CF models in total. Surprisingly, the results show that, under our CCL loss and a large negative sampling ratio, SimpleX can surpass most sophisticated state-of-the-art models by a large margin (e.g., max 48.5% improvement in NDCG@20 over LightGCN). We believe that SimpleX could not only serve as a simple strong baseline to foster future research on CF, but also shed light on the potential research direction towards improving loss function and negative sampling. Our source code will be available at https://reczoo.github.io/SimpleX.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-gowallaSimpleX
NDCG@20: 0.1557
Recall@20: 0.1872
collaborative-filtering-on-yelp2018SimpleX
NDCG@20: 0.0575
Recall@20: 0.0701
recommendation-systems-on-amazon-bookSimpleX
Recall@20: 0.0583
nDCG@20: 0.0468
recommendation-systems-on-gowallaSimpleX
Recall@20: 0.1872
nDCG@20: 0.1557
recommendation-systems-on-yelp2018SimpleX
NDCG@20: 0.0575
Recall@20: 0.0701

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