In-depth Recommendation Model Based on Self-Attention Factorization

Abstract

Rating prediction is an important issue in recommender systems, and its accuracy affects the experience of the user and the revenue of the company. Traditional recommender systems use Factorization Machines for rating predictions and each feature is selected with the same weight. Thus, there are problems with inaccurate ratings and limited data representation. This study proposes a deep recommendation model based on self-Attention Factorization (SAFMR) to solve these problems. Thismodel uses Convolutional Neural Networks to extract features from user and item reviews. The obtained features are fed into self-Attention mechanism Factorization Machines, where the self-Attention network automatically learns the dependencies of the features and distinguishes the weights of the different features, thereby reducing the prediction error. The model was experimentally evaluated using six classes of dataset. We compared MSE, NDCG and time for several real datasets. The experiment demonstrated that the SAFMR model achieved excellent rating prediction results and recommendation correlations, thereby verifying the effectiveness of the model.