融合矩阵分解和XGBoost的个性化推荐算法
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP301.6

基金项目:

国家自然科学基金青年科学基金资助项目(11701466)。


Personalized recommendation system based on matrix factorization and XGBoost algorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对传统的协同过滤推荐算法中评分矩阵过于稀疏和算法准确度不高的问题,提出一种融合矩阵分解和XGBoost算法的推荐算法(MFXGB,Matrix Factorization XGBoost),其特点是利用SVD++算法(SVD,Singular Value Decomposition)对用户项目评分矩阵进行填充,避免过多的缺失值对算法精确度的影响,再利用XGBoost(eXtreme Gradient Boosting)算法训练有监督的模型用于预测用户评分。为了克服计算成本过高的困难,提出利用K-均值聚类方法进行特征提取用于训练XGBoost模型。将MFXGB算法应用于MovieLens数据集进行实验分析,结果显示,MFXGB算法的推荐精确度比传统的3种方法分别提高了8.91%、10.18%和11.79%,效果明显优于传统的推荐算法。

    Abstract:

    In order to solve the data sparsity problem and improve the recommendation accuracy, we proposed a new matrix factorization XGBoost(MFXGB) recommendation algorithm which combined the matrix factorization method and the XGBoost (Extreme Gradient Boosting) algorithm. SVD++ algorithm (SVD, singular value decomposition) was used to fill the user-item score matrix to lessen the influence on the accuracy of the algorithm due to too many missing values and XGBoost was used to build a supervised learning model to predict the user's score. To reduce the computation time, feature extration based on the K-means clustering method was proposed to train XGBoot. The proposed MFXGB algorithm was applied to MovieLens dataset for experimental analysis and the results show that the recommendation accuracy was improved by 8.91%,10.18% and 11.79% respectively, compared with the three traditional algorithms.

    参考文献
    相似文献
    引证文献
引用本文

何婧,胡杰.融合矩阵分解和XGBoost的个性化推荐算法[J].重庆大学学报,2021,44(1):78-87.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2020-07-06
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2021-01-08
  • 出版日期:
文章二维码