---

RecBole (伯乐) 2.0

“世有伯乐，然后有千里马。千里马常有，而伯乐不常有。”——韩愈《马说》

RecBole 1.0 | HomePage | Datasets | Paper

Based on a popular recommendation framework RecBole, we develop an extended recommendation library called RecBole 2.0, consisting of benchmarking packages for up-to-date topics and architectures.

RecBole 2.0 includes 8 packages covering the up-to-date research topic in recommender system:

• Data augmentation (RecBole-DA)
• Meta recommendation (RecBole-MetaRec)
• Debiased recommendation (RecBole-Debias)
• Fairness-aware recommendation (RecBole-FairRec)
• Cross-domain recommendation (RecBole-CDR)
• Graph-based recommendation (RecBole-GNN)
• Transformer-based recommendation (RecBole-TRM)
• Person-job fit (RecBole-PJF)

For each package, we provide complete implementations from data loading, experimental setup, evaluation and algorithm implementation. This library is of great value to facilitate the up-to-date research in recommender systems.

RecBole-DA

RecBole-DA is a library built upon PyTorch and RecBole for reproducing and developing data augmentation for sequential recommendation.

1）Highlights

• Easy-to-use API: Our library provides extensive API based on common data augmentation strategies, users can further develop own new models based on our library.
• Full Coverage of Classic Methods: We provide seven data augmentation methods based on recommender systems in three major categories.

2）Implemented Models

Our library includes algorithms covering three major categories:

• Heuristic-based Methods: CL4SRec, DuoRec
• Model-based Methods: MMInfoRec, CauseRec
• Hybird Methods: CASR, CCL, CoSeRec

3）The Team

RecBole-DA is developed and maintained by members from RUCAIBox, the developer is Shuqing Bian (@fancybian).

RecBole-MetaRec

RecBole-MetaRec is an extended package for RecBole, which aims to help researchers to compare and develop their own models in the field of meta learning recommendation.

1) Highlights

The package can mainly provide researchers with the following advantages:

• Conveniently develop new meta learning recommendation models with the general meta learning framework.
• Conveniently learn and compare the meta learning recommendation models that we have implemented.
• Conveniently use the advantages and features of RecBole.

Moreover, we provide a document in detail for researchers.

2) Implemented Models

Our package includes three main types of algorithms:

• Meta learn to predict: MeLU, MAMO
• Meta learn to parameterize: LWA, NLBA, TaNP
• Meta learn to embed: MetaEmb, MWUF

3) Extended Modules

(1) MetaDataset: the meta learning task splitter. (2) MetaDataLoader: the meta learning task translator. (3) MetaRecommender: the template for meta learning models. (4) MetaTrainer: the base trainer for meta learning training process. (5) MetaCollector: the evaluation class for meta learning tasks. (6) MetaUtils: the toolkit for meta learning.

4) The Team

RecBole-MetaRec is developed and maintained by Zeyu Zhang (@Zeyu Zhang).

RecBole-Debias

RecBole-Debias is a toolkit built upon RecBole for reproducing and developing debiased recommendation algorithms.

1）Highlights

• Unified

  A unified framework, which includes several algorithms for different kinds of biases. Meanwhile, three datasets with the different distributions of training set and test set are provided for training and evaluation.

• Adaptive

  Adaptive to many base recommendation models. For simplicity, the current implementation is only based on MF model.

• Closely

  Closely related to Recbole. The toolkit fully adopts the functions of Recbole, except that certain algorithms need to design unique components like trainer.

2）Implemented Models

We list currently supported models according to category:

• Base Model: MF
• Selection Bias: MF-IPS
• Popularity Bias: PDA, MACR, DICE, CausE
• Exposure Bias: Rel-MF

3）The Team

RecBole-Debias is developed and maintained by members from RUCAIBox, the main developers is Jingsen Zhang (@Jingsen Zhang).

RecBole-FairRec

RecBole-FairRec is a library toolkit built upon PyTorch and RecBole for reproducing and developing fairness-aware recommendation algorithms.

1）Highlights

• Easy-to-use: Our library shares unified API and input(atomic files) as RecBole.
• Conveniently learn and compare: Our library provides several fairess-metrics and frameworks for learning and comparing.
• Extensive FairRec library: Recently proposed fairness-aware algorithms can be easily equipped in our library.

2）Implemented Models

We list the models and fairness-metrics that we have implemented up to now:

• Models: FOCF, PFCN(including PFCN_MLP, PFCN_BiasedMF, PFCN_DMF, PFCN_PMF), FairGo(including FairGo_PMF_WAP, FairGo_PMF_LBA, FairGo_PMF_LVA, FairGo_GCN_WAP, FairGo_GCN_LBA, FairGo_GCN_LVA), NFCF
• Fairness-Metrics:
  • Item-Oriented: GiniIndex, PopularityPercentage
  • User-Oriented: DifferentialFairness,ValueUnfairness, AbsoluteUnfairness, UnderUnfairness, OverUnfairness, NonParityUnfairness

3）The Team

RecBole-FairRec is developed and maintained by Jiakai Tang (@Jiakai Tang).

RecBole-CDR

RecBole-CDR is a library built upon RecBole for reproducing and developing cross-domain recommendation algorithms.

1) Highlights

• Automatic and compatible data processing for cross-domain recommendation: Our library designs a unified data structure for cross-domain recommendation, which inherits all the data pre-processing strategies in RecBole. The overlapped data in different domains can be matched automatically.
• Flexible and customized model training strategies: Our library provides four basic training modes for cross-domain recommendation, which can be combined arbitrarily by users. It is also easy to customize training strategy in original way.
• Extensive cross-domain recommendation algorithms: Based on unified data structure and flexible training strategies, several cross-domain recommendation algorithms are implemented and compared with others fairly.

2) Implemented Models

Our library includes algorithms covering three major categories:

• Algorithms based on the collective matrix factorization, such as CMF and CLFM.
• Algorithms that share or combine the representations of the overlapped data, for example, DTCDR, DeepAPF and NATR.
• Algorithms that transfer or map knowledge between different domains, such as CoNet, BiTGCF, EMCDR, SSCDR and DCDCSR.

3) The Team

RecBole-CDR is developed and maintained by members from RUCAIBox, the main developers are Zihan Lin (@linzihan-backforward), Gaowei Zhang (@Wicknight) and Shanlei Mu (@ShanleiMu).

RecBole-GNN

RecBole-GNN is a library built upon PyTorch and RecBole for reproducing and developing recommendation algorithms based on graph neural networks (GNNs).

1）Highlights

• Easy-to-use and unified API: Our library shares unified API and input (atomic files) as RecBole.
• Efficient and reusable graph processing: We provide highly efficient and reusable basic datasets, dataloaders and layers for graph processing and learning.
• Extensive graph library: Graph neural networks from widely-used library like PyG are incorporated. Recently proposed graph algorithms can be easily equipped and compared with existing methods.

2）Implemented Models

Our library includes algorithms covering three major categories:

• General Recommendation: NGCF, LightGCN, SGL, HMLET, NCL, SimGCL
• Sequential Recommendation: SR-GNN, GC-SAN, NISER, LESSR, TAGNN, GCE-GNN, SGNN-HN
• Social Recommendation: DiffNet, MHCN, SEPT

3）The Team

RecBole-GNN is developed and maintained by members from RUCAIBox, the main developers are Yupeng Hou (@hyp1231), Lanling Xu (@Sherry-XLL) and Changxin Tian (@ChangxinTian).

RecBole-TRM

RecBole-TRM is a library built upon [PyTorch](https://pytorc