Joint user attributes and item category in factor models for rating prediction

Abstract

One important problem of recommender system is rating prediction. In this paper, we use the movie rating data from Movie- Lens as an example to show how to use users’ attributes to improve the accuracy of rating prediction. Through data analysis, we observe that users having similar attributes tend to share more similar preferences and users with a special attribute have their own preferred items. Based on the two observations, we assume that a user’s rating to an item is determined by both the user intrinsic characteristics and the user common characteristics. Using the widely adopted latent factor model for rating prediction, in our proposed solution, we use two kinds of latent factors to model a user: one for the user intrinsic characteristics and the other for the user common characteristics. The latter encodes the influence of users’ attributes which include user age, gender and occupation. On the other hand, we jointly use user attributes or item category information and rating data for calculating similarity of users or items. The similarity calculating results are used in our proposed latent factor model as a regularization term to regularize users or items latent factors gap. Experimental results on MovieLens show that by incorporating users’ attributes influences, much lower prediction error is achieved than the state-of-the-art models. The prediction error is further reduced by incorporating influences from item category popularity and item popularity.