MCL4SRec: A Sequential Recommendation Model with Multi-level Contrastive Learning

Zhuohan Hu,Bo Yang,Jialiang Lin,Jiajin Wu,Wei Liu
DOI: https://doi.org/10.23919/fusion59988.2024.10706294
2024-01-01
Abstract:Sequential recommendation (SR) plays an important role across various platforms, aiming to predict users’ next items of interest based on their historical interaction sequences. Recent SR studies have employed deep learning techniques, such as Recurrent Neural Networks and Self-Attention (SA) mechanism, demonstrating promising results. Inspired by the emergence of contrastive learning methods, some SR models have utilized contrastive learning to improve the accuracy of recommendations. However, existing SR models employing contrastive learning primarily construct positive and negative sample pairs only from user interaction sequences, i.e., through sequence-level contrastive learning. In our research, we argue that there also exists semantic similarities between items, which can be used to conduct the item-level constructive learning, resulting in better recommendation accuracy. In this paper, we propose MCL4SRec, an SA-based SR model that combines sequence-level and item-level contrastive learning to enhance recommendation accuracy. In our proposed MCL4SRec, the item-level contrastive learning module utilizes items’ category information to construct positive and negative sample pairs, capturing semantic similarities and differences between items. Additionally, in MCL4SRec, we propose to use more side information such as category and brand to further improve the accuracy of recommendations. We conduct extensive experiments on three widely-used datasets to evaluate the proposed MCL4SRec. Experimental results indicate that the average improvements compared with the recent well-known baselines range from ${7. 7 3 \%}$ to ${1 6. 1 8 \%}$ in HR and NDCG, demonstrating the effectiveness of MCL4SRec for SR tasks.
What problem does this paper attempt to address?