结合Skip-gram和加权损失函数的神经网络推荐模型  

Neural Network Recommendation Model Combined with Skip-gram Model and Weighted Loss Function

作　　者：李淑芝[1] 余乐陶 邓小鸿[2] 李志军 LI Shuzhi;YU Letao;DENG Xiaohong;LI Zhijun(School of Information Engineering,Jiangxi University of Science and Technology,Ganzhou,Jiangxi 341000,China;College of Applied Science,Jiangxi University of Science and Technology,Ganzhou,Jiangxi 341000,China)

机构地区：[1]江西理工大学信息工程学院,江西赣州341000 [2]江西理工大学应用科学学院,江西赣州341000

出　　处：《计算机工程与应用》2020年第19期76-85,共10页Computer Engineering and Applications

基　　金：国家自然科学基金(No.61762046);江西省教育厅科学技术研究项目(No.GJJ181505)。

摘　　要：针对网络推荐系统中传统的协同过滤技术在实际应用中存在数据稀疏、导致准确率低、推荐单一性等问题,提出一种结合Skip-gram项目嵌入和加权损失函数的深度神经网络的推荐模型DSM。采用了3层ReLU层对输出向量进行回归,在未使用附加信息的前提下提高了推荐精度;利用Skip-gram进行项目嵌入得到更稠密的表示向量,减少了计算量;并且使用加权损失函数训练深度神经网络的参数,平衡了推荐项目的受欢迎程度,保证了新颖性。在APP数据集和Last.fm数据集的实验结果表明,DSM模型在推荐应用程序和歌曲时,准确性和多样性方面相比现有方法均有一定的提高。Aiming at the problems of sparse data, low accuracy and single recommendation in the application of traditional collaborative filtering technology in network recommendation system, this paper proposes a deep neural network recommendation model named DSM combining skip-gram item embedding model and weighted loss function. Firstly, the proposed model uses three layers of ReLU to regress the output vectors, which improves the recommendation accuracy without using additional information. Secondly, this paper uses the Skip-gram model for item embedding to obtain a denser representation vector, which reduces the computation. Moreover, the weighted loss function is used to train the parameters of the deep neural network, which balances the popularity of the recommended items and ensures the novelty. Finally, the experimental results in the APP dataset and the Last.fm dataset show that the proposed model has a certain improvement in accuracy and diversity compared to existing methods when recommending applications and songs.

关 键 词：推荐系统 数据稀疏 Skip-gram 加权损失函数 深度神经网络 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]