过度拟合
计算机科学
人工神经网络
推荐系统
水准点(测量)
人工智能
机器学习
机制(生物学)
深层神经网络
数据挖掘
大地测量学
认识论
哲学
地理
作者
Chang‐Dong Wang,Wu-Dong Xi,Ling Huang,Yin-Yu Zheng,Zi-Yuan Hu,Jianhuang Lai
标识
DOI:10.1109/tkde.2020.3023976
摘要
Recently, some attempts have been made in introducing deep neural networks (DNNs) to recommender systems for generating more accurate prediction due to the nonlinear representation learning capability of DNNs. However, they inevitably result in high computational and storage costs. Worse still, due to the relatively small number of ratings that can be fed into DNNs, they may easily suffer from the overfitting issue. To tackle these issues, we propose a novel recommendation framework based on Back Propagation (BP) neural network with attention mechanism, namely BPAM++. In particular, the BP neural network is utilized to learn the complex relationship between the target user and his/her neighbors and the complex relationship between the target item and its neighbors. Compared with DNNs, the shallow neural network, i.e., BP neural network, can not only reduce the computational and storage costs, but also alleviate the overfitting issues in DNNs caused by a relatively small number of ratings. In addition, an attention mechanism is designed to capture the global impact of the nearest users of the target user on their nearest target user sets. Extensive experiments conducted on eight benchmark datasets confirm the effectiveness of the proposed model.
科研通智能强力驱动
Strongly Powered by AbleSci AI