Attribute mapping and autoencoder neural network based matrix factorization initialization for recommendation systems

摘要

The recommendation algorithm is attracting increasing attention in analyzing big data. Matrix factorization (MF) is one of the recommendation methods and Singular Value Decomposition (SVD) is the most popular matrix factorization method. However, the existing SVD methods usually initialize user and item feature randomly, not fully utilize the information of the data, so require plenty of experiments to determine feature matrix dimension, with low convergence efficiency and low accuracy. This paper presents a hybrid initialization method based on attribute mapping and autoencoder neural network to solve these problems, which consists of three parts: (1) use the number of item attribute types to determine feature matrix dimension in order to avoid multiple experiments to select the optimal dimension value; (2) use items' attributes to initialize the item feature matrix in SVD++, and use an attribute mapping mechanism to get an item feature vector by fitting the rating matrix to accelerate the convergence; (3) adopt the autoencoder neural network to reduce feature dimension and obtain item latent features for initializing SVD++. The experimental results show that our methods achieve better performance than SVD++ random initialization and also be adopted to other matrix factorization methods. (C) 2018 Elsevier B.V. All rights reserved.