Matrix factorisation with word embeddings for rating prediction on location-based social networks

Abstract

With vast amounts of data being created on location-based social networks (LBSNs) such as Yelp and Foursquare, making effective personalised suggestions to users is an essential functionality. Matrix Factorisation (MF) is a collaborative filtering-based approach that is widely used to generate suggestions relevant to user’s preferences. In this paper, we address the problem of predicting the rating that users give to venues they visit. Previous works have proposed MF-based approaches that consider auxiliary information (e.g. social information and users’ comments on venues) to improve the accuracy of rating predictions. Such approaches leverage the users’ friends’ preferences, extracted from either ratings or comments, to regularise the complexity of MF-based models and to avoid over-fitting. However, social information may not be available, e.g. due to privacy concerns. To overcome this limitation, in this paper, we propose a novel MF-based approach that exploits word embeddings to effectively model users’ preferences and the characteristics of venues from the textual content of comments left by users, regardless of their relationship. Experiments conducted on a large dataset of LBSN ratings demonstrate the effectiveness of our proposed approach compared to various state-of-the-art rating prediction approaches.