A Self-Supervised Learning Framework for Sequential Recommendation

No Thumbnail Available
File version
Author(s)
Jia, Renqi
Bai, Xu
Zhou, Xiaofei
Pan, Shirui
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2021
Size
File type(s)
Location

Shenzhen, China

License
Abstract

Sequential recommendation that aims to predict user preference with historical user interactions becomes one of the most popular tasks in the recommendation area. The existing methods concentrated on user's sequential features among exposed items have achieved good performance. However, they only rely on single item prediction optimization to learn data representation, which ignores the association between context data and sequence data. In this paper, we propose a novel self-supervised learning based sequential recommendation network (SSLRN), which contrastively learns data correlation to promote data representation of users and items. We design two auxiliary contrastive learning tasks to regularize user and item representation based on mutual information maximization (MIM). In particular, the item contrastive learning captures sequential contrast feature with sequence-item MIM, and the user contrastive learning regularizes user latent representation with user-item MIM. We evaluate our model on five real-world datasets and the experimental results show that the proposed framework significantly and consistently outperforms state-of-the-art sequential recommendation techniques.

Journal Title
Conference Title

2021 International Joint Conference on Neural Networks (IJCNN)

Book Title
Edition
Volume
Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject

Neural networks

Computer Science, Artificial Intelligence

Computer Science, Hardware & Architecture

contrastive learning

Engineering

Persistent link to this record
Citation

Jia, R; Bai, X; Zhou, X; Pan, S, A Self-Supervised Learning Framework for Sequential Recommendation, 2021 International Joint Conference on Neural Networks (IJCNN), 2021