计算机工程与科学 ›› 2024, Vol. 46 ›› Issue (09): 1648-1659.
周祺,周宁宁
ZHOU Qi,ZHOU Ning-ning
摘要: 基于神经网络增强的因子分解机模型因可以捕捉更多高阶特征的交互,使预测结果愈加精准而成为了当前推荐算法的研究热点。针对现有模型在对用户与物品交互特征进行建模时,并没有综合考虑高阶交互特征和原始低阶特征的问题,同时为了提高模型对用户偏好的建模能力,采用深度神经网络,并且结合成对学习提出了新的深度神经网络增强的成对双线性因子分解机模型DeepPRBFM。该模型采用一对分别包含正样本和负样本输入的双线性结构,利用多层ResNet保留低阶特征,利用DNN增强高阶特征的交互,并采用了基于Pairwise Ranking的损失函数。此外,双线性结构中,通过增加负样本的比例,不仅能大大减缓推荐系统的冷启动问题,而且还能提升模型的预测效果。在2个真实数据集上的实验结果表明,所提出的模型获得了更高的推荐准确率,在HR和NDCG等客观指标上都优于其他对比模型。