A new item-based deep network structure using a restricted Boltzmann machine for collaborative filtering

作者：Du, Yong-ping; Yao, Chang-qing^*; Huo, Shu-hua; Liu, Jing-xuan

来源：Frontiers of Information Technology and Electronic Engineering, 2017, 18(5): 658-666.

DOI：10.1631/FITEE.1601732

• Restricted Boltzmann machine
• Deep network structure
• Collaborative filtering
• Recommendation system

The collaborative filtering (CF) technique has been widely used recently in recommendation systems. It needs historical data to give predictions. However, the data sparsity problem still exists. We propose a new item-based restricted Boltzmann machine (RBM) approach for CF and use the deep multilayer RBM network structure, which alleviates the data sparsity problem and has excellent ability to extract features. Each item is treated as a single RBM, and different items share the same weights and biases. The parameters are learned layer by layer in the deep network. The batch gradient descent algorithm with minibatch is used to increase the convergence speed. The new feature vector discovered by the multilayer RBM network structure is very effective in predicting a rating and achieves a better result. Experimental results on the data set of MovieLens show that the item-based multilayer RBM approach achieves the best performance, with a mean absolute error of 0.6424 and a root-mean-square error of 0.7843.

• 出版日期2017-5
• 单位中国科学技术信息研究所; 北京工业大学