Paper
19 July 2024 S-BERT: A BERT-based personalised recommendation model
Jun Chen, Xuefeng Fu, Kailiang Wang, Yanping Liu, Weikun Chen
Author Affiliations +
Proceedings Volume 13181, Third International Conference on Electronic Information Engineering, Big Data, and Computer Technology (EIBDCT 2024); 131818C (2024) https://doi.org/10.1117/12.3031102
Event: Third International Conference on Electronic Information Engineering, Big Data, and Computer Technology (EIBDCT 2024), 2024, Beijing, China
Abstract
Sequential dynamics of users is an important feature of modern sequential recommender systems. It helps recommender systems to capture 'context' information about the user's activities. The Markov chain assumes that the user's next action is determined only by the current action and is uncorrelated with all previous actions in the time series. The RNN, on the other hand, is able to capture longer user sequences. In recent years, with the advancement of deep learning techniques, especially the various attention mechanisms and their architectures, it is possible to learn better user behaviour sequences. However, despite the great progress in sequence recommendation based on the above methods, both Markov chains, RNNs and various models based on attention mechanisms are encoded strictly according to user behaviour sequences, which is a unidirectional structure that limits the representation of user behaviour sequences, and the unidirectional structure is not practical in real life. In addition, these existing model structures have another drawback, they are all too idealized, thinking that the global preference of a user can be learned only from the user's behaviour sequences, in fact, even the attention mechanism can only extract the user's interest in the short term. To address the above issues, we propose a model, S-BERT, which is based on the BERT model and further improved to be more applicable in recommendation tasks. The model uses deep bidirectional self-attention to model the user's behavioural sequences, which is used to learn the user's short-term preferences, and the user's own embedding is added to learn the user's long-term preferences. The S-BERT model outperforms the same BERT-based model, BERT4Rec, with three real data sets and the NDCG@10 metric by about 2%.
(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Jun Chen, Xuefeng Fu, Kailiang Wang, Yanping Liu, and Weikun Chen "S-BERT: A BERT-based personalised recommendation model", Proc. SPIE 13181, Third International Conference on Electronic Information Engineering, Big Data, and Computer Technology (EIBDCT 2024), 131818C (19 July 2024); https://doi.org/10.1117/12.3031102
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Data modeling

Systems modeling

Education and training

Matrices

Neural networks

Deep learning

Overfitting

Back to Top