RecBole 2.0: Towards a More Up-to-Date Recommendation Library
Wayne Xin Zhao,Chen Yang,Jingsen Zhang,Wenqi Sun,Gaowei Zhang,Shanlei Mu,Yupeng Hou,Zeyu Zhang,Shuqing Bian,Yushuo Chen,Zhen Tian,Xinyan Fan,Ji-Rong Wen,Xingyu Pan,Zihan Lin,Jiakai Tang,Lanling Xu,Changxin Tian,Xu Chen
DOI: https://doi.org/10.1145/3511808.3557680
2022-01-01
Abstract:In order to support the study of recent advances in recommender systems, this paper presents an extended recommendation library consisting of eight packages for up-to-date topics and architectures. First of all, from a data perspective, we consider three important topics related to data issues (ie sparsity, bias and distribution shift ), and develop five packages accordingly, including meta-learning, data augmentation, debiasing, fairness and cross-domain recommendation. Furthermore, from a model perspective, we develop two benchmarking packages for Transformer-based and graph neural network~(GNN)-based models, respectively. All the packages (consisting of 65 new models) are developed based on a popular recommendation framework RecBole, ensuring that both the implementation and interface are unified. For each package, we provide complete implementations from data loading, experimental setup, evaluation and algorithm implementation. This library provides a valuable resource to facilitate the up-to-date research in recommender systems. The project is released at the link: \urlhttps://github.com/RUCAIBox/RecBole2.0.