DeepRec
Implementation of Deep Learning based Recommender Algorithms with Tensorflow.
view repo
Deep learning based recommender systems have been extensively explored in recent years. However, the large number of models proposed each year poses a big challenge for both researchers and practitioners in reproducing the results for further comparisons. Although a portion of papers provides source code, they adopted different programming languages or different deep learning packages, which also raises the bar in grasping the ideas. To alleviate this problem, we released the open source project: DeepRec. In this toolkit, we have implemented a number of deep learning based recommendation algorithms using Python and the widely used deep learning package - Tensorflow. Three major recommendation scenarios: rating prediction, top-N recommendation (item ranking) and sequential recommendation, were considered. Meanwhile, DeepRec maintains good modularity and extensibility to easily incorporate new models into the framework. It is distributed under the terms of the GNU General Public License. The source code is available at github: <https://github.com/cheungdaven/DeepRec>
READ FULL TEXT VIEW PDFImplementation of Deep Learning based Recommender Algorithms with Tensorflow.
In recent years, deep learning has achieved tremendous success in many application domains, such as computer vision, natural language processing, and speech recognition. Inspired by the effectiveness of deep neural networks, researchers and practitioners started to apply deep learning techniques to recommendation tasks and have achieved a huge improvement on many recommendation tasks. Meanwhile, the number of publications on deep neural network based recommendation models also increases exponentially
[Zhang et al.2019]. Moreover, many companies resort to deep learning techniques to improve their recommendation quality for better user experience and increasing revenue [Covington et al.2016, Cheng et al.2016]. In general, deep learning techniques can be utilized to learn representations for user and item side information (e.g., user profile, item content, etc.), introduce nonlinearities to recommendation models and model the sequential patterns of user consumption history. In addition, deep neural networks are of high flexibility, which makes it possible to develop more powerful models by combining different neural networks.However, this hype also poses a big challenge to researchers and practitioners, that it, reproducing results for those proposed deep learning based models. Although some authors provide source code for reproductivity, many of them are written in different programming languages, or use different deep learning packages, let alone a large portion of work do not provide source code, which makes it hard to understand and hinder the process of justifying newly proposed methods. To this end, we start the project - DeepRec. Our goal is to make reproducing results easier and lower the bar of developing deep learning based recommendation models and increase their practicality.
Although there are some open source recommendation packages available, such as MyMediaLite222http://mymedialite.net/index.html, LibRec [Guo et al.2015], Suprise333http://surpriselib.com/ and OpenRec [Yang et al.2018], most of them only include traditional recommendation algorithms or have not updated for a long time. The OpenRec library do provide several deep neural network based recommendation models, the number is still very small (only four models implemented) compared with DeepRec.
To this end, we develop an open-source toolkit named ‘DeepRec’. This toolkit provides a flexible framework and unified interfaces for deep learning based recommendation models. The prime features of DeepRec are twofold:
In this toolkit, we addressed three common scenarios in recommendation research: rating prediction, top-n recommendation (item ranking) and sequence-aware recommendation. It offers state-of-the-art algorithms for those three tasks.
DeepRec is extensible, adaptable and efficient. Users could add algorithm easily. Moreover, DeepRec is able to make use of the computation power of GPUs, making it scalable to large-scale datasets.
In this section, we briefly introduce the archecture of the DeepRec, plus implemented recommendation algorithms and comparison with other frameworks.
DeepRec is made up of three components: recommender, backend and utility. We adopt the widely used deep learning framework Tensorflow and several other frequent used scientific calculation libraries such as Numpy, Scipy, Sklearn and Pandas as the core backend; A number of representative and advanced algorithms deep learning based recommendation models forms the recommender components. We split the algorithms into three categories: rating prediction, top-n recommendation and sequence-aware recommendation. We put the common evaluators and data preprocess functions into the utility component.
A number of algorithms that were implemented in the current version are listed below. Note that a few non deep learning methods (e.g., biasedSVD, Factorization Machines, BPRMF) are also implemented with Tensorflow.
In this category, we implemented methods like classical method like biasedSVD (or MF) [Koren et al.2009], Factorization Machines (FM) [Rendle2010], user and item based AutoRec [Sedhain et al.2015], neural network matrix factorization (NNMF) [Dziugaite and Roy2015], neural rating regression (NRR) [Li et al.2017], neural factorization machines (NFM) [He and Chua2017] and attentive factorization machines (AFM) [Xiao et al.2017]. For factorization machines, we use the same feature format as libfm [Rendle2012] to make them compatible.
Here, we included classical pairwise personalized ranking methods BPRMF [Rendle et al.2009], collaborative metric learning (CML) [Hsieh et al.2017], collaborative denoising auto-Encoder (CDAE) [Wu et al.2016]
, multilayer-perceptron (MLP), generalized matrix factorization (GMF) and neural collaborative filtering (NeuMF)
[he2017neural], NeuRec [Zhang et al.2018b] and latent relational metric learning [Tay et al.2018].For sequence-aware recommendation models, we implemented the personalized ranking metric embedding model (PRME) [Feng et al.2015], LSTM and GRU based methods [Hidasi et al.2015], convolution based model (Caser) [Tang and Wang2018], and self-attention based approach (AttRec) [Zhang et al.2018a].
Several commonly used evaluation metrics were included in this toolkit. For rating prediction, we can use Root Mean Square Error (RMSE) and Mean Average Error (MAE) to measure the effectiveness. For top-n recommendation and sequence-aware recommendation, we included several ranking-aware metrics, including, Recall@n, Precision@n, Normalized Discounted Cumulative Gain (NDCG) and Mean Reciprocal Rank (MRR) to measure both the accuracy and quality of ranking lists. For detailed definitions, readers are referred to
[Shani and Gunawardana2011].Library | backend | GPUs | 3 scenarios | #DL |
Suprise | SciKits | N | N | 0 |
MyMediaLite | N/A | N | N | 0 |
LibRec | N/A | N | N | 0 |
OpenRec | Tensorflow | Y | N | 4 |
DeepRec | Tensorflow | Y | Y | > 20 |
Table 1 summarizes the difference between our toolkit and other existing Toolkits. Suprise, MyMedialite, LibRec do not include neural network based models and can not utilize the computation power of GPUs. OpenRec also adopts Tensorflow as the backend, but it only implemented four ranking based recommendation methods, while DeepRec includes over twenty state-of-the-art deep learning based recommendation model. In addition, DeepRec is the first library that includes sequence-aware models.
In this demo paper, we introduce the open-source library DeepRec, in the hope that it could ease the burden of researchers and practitioners on reproducing existing models. DeepRec is a flexible, extensible and efficient framework which includes a number of state-of-the-art deep learning based recommendation algorithms.
Moving forward, future work will include: implementing more models and evaluation measures, improving the documentation and tutorials as well as updating it to follow the scheme of Tensorflow 2.0.
Twenty-Fourth International Joint Conference on Artificial Intelligence
, 2015.Autorec: Autoencoders meet collaborative filtering.
In WWW, 2015.
Comments
There are no comments yet.