Abstract

In a real e-commerce website, usually only a small number of users will give ratings to the items they purchased, and this can lead to the very sparse user-item rating data. The data sparsity issue will greatly limit the recommendation performance of most recommendation algorithms. However, a user may register accounts in many e-commerce websites. If such users’ historical purchasing data on these websites can be integrated, the recommendation performance could be improved. But it is difficult to align the users and items between these websites, and thus how to effectively borrow the users’ rating data of one website (source domain) to help improve the recommendation performance of another website (target domain) is very challenging. To this end, this paper extended the traditional one-dimensional psychometrics model to multidimension. The extended model can effectively capture users’ multiple interests. Based on this multidimensional psychometrics model, we further propose a novel transfer learning algorithm. It can effectively transfer users’ rating preferences from the source domain to the target domain. Experimental results show that the proposed method can significantly improve the recommendation performance.

1. Introduction

A recommendation system aims to find the favorite items (goods) for users. Over the past decade, many recommendation algorithms have been proposed, among which collaborative filtering (CF) has attracted much more attention because of its high recommendation accuracy and wide applicability. These CF algorithms can be grouped into two categories: memory-based CF and model-based CF [1, 2]. Memory-based CF mainly includes user-based CF and item-based CF. User-based CF supposes that the test user will like the items that his or her similar users like, and the estimated rating of the test user for the test item is adjusted by the ratings of his or her similar users. Item-based CF deems that the test user will like the other items similar to the items that he or she previously liked, and the estimated rating of the test user for the test item is calculated by the items this user previously rated. Therefore, the key step in memory-based CF is to calculate the similarity of users or items. Usually, the similarity is calculated directly from the user-item rating matrix. In this matrix, each row denotes a feature vector of a user, while each column denotes a feature vector of an item. The similarity between two vectors depends on the elements at which both vectors have a rating. But usually, only a few users evaluated a few items on websites; thus, the user-item rating matrix is very sparse, and the number of commonly rated elements of the two vectors is very small. This results in an unreliable computed similarity and hence a low recommendation accuracy. Furthermore, with the growth of users and items, the complexity of similarity computation increases in a nonlinear fashion, which restricts its scalability.

To solve these problems, a variety of machine learning and data mining models have been proposed, which have led to the development of model-based CF algorithms. The basic idea of model-based CF algorithms is to reduce the dimension of a rating matrix that will combat the scalability and data sparsity problems. Model-based CF mainly contains clustering methods [3, 4], regression methods [5], graph model-based methods, matrix factorization methods, and neural network methods. Graph model-based methods include Bayesian network [6, 7], PLSA (Probabilistic Latent Semantic Analysis) [8, 9], and LDA (Latent Dirichlet Allocation) [10]. Representative Matrix factorization methods include SVD (Singular Value Decomposition) [1115], PCA (Principal Component Analysis) [16], and NMF (Nonnegative Matrix Factorization) [17, 18]. Neural network methods include RBM (Restrained Boltzmann Machine) [19]. Generally, model-based CF has better performance than memory-based CF; however, the recommendation produced by memory-based CF has better interpretability and makes convincing users easier. However, these methods only utilize the rating data coming from a single website and ignore exploiting the ratings from other websites.

In order to exploit big data gathered from different websites to cope with data sparsity problems, recently, the transfer learning methods [2229] were proposed. They utilize the data from the source domain to improve the prediction accuracy of the target domain. However, transfer learning can potentially hinder performance if there is a significant difference between the source domain and the target domain [30]. For successful knowledge transfer, there are two critical problems that must be resolved. First, the two domains share the same feature space (in recommendation system, it means that the two domains have aligned users or items); however, their distributions are significantly different. Second, the two domains do not share the same feature space (in recommendation system, it means that the two domains have no aligned users or items). To solve the first problem, some methods [2426, 31] were proposed to supervise the knowledge transfer. The second problem is very difficult, because a common latent space is needed in which the two domains are similar, and the common factors learned from the source domain can be used to constrain the algorithm to achieve better performance in the target domain. A representative method to solve this tough problem is CBT (CodeBook Transfer), proposed in [23]. The algorithm CBT exploits nonnegative matrix trifactorization to transfer information from the source domain to the target domain, which can be expressed by where denotes the rating matrix in the source domain, denotes the rating matrix in the target domain, and is a tradeoff parameter to balance the target and source domain data. , , and are nonnegative matrices. matrix is learned from the source domain according to (1), and is then fixed to learn the factor matrices of the target domain by (2). This method expects to find a common latent space (here, it is represented by the middle matrix) in which the information obtained from the source domain data (i.e., matrix ) can be used to improve the algorithm’s performance in the target domain. However, there are several implementation methods of nonnegative matrix trifactorization. Some constraints are usually needed in these methods to guarantee that the columns of the left matrix and right matrix are orthogonal [17, 24, 25, 32]. Note that too strict or too loose constraints would damage the performance of the nonnegative matrix trifactorization based transfer learning method.

