应用科学学报 ›› 2022, Vol. 40 ›› Issue (1): 145-154.doi: 10.3969/j.issn.0255-8297.2022.01.013

• 计算机应用专辑 • 上一篇    下一篇

KNN-GWD推荐模型及其应用

季德强, 王海荣, 车淼, 王嘉鑫   

  1. 北方民族大学 计算机科学与工程学院, 宁夏 银川 750021
  • 收稿日期:2021-07-17 出版日期:2022-01-28 发布日期:2022-01-28
  • 通信作者: 王海荣,博士,研究方向为大数据与知识工程。E-mail:bmdwhr@163.com E-mail:bmdwhr@163.com
  • 基金资助:
    宁夏自然科学基金(No.2020AAC03218,No.2020AAC03307);省部级前期培育项目基金(No.PY1906);大学生创新项目基金(No.2021-XJ-JSJ-021);北方民族大学校级一般项目基金(No.2021XYZJK06)资助

KNN-GWD Recommendation Model and Its Application

JI Deqiang, WANG Hairong, CHE Miao, WANG Jiaxin   

  1. School of Computer Science and Engineering, North Minzu University, Yinchuan 750021, Ningxia, China
  • Received:2021-07-17 Online:2022-01-28 Published:2022-01-28

摘要: 为解决传统推荐存在精准性差的问题,构建了一个融合K-最近邻算法(K-nearest neighbor,KNN)、图神经网络(graph neural network,GNN)和深度宽度(Wide&Deep)网络的推荐模型。融入KNN分类方法对数据进行噪声过滤,以提高数据质量;利用GNN提取用户会话图的节点嵌入表示,基于注意力机制加权用户全局特征以捕获用户短期兴趣;调用Wide&Deep以解决稀疏数据时的模型过度泛化问题。为了验证模型的有效性,分别在MovieLens-1M、Bing-News、Book-Crossing数据集与6种传统推荐方法进行对比实验,结果表明所提模型的各项评价指标更佳。为了进一步验证该模型在实际应用领域中的可行性,面向农业领域搭建了农业一体管理App化肥推荐系统,得到推荐结果的准确率为0.721,曲线下面积为0.784,满足了预期的应用需求。

关键词: 推荐系统, 特征表示, 用户偏好, 农业领域

Abstract: In order to solve the problem of poor accuracy in traditional recommendation, a multi-layer K-nearest neighbor (KNN) network recommend model KNN-GWD, in combination of graph neural network (GNN) and wide & deep network was constructed. In the model, the KNN classification method is for data noise filtering to improve data quality. GNN is used to extract the node embedding representation of user's conversation graphs, and capture user's short-term interest by weighting user's global characteristics based on attention mechanism. Wide&Deep is used to solve the problem of model overgeneralization in the case of sparse data. In order to verify the effectiveness of the model, comparative experiments were carried out on MovieLens-1M, Bing-News and Book-Crossing data sets with this model and six other traditional recommendation methods. Experimental results show that the evaluation indicators of this model are better. In order to further verify the feasibility of the proposed model in the actual application field, an agricultural integrated management App fertilizer recommendation system was built with the accuracy of recommended results of 0.721 and the area under curve of 0.784, which met the expected application requirements.

Key words: recommendation system, feature representation, user preference, agricultural field

中图分类号: