IEEE Access (Jan 2024)

Design and Implementation of Attention-Based CR System in the Context of Big Data

  • Zheyi Wang,
  • Yanli Kuang,
  • Xin Lyu

DOI
https://doi.org/10.1109/ACCESS.2024.3378521
Journal volume & issue
Vol. 12
pp. 58639 – 58650

Abstract

Read online

The recommendation accuracy of traditional product recommendation systems is insufficient. Therefore, an improved deep factor decomposition machine algorithm combining adaptive regularization and attention mechanisms is proposed, and big data components are integrated to enable the algorithm to support more data input types. The publicly available datasets Criteo and Avazu from the Kaggle competition were selected for testing experiments in the study. The experimental results are as follows. During the training phase, after the convergence of each model, the loss function value of the model designed in this study is 1.26, which is the lowest among all comparison models. Moreover, when the number of recommended products is 7, the overall recommendation effect of each model is the best. The area under the curve of the subject operation characteristic curve of the model designed in this study on the Criteo dataset is 0.809, which is significantly higher than all comparison models. It is believed that this model has higher recommendation accuracy and can be used in application scenarios that require higher recommendation quality.

Keywords