Compared with our previous work [33], the differences between them and the new contributions of this paper are as follows. In our previous work, a multidimensional psychometrics-based collaborative filtering algorithm was proposed, which is presented in Section 2.1. In this paper, we extend the algorithm and propose a novel transfer learning algorithm in Section 2.2. This new proposed method can effectively transfer knowledge from source domain to target domain even when the two domains have no aligned users or items. In Section 3.2, MovieLens dataset and BookCrossing dataset are used to evaluate the two algorithms to validate the effectiveness of knowledge transfer. Comprehensive experiments are conducted in Section 3.3, and the results show that our novel transfer learning algorithm outperforms representative state-of-the-art algorithms including nonnegative matrix trifactorization based transfer learning method [23].

2. Materials and Methods

In this section, a multidimensional psychometrics model is proposed. Compared with one-dimensional psychometrics model, it can model user’s multilatent interests. Furthermore, a novel transfer learning method based on this multidimensional psychometrics model is proposed, which can mine rating preferences parameters and transfer them from source domain to target domain to improve the recommendation accuracy further.

2.1. Collaborative Filtering Using a Multidimensional Psychometrics Model

Hu et al. [20, 21] first introduced the psychometrics model into CF and used the software Winsteps [34] to estimate the parameters in the model. It can be expressed aswhere and are scalar quantities, which denote the interest of user and the drawback of item . Higher means that user has a great interest in the items, and he or she will assign a higher score to the items. Smaller indicates that item is of good quality and will receive higher scores from users. The parameter denotes the probability of user assigning score to item , and is the ordered threshold, which denotes the difficulty for users to give score relative to giving score for an item. Each item has -level scores, that is, , and parameters , , and need to be estimated.

