GRecX

December 6, 2021 ยท View on GitHub

GRecX

An Efficient and Unified Benchmark for GNN-based Recommendation.

Homepage and Documentation

Example Benchmark: Performance on Yelp and Gowalla with BPR Loss

Performance on Yelp with BPR Loss:

Performance on Gowalla with BPR Loss:

Demo

We recommend you get started with some demos.

Preliminary Comparison

LightGCN-Yelp dataset (featureless)

  • BCE-loss
AlgoPrecision@10Precision@20Recall@10Recall@20nDCG@10nDCG@20
MF0.0295970.0254950.0327330.0560860.0373320.045805
NGCF0.0247130.0218930.0282510.0496110.0313570.039549
LightGCN------------0.0373500.045872
UltraGCN-single0.0306520.0267900.0339130.0588860.0385760.047766
UltraGCN0.035530.0303460.0395260.0670280.0453650.055376
  • BPR-loss
AlgoPrecision@10Precision@20Recall@10Recall@20nDCG@10nDCG@20
MF0.0314890.0273030.0347330.0603330.0401030.049406
NGCF0.0303750.0266990.0345020.0599840.0387320.048351
LightGCN0.0335440.0289960.0372770.0641280.0429070.052667
UltraGCN-single------------------
UltraGCN------------------

Note that "UltraGCN-single" uses loss with one negative sample and one negatvie loss weight

Cite

If you use GRecX in a scientific publication, we would appreciate citations to the following paper:

@misc{cai2021grecx,
title={GRecX: An Efficient and Unified Benchmark for GNN-based Recommendation},
author={Desheng Cai and Jun Hu and Shengsheng Qian and Quan Fang and Quan Zhao and Changsheng Xu},
year={2021},
eprint={2111.10342},
archivePrefix={arXiv},
primaryClass={cs.IR}
}