NILSIC-BERT4Rec: Sequential Recommendation with Non-Invasive and Interest Capturing Self-Attention Mechanism

Haolin Liu,C. L. Luo,Rui Zhang,Zhenye Liu,Shi‐Min Cai
DOI: https://doi.org/10.1007/978-981-99-3925-1_22
2023-01-01
Abstract:Recently, recommendation models based on Transformer have become a hot research topic. Despite the large number of recommendation models proposed in recent years, we argue that the use of invasive fusion when using heterogeneous auxiliary information and mixing multiple different vector spaces results in side effects in existing self-attention models. Additionally, the self-attention layer in Transformer overly focuses on distant items in user interaction sequence, lacking the ability to capture local dependencies of items. Thus, we propose a novel sequential recommendation model, Non-Invasive Long-term Short-term Interest Capturing BERT4Rec (NILSIC-BERT4Rec). NILSIC-BERT4Rec retains the ability of the representation layer and self-attention layer to output in the same vector space. We also introduce a local encoder in the self-attention layer to strengthen the role of item context in the attention function, enhancing the self-attention layer’s ability to capture local dependencies. Experiment results show that NILSIC-BERT4Rec can learn a better attention distribution and achieves the best experiment results on all datasets, with an average of 7.36% improvement on evaluation metrics.
What problem does this paper attempt to address?