In (3), and are scalar quantities and denote the user’s interest and the item’s quality, respectively. The model expressed by (3) assumes that user’s interest and item’s quality are one-dimensional. However, it is more feasible to represent the user’s interest and the item’s quality with vectors because users usually have varied interests and gain satisfaction from item in a number of different aspects [33]. For example, a user can have an interest in science fiction films with a value of 2, an interest in action movies with a value of 8, an interest in romantic films with a value of 1, an interest in movie scenes with a value of 3, and an interest in actors with a value of 5. For a film that is 30% science fiction and 20% action, the scene quality could earn a value of 2 while the actors’ performance earns a value of 3. A user’s rating for a movie is determined from a comprehensive assessment of the satisfaction that the movie brings to him or her in the various dimensions. Based on this idea, the psychometrics model is extended to multidimension, which is expressed aswhere is the probability of user assigning rating score to item in dimension and is the interest value of user in dimension . denotes the interest vector of user . A high value of indicates that user has a strong interest in dimension , which implies that if a film can satisfy a user’s interest in this area, then he or she will assign a high latent rating value to this film in this dimension. is the quality value of item in dimension ; therefore, can denote the quality vector of item . Higher indicates that item has less content related to dimension and is less probable of earning a high rating in dimension , and is the ordered threshold that measures the difficulty for all users to assign points relative to assigning points to an item. Each item has a K-level rating. From the MovieLens (http://www.grouplens.org/) dataset, is set to 5. is the number of dimensions, which denotes the number of latent interests or demands each user has, and it is a metaparameter, which should be set in advance. If is set to 1, this model will degenerate into the model represented by (3). Parameters ,  , and should be estimated using the training set. To simplify, the notation is introduced, and (6)–(8) can be deduced from (5). Hence,Equation (8) shows that the rating of user for item in dimension is of a multinomial distribution with categories determined by , , and . After    is obtained, there are two methods to estimate the latent rating of user for item in dimension (denoted by ), and they are as follows:(1)The rating score corresponding to the biggest probability:(2)The expectation given by

In this paper, for simplicity, the second method is used to estimate the latent rating of user for item in dimension because the function of described by (10) is continuous while the function of (9) is not continuous. After    is obtained, there are three methods to estimate the rating value of user for item , and they are formulated by (11)–(13). In this paper, for simplicity, (13) is used to estimate the rating value of user for item . The three methods are as follows:(1)The maximum value of    as follows:(2)The minimum value of      as follows:(3)The weighted summation of    as follows:where and denotes the weight of user in dimension . It also represents the importance of the th interest or demand for user ; therefore, the weight vector of user can be determined by , and it will be estimated using the training set.

In order to learn the parameters, the loss function is designed aswhere denotes the training set. The first term on the right-hand side of the equation is the sum of the square errors, and the other four terms are the regularization terms that are applied to the parameters to avoid overfitting. The regularization constant is a parameter that requires tuning.

It is easy to find that the partial derivative of the loss function is an elementary function; therefore, the partial derivative of the loss function is continuous, and the loss function is differentiable. Thus, the stochastic gradient descent method [14] can be used to learn the parameters. Convergence of the stochastic gradient descent method has been empirically verified in many previous works [14, 15, 35], given the appropriate learning ratio parameters. Each rating in the training set is used to update the parameters according to (17)–(20). Parameter should not be updated and is fixed as 0 according to (7). Learning ratio parameters , , and are parameters that should be tuned. For easy representation, notations and are introduced by (15) and (16). Hence, we have the following:

The pseudo codes of the collaborative filtering using multidimensional psychometrics model (CFMPM) are shown in Algorithm 1. After the parameters have been estimated by the algorithm, the ratings in the test set can be predicted by (13).

Input: Training set
Output: , , , and ()
()Initialize parameters , , , , , , , and .
()For each iteration
()Randomly permutate the data in .
()For each in
()    Compute by (15);
()    Compute (; ) by (16);
()    Update the parameters according to (17)–(20).
()End for
()Decrease , , ; they are multiplied by 0.9.
() End for
2.2. Transfer Learning Based on Multidimensional Psychometrics Model

In this section, we will represent the notion that, in the multidimensional psychometrics model, the rating preferences parameters of all the datasets are similar, and these parameters may be the common knowledge that could be shared from source domain to target domain. Based on this idea, a novel transfer learning method is proposed.

Equations (21)–(25) can be derived from (8), (10), and (13), where is defined by (6) and (7), and the notation “” means that the two sides of the equation have the same sign. denotes the probability for user to give score to item , and denotes the probability of getting score for an item. The notation “” means that the two sides of the equation have the same increase or decrease trend, and the notation denotes the number of ratings in the training set. If the entry is observed in the training set, , or not, ,

From (25), we can see that the sign of the partial derivative of with respect to is constant and positive, which means that when increases, users have a higher probability (i.e., ) of assigning score to item and vice versa. It also means that a higher value of will result in a higher value of , and a lower value of will result in a lower value of . In reality, users rarely assign a bad evaluation. Users usually give a score of 4 or 5 (in a 1–5-scaled rating system) to an item because if a user does not like the item, the user will not buy it at all. The proportion of each rating score in three different datasets is shown in Figure 1, which shows that the rating score distributions in the three datasets are similar. It means that the users in the different domains have similar rating preferences; that is, the threshold is usually similar in different domains. Based on this idea, if we learn parameter from the source domain and transfer it to the target domain, the prediction accuracy in the target domain will possibly be improved. Because of the similarity of parameter across different domains, the parameter learned from the source domain contains the information about the target domain. Meanwhile, transferring into the target domain will increase the model’s generalization capability according to Occam’s Razor, as it reduces the number of parameters to be learned. Therefore, a new transfer learning method is designed as shown in Algorithm 2, which is called TLMPM (transfer learning based on multidimensional psychometrics model).

Input: The data of the source domain denoted by and the training data set of the target domain denoted by .
Output: Parameters , , , and () of the target domain.
()Learn the parameters of the source domain by algorithm CFMPM using , and get parameters () of the
 source domain.
()Set parameters () of the target domain equal to the parameters () of the source domain,
 respectively.
()Initialize parameters , , , , and of the target domain.
()For each iteration
()Randomly permutate the data in .
()For each in .
()    Compute by (15);
()    Compute (; ) by (16);
()    Update the parameters according to (17)–(19).
() End for
() Decrease , ; they are multiplied by 0.9.
() End for

3. Results

Two experiments are designed to evaluate the performance of the algorithm TLMPM. The first one is to estimate the sensitivity of parameter and to compare algorithms TLMPM and CFMPM with the collaborative filtering algorithms proposed in [20, 21], which are based on one-dimensional psychometrics. The other experiment is used to compare TLMPM with other representative state-of-the-art algorithms including nonnegative matrix trifactorization based transfer learning method CBT, which is reported in [23].

3.1. Datasets and Evaluation Metrics

We use the following datasets in our empirical tests.

EachMovie (http://www.cs.cmu.edu/~lebanon/IR-lab.htm) (source domain) is a movie rating dataset that contains 2.8 million ratings (scales 0–5) rated by 72,916 users for 1,628 movies. Because user’s implicit feedback is denoted by score 0 in the EachMovie dataset and this implicit feedback has many noises, we removed the score 0. Similar to Li et al. [23], we extract a submatrix to simulate the source domain by selecting 500 users and 500 movies with the most ratings (rating ratio 54.67%).

MovieLens (http://www.grouplens.org/) (target domain) is a movie rating dataset that contains 100,000 rating scores (1–5 scales) rated by 943 users for 1682 items, where each user has more than 20 ratings.

BookCrossing (http://www.informatik.uni-freiburg.de/~cziegler/BX/) (target domain) contains 77,805 users, 433,671 ratings (scales 1–10), and approximately 185,973 books. We normalize the rating scales from 1 to 5.

We use Mean Absolute Error (MAE) as the evaluation metric. It is used to average the absolute deviation between the predicted values and the true values:where is the number of tested ratings in test set . The lower the MAE is, the better the performance is.

3.2. Performance Comparison between Psychometrics-Based CF Algorithms

In this section, we will evaluate the proposed algorithms using the MovieLens dataset and the BookCrossing dataset. Here, we named the one-dimensional psychometrics model-based algorithms proposed in [20, 21] CFOPM_1 and CFOPM_2, respectively. To estimate the necessity of the parameter in our algorithm CFMPM, a method called CFMPM_L is designed which differs from CFMPM only in that its parameter is set to 0, and it will always remain unchanged. Fivefold cross-validation was performed for each dataset, and, for every test set, the algorithms were run independently 5 times, and the average MAE over 25 times is reported. In this experiment, the parameters in TLMPM, CFMPM_L, and CFMPM are set as follows: the initial values of and are sampled from the uniform distribution on ; the initial value of is set to 1; and the initial value of    is set to 0; the values in are used to tune the parameter ; and the values in are used to tune the parameters , , and ; the values in are used to tune , the dimension of latent interests. To evaluate algorithm TLMPM, firstly, algorithm CFMPM is used to learn parameters    from EachMovie dataset as described in the previous section, and then the learned parameters    are copied to the target domain and held constant to learn the other parameters. The results are shown in Table 1. In Table 1, the results of CFOPM_1 and CFOPM_2 for the MovieLens dataset are extracted directly from Hu’s papers [20, 21]. The MAE values for these two algorithms on BookCrossing dataset are our experimental results, and our experimental settings are according to the papers of Hu et al. [20, 21].

From Table 1, it is shown that, of the two datasets, algorithm CFMPM outperforms CFOPM_1. Although the MAE of CFMPM is not significantly lower than that of CFOPM_2 on MovieLens, CFMPM has a significantly lower MAE than CFOPM_2 has on BookCrossing dataset. The time complexity of CFMPM is , while the time complexity of CFOPM_2 is . Here, is the iterate times, is the number of ratings in training set, is the dimensionality of user’s interest, is the rating levels, is the number of users, and is the number of items. In our experiments, is smaller than 25 and is smaller than 30. Therefore, on the MovieLens dataset, the time complexity of CFMPM is and it is for CFOPM_2. On the BookCrossing dataset, the time complexities for CFMPM and CFOPM_2 are and , respectively. Thus, if the rating matrix is extremely sparse, for example, BookCrossing, CFMPM will have a lower computational time than CFOPM_2. This means that extending the psychometrics model from one dimension to multidimension is effective. Meanwhile, algorithm CFMPM outperforms CFMPM_L, which verifies that parameters    are essential in this model, although the number of these parameters is only 5. Furthermore, the algorithm TLMPM outperforms CFMPM, which means that the transfer is positive even though only four parameters (i.e., , , , and ) were transferred from the source domain to the target domain.

3.3. Performance Comparison between TLMPM and Other State-of-the-Art Algorithms

The compared algorithms are Codebook Transfer (CBT) [23] and the other baseline algorithms used in [23], which are Scalable Cluster-Based Smoothing (CBS) [36] and user-based CF based on Pearson correlation coefficients (PCC) similarity [37]. Algorithm CBT [23] is a representative transfer learning method based on nonnegative matrix trifactorization. Except for TLMPM and CBT, the other algorithms only use the data in the target domain to predict the ratings in the test set.

To consistently compare algorithm TLMPM with the other algorithms described above, as in [23], we also extract a subset of the MovieLens dataset, which contains 1000 items and 500 users, where each user has more than 40 ratings (rating ratio 10.35%). For each run of algorithm TLMPM, the training set and the test set are generated as follows. First, 500 users were randomly sorted. Second, the first 100, 200, or 300 users were selected as training sets and denoted as ML_100, ML_200, and ML_300, respectively, and the test set contains the last 200 users. Additionally, in order to evaluate the performance of the algorithm at different data sparsity levels, as [23] did, the available ratings of each test user are also split into an observed set and a held-out set. The observed ratings are used to predict the held-out ratings. We randomly selected 5, 10, and 15 items rated by each test user to constitute the observed set, and they are named Given5, Given10, and Given15, respectively, while the other items rated by the test user constitute the held-out set. The BookCrossing dataset is also preprocessed using the same method described above and then produces three training datasets denoted by BX_100, BX_200, and BX_300, respectively.

In this experiment, the parameters in algorithm TLMPM are initialized in the same way as described in the previous section. The comparison results are shown in Figures 27, and the reported results of algorithm TLMPM are the average MAE over 10 independent run times.

From Figures 27, we can see that the algorithm TLMPM almost outperforms all the other algorithms at each sparsity level on all datasets, which verifies that transferring rating preferences from the source domain to the target domain is effective.

4. Discussion

This paper extended the psychometric model from one dimension to multidimension to model user’s multi-interests and then proposed algorithm CFMPM, which outperforms the algorithm based on the one-dimensional psychometric model. After that, a novel transfer learning algorithm based on CFMPM was developed. In this transfer algorithm, the psychometric model is used to learn user’s rating preferences from the source domain and then transfer them to the target domain. This transfer algorithm does not require two domains having overlapped users or items. Experimental results show that it is a positive transfer and that it outperforms other representative state-of-the-art algorithms such as CBT, the transfer algorithm based on nonnegative matrix trifactorization. But, in a 5-scaled rating system, the number of transferred parameters is only 4, which limits the algorithm’s effectiveness. In future work, we intend to study how to break this limitation and how to speed up the algorithm.

Competing Interests

The authors declare that there are no competing interests regarding the publication of this paper.

Acknowledgments

The authors appreciate Yan Chen, Senzhang Wang, Yang Yang, and Xiaoming Zhang for their constructive suggestions for polishing this paper. This work is supported by the National Natural Science Foundation of China (nos. 61370126, 71540028, and 61571052), the Fund of Beijing Social Science (no. 14JGC103), the Statistics Research Project of National Bureau (no. 2013LY055), the Fund of Beijing Key Laboratory (no. BZ0211), and the High-Level Research Projects’ Cultivation Fund of Beijing Wuzi University (no. GJB20141002).