Field-Aware Matrix Factorization for Recommender Systems

Abstract

Predicting user response is one of the core machine learning tasks in recommender systems (RS). The matrix factorization (MF)-based model has been proved to be a useful tool to improve the performance of recommendation. Many existing matrix factorization-based models mainly rely on adding some side information into basic MF to enable the model to fully express the data. However, most of the side information is measured based on the statistics or empirical formula. Also, the latent features of side information cannot be deeply mined. In this paper, we focus on mining the influence of field information (useful side information) to improve the performance of prediction. Based on the MF framework, we propose a field-aware matrix factorization (FMF) model. In FMF, the interactions between user/item and field can be captured and learned in the latent vector spaces. We propose efficient implementations to train FMF. Then, we comprehensively analyze FMF and compare this model with the state-of-the-art models. The analysis of experiments on two large data sets demonstrates that our method is very useful in RS.